From e04b91fa63369fb0bdaf3ac7c24bd159394e3c83 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Mon, 21 Jul 2025 14:20:02 +0100 Subject: [PATCH] Fix the tests following metadata change in agent extractor --- .../test_knowledge_graph/test_agent_extraction.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/tests/unit/test_knowledge_graph/test_agent_extraction.py b/tests/unit/test_knowledge_graph/test_agent_extraction.py index 9e8662e3..be5553df 100644 --- a/tests/unit/test_knowledge_graph/test_agent_extraction.py +++ b/tests/unit/test_knowledge_graph/test_agent_extraction.py @@ -354,7 +354,12 @@ class TestAgentKgExtractor: mock_publisher.send.assert_called_once() sent_triples = mock_publisher.send.call_args[0][0] assert isinstance(sent_triples, Triples) - assert sent_triples.metadata == sample_metadata + # Check metadata fields individually since implementation creates new Metadata object + assert sent_triples.metadata.id == sample_metadata.id + assert sent_triples.metadata.user == sample_metadata.user + assert sent_triples.metadata.collection == sample_metadata.collection + # Note: metadata.metadata is now empty array in the new implementation + assert sent_triples.metadata.metadata == [] assert len(sent_triples.triples) == 1 assert sent_triples.triples[0].s.value == "test:subject" @@ -375,7 +380,12 @@ class TestAgentKgExtractor: mock_publisher.send.assert_called_once() sent_contexts = mock_publisher.send.call_args[0][0] assert isinstance(sent_contexts, EntityContexts) - assert sent_contexts.metadata == sample_metadata + # Check metadata fields individually since implementation creates new Metadata object + assert sent_contexts.metadata.id == sample_metadata.id + assert sent_contexts.metadata.user == sample_metadata.user + assert sent_contexts.metadata.collection == sample_metadata.collection + # Note: metadata.metadata is now empty array in the new implementation + assert sent_contexts.metadata.metadata == [] assert len(sent_contexts.entities) == 1 assert sent_contexts.entities[0].entity.value == "test:entity"