varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-12 08:00:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

small tweak

Browse Source 
Signed-off-by: GautamBytes <manchandanigautam@gmail.com>
This commit is contained in:
GautamBytes
2025-07-20 09:30:21 +00:00
parent 187418378a
commit 227a5c6441
2 changed files with 11 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/transport/test_websocket.py
View File

@ -22,13 +22,13 @@ PLAINTEXT_PROTOCOL_ID = "/plaintext/1.0.0"
async def make_host(
listen_addrs: Sequence[Multiaddr] | None = None,
) -> tuple[BasicHost, Any | None]:
# 1) Identity
# 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
upgrader = TransportUpgrader(
secure_transports_by_protocol={
TProtocol(PLAINTEXT_PROTOCOL_ID): InsecureTransport(key_pair)
@ -36,12 +36,12 @@ async def make_host(
muxer_transports_by_protocol={TProtocol(MPLEX_PROTOCOL_ID): Mplex},
)
# 3) Transport + Swarm + Host
# Transport + Swarm + Host
transport = WebsocketTransport()
swarm = Swarm(peer_id, peer_store, upgrader, transport)
host = BasicHost(swarm)
# 4) Optionally run/listen
# Optionally run/listen
ctx = None
if listen_addrs:
ctx = host.run(listen_addrs)
@ -52,20 +52,15 @@ async def make_host(
@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)