summaryrefslogtreecommitdiffstats
path: root/packages/meshbay-node/tests/test_quic_transport.py
diff options
context:
space:
mode:
Diffstat (limited to 'packages/meshbay-node/tests/test_quic_transport.py')
-rw-r--r--packages/meshbay-node/tests/test_quic_transport.py20
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")