diff options
Diffstat (limited to 'packages/meshbay-node/tests/test_quic_transport.py')
| -rw-r--r-- | packages/meshbay-node/tests/test_quic_transport.py | 20 |
1 files changed, 9 insertions, 11 deletions
diff --git a/packages/meshbay-node/tests/test_quic_transport.py b/packages/meshbay-node/tests/test_quic_transport.py index 5720043..fb28295 100644 --- a/packages/meshbay-node/tests/test_quic_transport.py +++ b/packages/meshbay-node/tests/test_quic_transport.py @@ -13,7 +13,7 @@ from cryptography.hazmat.primitives.asymmetric.ed25519 import Ed25519PrivateKey from cryptography.hazmat.primitives import serialization from meshbay_common.crypto import generate_gek, pk_to_b64 -from meshbay_node.indexer import DirectoryIndexer, GroupIndex +from meshbay_node.indexer import DirectoryIndexer from conftest import one_root from meshbay_node.transport.quic_server import QuicChunkServer, Denylist from meshbay_node.transport.quic_client import QuicChunkClient @@ -95,7 +95,7 @@ async def test_quic_chunk_roundtrip(sk_node, sk_hub, gek, shared_dir, tmp_path): @pytest.mark.asyncio async def test_quic_fetch_index(sk_node, sk_hub, gek, shared_dir, tmp_path): - """QUIC index sync returns deserializable GroupIndex.""" + """QUIC index sync returns the same message shape WebRTC sends.""" hub_pk_pem = sk_hub.public_key().public_bytes( serialization.Encoding.PEM, serialization.PublicFormat.SubjectPublicKeyInfo) @@ -121,9 +121,10 @@ async def test_quic_fetch_index(sk_node, sk_hub, gek, shared_dir, tmp_path): pk_node_b64=pk_to_b64(sk_node.public_key()), group_id="g", ) as client: - wire = await client.fetch_index() - recovered = GroupIndex.deserialize(wire, sk_node=sk_node, gek=gek) - assert recovered.count == 2 + msg = await client.fetch_index() + assert msg["group_id"] == "g" + assert len(msg["entries"]) == 2 + assert "dirs" in msg and "roots" in msg await server.stop() @@ -226,8 +227,7 @@ async def test_quic_session_resumption(sk_node, sk_hub, gek, shared_dir, tmp_pat jwt_token=token, gek=gek, pk_node_b64=pk_b64, group_id="g", ) as client: - wire = await client.fetch_index() - assert GroupIndex.deserialize(wire, sk_node=sk_node, gek=gek).count == 2 + assert len((await client.fetch_index())["entries"]) == 2 saved_ticket = client.session_ticket saved_cert = client.peer_cert_der @@ -244,8 +244,7 @@ async def test_quic_session_resumption(sk_node, sk_hub, gek, shared_dir, tmp_pat peer_cert_der=saved_cert, group_id="g", ) as client: - wire = await client.fetch_index() - assert GroupIndex.deserialize(wire, sk_node=sk_node, gek=gek).count == 2 + assert len((await client.fetch_index())["entries"]) == 2 await server.stop() @@ -281,8 +280,7 @@ async def test_quic_denylist_blocks_user(sk_node, sk_hub, gek, shared_dir, tmp_p jwt_token=token, gek=gek, pk_node_b64=pk_b64, group_id="g", ) as client: - wire = await client.fetch_index() - assert GroupIndex.deserialize(wire, sk_node=sk_node, gek=gek).count == 2 + assert len((await client.fetch_index())["entries"]) == 2 # Add user to denylist denylist.deny_user("user-001") |