fix(get_instance): return None with invalid id

Author: jourdain

src/trame_dataclass/v2.py

@@ -442,7 +442,7 @@ def decode_dataclass_dict(data):
 def get_instance(instance_id: str):
     # print(f"get_instance({instance_id})")
     # print(" => ", INSTANCES[instance_id])
-    return INSTANCES[instance_id]
+    return INSTANCES.get(instance_id)
 
 
 class ServerOnly:

tests/test_v2.py

@@ -5,6 +5,7 @@
     StateDataModel,
     Sync,
     TypeValidation,
+    get_instance,
 )
 
 
@@ -82,3 +83,12 @@ class BasicSerial(StateDataModel):
     assert isinstance(set_as_list, list)
     for obj in data.f:
         assert obj._id in set_as_list
+
+
+def test_get_instance():
+    class Dummy(StateDataModel):
+        a = Sync(int, 0)
+
+    inst = Dummy()
+    assert get_instance(inst._id) is inst
+    assert get_instance("not an id") is None