Add test case test_get_key_creates_shares_on_hubs

Author: brunorijsman

system_tests/system_test_common.py

@@ -163,7 +163,7 @@ def check_wait_for_all_nodes_started_output(output):
     )
 
 
-def status_topology():
+def status_topology() -> dict:
     """
     Get status for a topology.
     """
@@ -176,7 +176,7 @@ def status_topology():
     return status
 
 
-def status_node(node_type, node_name):
+def status_node(node_type: str, node_name: str) -> dict:
     """
     Get status for a node.
     """

system_tests/test_share_storage.py

@@ -0,0 +1,33 @@
+"""
+Test share storage at hubs.
+"""
+
+import sys
+import pytest
+from . import system_test_common
+
+
+@pytest.fixture(autouse=True)
+def setup_and_teardown():
+    """
+    Setup and teardown for each test.
+    """
+    system_test_common.start_topology()
+    yield
+    system_test_common.stop_topology()
+
+
+def test_get_key_creates_shares_on_hubs():
+    """
+    Test that getting a key on the master client results in shares being created on the hubs.
+    """
+    key_id = system_test_common.get_key("sam", "sofia")
+    status = system_test_common.status_node("hub", "hank")
+    assert "shares" in status
+    shares = status["shares"]
+    assert len(shares) == 1
+    share = shares[0]
+    assert share["key_id"] == key_id
+    assert share["master_sae_id"] == "sam"
+    assert share["slave_sae_id"] == "sofia"
+    assert share["share_index"] == 0