test(typed state): add unit test to verify child server handling

Justineantoine · jourdain · commit d78810a6e189 · 2025-09-24T10:50:23.000-06:00
diff --git a/tests/test_typed_state.py b/tests/test_typed_state.py
@@ -17,11 +17,20 @@
 
 
 @pytest.fixture
-def state():
-    server = Server()
+def server():
+    return Server()
+
+@pytest.fixture
+def state(server):
     server.state.ready()
     return server.state
 
+@pytest.fixture
+def child_state(server):
+    child_server = server.create_child_server(prefix="child_")
+    child_server.state.ready()
+    return child_server.state
+
 
 @dataclass
 class MyData:
@@ -101,6 +110,17 @@ def on_a_change(**_):
     mock.assert_called_once_with(53)
 
 
+def test_can_handle_child_server(state, child_state):
+    typed_state_main = TypedState(state, MyData)
+    typed_state_child = TypedState(child_state, MyData)
+
+    typed_state_main.data.a = 40
+    typed_state_child.data.a = 3
+
+    assert typed_state_main.data.a != typed_state_child.data.a
+    assert typed_state_child.data.a == state[f"child_{typed_state_child.name.a}"]
+
+
 class MyEnum(Enum):
     A = auto()
     B = auto()
