From ecb5a6feb118ec13f51bfc5eb27449710a3141b9 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 21 Apr 2026 11:30:07 +0100 Subject: [PATCH] Fixing tests --- tests/unit/test_storage/test_triples_falkordb_storage.py | 4 ++-- tests/unit/test_storage/test_triples_memgraph_storage.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/unit/test_storage/test_triples_falkordb_storage.py b/tests/unit/test_storage/test_triples_falkordb_storage.py index 8276de31..c5b0848e 100644 --- a/tests/unit/test_storage/test_triples_falkordb_storage.py +++ b/tests/unit/test_storage/test_triples_falkordb_storage.py @@ -235,8 +235,8 @@ class TestFalkorDBStorageProcessor: with patch.object(processor, 'collection_exists', return_value=True): - await processor.store_triples(mock_message) - + await processor.store_triples('test_workspace', mock_message) + # Verify queries were called in the correct order expected_calls = [ # Create subject node diff --git a/tests/unit/test_storage/test_triples_memgraph_storage.py b/tests/unit/test_storage/test_triples_memgraph_storage.py index d6ca6bf5..6a0c68a2 100644 --- a/tests/unit/test_storage/test_triples_memgraph_storage.py +++ b/tests/unit/test_storage/test_triples_memgraph_storage.py @@ -313,8 +313,8 @@ class TestMemgraphStorageProcessor: with patch.object(processor, 'collection_exists', return_value=True): - await processor.store_triples(mock_message) - + await processor.store_triples('test_workspace', mock_message) + # Verify execute_query was called for create_node, create_literal, and relate_literal # (since mock_message has a literal object) assert processor.io.execute_query.call_count == 3