summaryrefslogtreecommitdiffstats
path: root/packages/meshbay-node/tests/conftest.py
diff options
context:
space:
mode:
Diffstat (limited to 'packages/meshbay-node/tests/conftest.py')
-rw-r--r--packages/meshbay-node/tests/conftest.py31
1 files changed, 31 insertions, 0 deletions
diff --git a/packages/meshbay-node/tests/conftest.py b/packages/meshbay-node/tests/conftest.py
index 3dc9cd9..ba86c13 100644
--- a/packages/meshbay-node/tests/conftest.py
+++ b/packages/meshbay-node/tests/conftest.py
@@ -39,3 +39,34 @@ def one_root(path: Path, *, name: str = "", kind: str = "generic",
"""
return RootSet.build([{"path": str(path), "name": name, "kind": kind,
"writable": writable}])
+
+
+def sealed_upload(session, *, filename: str, data: bytes,
+ chunk_index: int = 0, total_chunks: int = 1,
+ dir: str = "", root: str = "",
+ upload_id: str = "up-test") -> dict:
+ """
+ A `file_upload` message as the shipping client builds one (MNP 2.0).
+
+ Built through `file_upload_wire`, not by hand: a test that assembles the
+ wire shape itself is a second encoder, and a second encoder is how
+ `file_chunk` and `index_sync` forked between the transports (finding C6)
+ with nobody noticing. The key and the AAD are taken off the session, so
+ these agree with the handler by construction rather than by copying.
+ """
+ from meshbay_common.protocol import file_upload_wire
+
+ ctx = session._group_ctx()
+ return file_upload_wire(
+ ctx["gek"], session._group_id or "",
+ upload_id=upload_id, chunk_index=chunk_index, total_chunks=total_chunks,
+ filename=filename, data=data, dir=dir, root=root,
+ )
+
+
+def opened_ack(session, msg: dict) -> dict:
+ """The payload of a `file_upload_ack` the node sent, opened as a client would."""
+ from meshbay_common.protocol import file_upload_ack_payload
+
+ ctx = session._group_ctx()
+ return file_upload_ack_payload(ctx["gek"], session._group_id or "", msg)