@@ -22,6 +22,7 @@ def mock_get_client():
2222
2323# ── create_entity ─────────────────────────────────────────────────────────────
2424
25+
2526def test_create_entity_stores_visibility (mock_get_client ):
2627 mock_get_client .update_entities .return_value = [
2728 EntityUpdate (id = "1" , type = "guideline" , content = "test" , event = "ADD" , metadata = {"visibility" : "private" })
@@ -48,16 +49,15 @@ def test_create_entity_invalid_visibility(mock_get_client):
4849
4950
5051def test_create_entity_no_owner_id_omitted (mock_get_client ):
51- mock_get_client .update_entities .return_value = [
52- EntityUpdate (id = "1" , type = "note" , content = "x" , event = "ADD" , metadata = {})
53- ]
52+ mock_get_client .update_entities .return_value = [EntityUpdate (id = "1" , type = "note" , content = "x" , event = "ADD" , metadata = {})]
5453 create_entity (content = "x" , entity_type = "note" )
5554 entities = mock_get_client .update_entities .call_args [1 ]["entities" ]
5655 assert "owner_id" not in entities [0 ].metadata
5756
5857
5958# ── publish_entity ─────────────────────────────────────────────────────────────
6059
60+
6161def _make_entity (entity_id = "42" , visibility = "private" , owner_id = None ):
6262 meta = {"visibility" : visibility }
6363 if owner_id :
@@ -92,6 +92,7 @@ def test_publish_entity_no_user_id(mock_get_client):
9292
9393def test_publish_entity_not_found (mock_get_client ):
9494 from altk_evolve .schema .exceptions import EvolveException
95+
9596 mock_get_client .patch_entity_metadata .side_effect = EvolveException ("Entity '99' not found" )
9697
9798 result = json .loads (publish_entity (entity_id = "99" , user_id = "alice" ))
@@ -100,6 +101,7 @@ def test_publish_entity_not_found(mock_get_client):
100101
101102# ── unpublish_entity ───────────────────────────────────────────────────────────
102103
104+
103105def test_unpublish_entity_reverts_to_private (mock_get_client ):
104106 updated = _make_entity (visibility = "private" )
105107 mock_get_client .patch_entity_metadata .return_value = updated
@@ -114,6 +116,7 @@ def test_unpublish_entity_reverts_to_private(mock_get_client):
114116
115117# ── get_entities with include_public ──────────────────────────────────────────
116118
119+
117120def test_get_entities_include_public_annotates_results (mock_get_client ):
118121 private = _make_entity (entity_id = "1" , visibility = "private" )
119122 public = _make_entity (entity_id = "2" , visibility = "public" , owner_id = "bob" )
@@ -149,11 +152,10 @@ def test_get_entities_deduplicates_public_already_in_private(mock_get_client):
149152
150153# ── backward compatibility ─────────────────────────────────────────────────────
151154
155+
152156def test_create_entity_backward_compat_no_visibility_args (mock_get_client ):
153157 """Calling create_entity without new args should still work and default to private."""
154- mock_get_client .update_entities .return_value = [
155- EntityUpdate (id = "1" , type = "fact" , content = "x" , event = "ADD" , metadata = {})
156- ]
158+ mock_get_client .update_entities .return_value = [EntityUpdate (id = "1" , type = "fact" , content = "x" , event = "ADD" , metadata = {})]
157159 result = json .loads (create_entity (content = "x" , entity_type = "fact" ))
158160 assert "error" not in result
159161 entities = mock_get_client .update_entities .call_args [1 ]["entities" ]
0 commit comments