From 364e30b0f9f2f988cb982404fcd22c0ddceed527 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 21 Apr 2026 11:34:21 +0100 Subject: [PATCH] Fixing tests --- .../test_query/test_triples_falkordb_query.py | 18 +++++++++--------- .../test_query/test_triples_memgraph_query.py | 18 +++++++++--------- .../test_query/test_triples_neo4j_query.py | 8 ++++---- 3 files changed, 22 insertions(+), 22 deletions(-) diff --git a/tests/unit/test_query/test_triples_falkordb_query.py b/tests/unit/test_query/test_triples_falkordb_query.py index d5c047d7..d1230720 100644 --- a/tests/unit/test_query/test_triples_falkordb_query.py +++ b/tests/unit/test_query/test_triples_falkordb_query.py @@ -131,7 +131,7 @@ class TestFalkorDBQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_graph.query.call_count == 2 @@ -172,7 +172,7 @@ class TestFalkorDBQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_graph.query.call_count == 2 @@ -217,7 +217,7 @@ class TestFalkorDBQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_graph.query.call_count == 2 @@ -262,7 +262,7 @@ class TestFalkorDBQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_graph.query.call_count == 2 @@ -307,7 +307,7 @@ class TestFalkorDBQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_graph.query.call_count == 2 @@ -352,7 +352,7 @@ class TestFalkorDBQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_graph.query.call_count == 2 @@ -397,7 +397,7 @@ class TestFalkorDBQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_graph.query.call_count == 2 @@ -442,7 +442,7 @@ class TestFalkorDBQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_graph.query.call_count == 2 @@ -484,7 +484,7 @@ class TestFalkorDBQueryProcessor: # Should raise the exception with pytest.raises(Exception, match="Database connection failed"): - await processor.query_triples(query) + await processor.query_triples('test_user', query) def test_add_args_method(self): """Test that add_args properly configures argument parser""" diff --git a/tests/unit/test_query/test_triples_memgraph_query.py b/tests/unit/test_query/test_triples_memgraph_query.py index f4222af1..1d1cf8a3 100644 --- a/tests/unit/test_query/test_triples_memgraph_query.py +++ b/tests/unit/test_query/test_triples_memgraph_query.py @@ -130,7 +130,7 @@ class TestMemgraphQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -172,7 +172,7 @@ class TestMemgraphQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -218,7 +218,7 @@ class TestMemgraphQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -264,7 +264,7 @@ class TestMemgraphQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -310,7 +310,7 @@ class TestMemgraphQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -356,7 +356,7 @@ class TestMemgraphQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -402,7 +402,7 @@ class TestMemgraphQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -448,7 +448,7 @@ class TestMemgraphQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -488,7 +488,7 @@ class TestMemgraphQueryProcessor: # Should raise the exception with pytest.raises(Exception, match="Database connection failed"): - await processor.query_triples(query) + await processor.query_triples('test_user', query) def test_add_args_method(self): """Test that add_args properly configures argument parser""" diff --git a/tests/unit/test_query/test_triples_neo4j_query.py b/tests/unit/test_query/test_triples_neo4j_query.py index e379ed21..6765fd5b 100644 --- a/tests/unit/test_query/test_triples_neo4j_query.py +++ b/tests/unit/test_query/test_triples_neo4j_query.py @@ -130,7 +130,7 @@ class TestNeo4jQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -172,7 +172,7 @@ class TestNeo4jQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -218,7 +218,7 @@ class TestNeo4jQueryProcessor: limit=100 ) - result = await processor.query_triples(query) + result = await processor.query_triples('test_user', query) # Verify both literal and URI queries were executed assert mock_driver.execute_query.call_count == 2 @@ -258,7 +258,7 @@ class TestNeo4jQueryProcessor: # Should raise the exception with pytest.raises(Exception, match="Database connection failed"): - await processor.query_triples(query) + await processor.query_triples('test_user', query) def test_add_args_method(self): """Test that add_args properly configures argument parser"""