From d2c8319b6bb7a19ca8001e58ccd8b87210639a36 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Mon, 9 Feb 2026 14:02:39 +0000 Subject: [PATCH] Fix tests --- tests/integration/test_kg_extract_store_integration.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/integration/test_kg_extract_store_integration.py b/tests/integration/test_kg_extract_store_integration.py index 7a3af18c..c22524fd 100644 --- a/tests/integration/test_kg_extract_store_integration.py +++ b/tests/integration/test_kg_extract_store_integration.py @@ -16,7 +16,7 @@ from trustgraph.extract.kg.definitions.extract import Processor as DefinitionsPr from trustgraph.extract.kg.relationships.extract import Processor as RelationshipsProcessor from trustgraph.storage.knowledge.store import Processor as KnowledgeStoreProcessor from trustgraph.schema import Chunk, Triple, Triples, Metadata, Value, Error -from trustgraph.schema import EntityContext, EntityContexts, GraphEmbeddings +from trustgraph.schema import EntityContext, EntityContexts, GraphEmbeddings, EntityEmbeddings from trustgraph.rdf import TRUSTGRAPH_ENTITIES, DEFINITION, RDF_LABEL, SUBJECT_OF @@ -501,12 +501,12 @@ class TestKnowledgeGraphPipelineIntegration: await definitions_processor.on_message(mock_msg, mock_consumer, mock_flow_context) # Assert - # Should still call producers but with empty results + # Should NOT call producers with empty results (avoids Cassandra NULL issues) triples_producer = mock_flow_context("triples") entity_contexts_producer = mock_flow_context("entity-contexts") - - triples_producer.send.assert_called_once() - entity_contexts_producer.send.assert_called_once() + + triples_producer.send.assert_not_called() + entity_contexts_producer.send.assert_not_called() @pytest.mark.asyncio async def test_invalid_extraction_format_handling(self, definitions_processor, mock_flow_context, sample_chunk):