Add shamir create test.

Scille · May 16, 2024 · 8cd5970 · 8cd5970
1 parent f8a026a
commit 8cd5970
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/server/parsec/components/memory/factory.py b/server/parsec/components/memory/factory.py
@@ -18,6 +18,7 @@
 from parsec.components.memory.pki import MemoryPkiEnrollmentComponent
 from parsec.components.memory.realm import MemoryRealmComponent
 from parsec.components.memory.sequester import MemorySequesterComponent
+from parsec.components.memory.shamir import MemoryShamirComponent
 from parsec.components.memory.user import MemoryUserComponent
 from parsec.components.memory.vlob import MemoryVlobComponent
 from parsec.config import BackendConfig
@@ -44,6 +45,7 @@ async def components_factory(config: BackendConfig) -> AsyncGenerator[dict[str,
             ping = MemoryPingComponent(event_bus)
             pki = MemoryPkiEnrollmentComponent(data, event_bus)
             sequester = MemorySequesterComponent(data, event_bus)
+            shamir = MemoryShamirComponent()
             blockstore = blockstore_factory(config.blockstore_config, mocked_data=data)
             block = MemoryBlockComponent(data, blockstore)
             events = MemoryEventsComponent(data, config, event_bus)

diff --git a/server/parsec/components/memory/shamir.py b/server/parsec/components/memory/shamir.py
@@ -0,0 +1,4 @@
+from parsec.components.shamir import BaseShamirComponent
+
+class MemoryShamirComponent(BaseShamirComponent):
+    pass
diff --git a/server/parsec/components/shamir.py b/server/parsec/components/shamir.py
@@ -0,0 +1,8 @@
+from parsec.api import api
+from parsec._parsec import authenticated_cmds
+from parsec.client_context import AuthenticatedClientContext
+
+class BaseShamirComponent:
+    @api
+    async def create_shared_recovery_device(self, client_ctx: AuthenticatedClientContext, req: authenticated_cmds.latest.shamir_recovery_setup.Req) -> authenticated_cmds.latest.shamir_recovery_setup.Rep:
+        pass