mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2026-02-12 16:10:57 +00:00
added WebSocket transport support
Signed-off-by: GautamBytes <manchandanigautam@gmail.com>
This commit is contained in:
72
tests/transport/test_websocket.py
Normal file
72
tests/transport/test_websocket.py
Normal file
@ -0,0 +1,72 @@
|
||||
from collections.abc import Sequence
|
||||
from typing import Any
|
||||
|
||||
import pytest
|
||||
from multiaddr import Multiaddr
|
||||
|
||||
from libp2p.crypto.secp256k1 import create_new_key_pair
|
||||
from libp2p.custom_types import TProtocol
|
||||
from libp2p.host.basic_host import BasicHost
|
||||
from libp2p.network.swarm import Swarm
|
||||
from libp2p.peer.id import ID
|
||||
from libp2p.peer.peerinfo import PeerInfo
|
||||
from libp2p.peer.peerstore import PeerStore
|
||||
from libp2p.security.insecure.transport import InsecureTransport
|
||||
from libp2p.stream_muxer.mplex.mplex import MPLEX_PROTOCOL_ID, Mplex
|
||||
from libp2p.transport.upgrader import TransportUpgrader
|
||||
from libp2p.transport.websocket.transport import WebsocketTransport
|
||||
|
||||
PLAINTEXT_PROTOCOL_ID = "/plaintext/1.0.0"
|
||||
|
||||
|
||||
async def make_host(
|
||||
listen_addrs: Sequence[Multiaddr] | None = None,
|
||||
) -> tuple[BasicHost, Any | None]:
|
||||
# 1) Identity
|
||||
key_pair = create_new_key_pair()
|
||||
peer_id = ID.from_pubkey(key_pair.public_key)
|
||||
peer_store = PeerStore()
|
||||
peer_store.add_key_pair(peer_id, key_pair)
|
||||
|
||||
# 2) Upgrader
|
||||
upgrader = TransportUpgrader(
|
||||
secure_transports_by_protocol={
|
||||
TProtocol(PLAINTEXT_PROTOCOL_ID): InsecureTransport(key_pair)
|
||||
},
|
||||
muxer_transports_by_protocol={TProtocol(MPLEX_PROTOCOL_ID): Mplex},
|
||||
)
|
||||
|
||||
# 3) Transport + Swarm + Host
|
||||
transport = WebsocketTransport()
|
||||
swarm = Swarm(peer_id, peer_store, upgrader, transport)
|
||||
host = BasicHost(swarm)
|
||||
|
||||
# 4) Optionally run/listen
|
||||
ctx = None
|
||||
if listen_addrs:
|
||||
ctx = host.run(listen_addrs)
|
||||
await ctx.__aenter__()
|
||||
|
||||
return host, ctx
|
||||
|
||||
|
||||
@pytest.mark.trio
|
||||
async def test_websocket_dial_and_listen():
|
||||
# Start server
|
||||
server_host, server_ctx = await make_host([Multiaddr("/ip4/127.0.0.1/tcp/0/ws")])
|
||||
# Client
|
||||
client_host, _ = await make_host(None)
|
||||
|
||||
# Dial
|
||||
peer_info = PeerInfo(server_host.get_id(), server_host.get_addrs())
|
||||
await client_host.connect(peer_info)
|
||||
|
||||
# Verify connections
|
||||
assert client_host.get_network().connections.get(server_host.get_id())
|
||||
assert server_host.get_network().connections.get(client_host.get_id())
|
||||
|
||||
# Cleanup
|
||||
await client_host.close()
|
||||
if server_ctx:
|
||||
await server_ctx.__aexit__(None, None, None)
|
||||
await server_host.close()
|
||||
Reference in New Issue
Block a user