apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-25 05:01:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-04 00:57:25 +01:00
parent 9dfdef0110
commit 0fad26872c
2 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/unit/test_cores/test_knowledge_manager.py
View file

@ -56,7 +56,7 @@ def knowledge_manager(mock_flow_config):
with patch('trustgraph.cores.knowledge.KnowledgeTableStore') as mock_store_class: with patch('trustgraph.cores.knowledge.KnowledgeTableStore') as mock_store_class:
manager = KnowledgeManager( manager = KnowledgeManager(
cassandra_host=["localhost"], cassandra_host=["localhost"],
cassandra_user="test_user", cassandra_username="test_user",
cassandra_password="test_pass", cassandra_password="test_pass",
keyspace="test_keyspace", keyspace="test_keyspace",
flow_config=mock_flow_config flow_config=mock_flow_config

18
tests/unit/test_storage/test_cassandra_config_integration.py
View file

@ -216,7 +216,7 @@ class TestKgStoreConfiguration:
# Verify KnowledgeTableStore was called with resolved config # Verify KnowledgeTableStore was called with resolved config
mock_table_store.assert_called_once_with( mock_table_store.assert_called_once_with(
cassandra_host=['kg-env-host1', 'kg-env-host2', 'kg-env-host3'], cassandra_host=['kg-env-host1', 'kg-env-host2', 'kg-env-host3'],
cassandra_user='kg-env-user', cassandra_username='kg-env-user',
cassandra_password='kg-env-pass', cassandra_password='kg-env-pass',
keyspace='knowledge' keyspace='knowledge'
) )
@ -237,28 +237,28 @@ class TestKgStoreConfiguration:
# Verify KnowledgeTableStore was called with explicit config # Verify KnowledgeTableStore was called with explicit config
mock_table_store.assert_called_once_with( mock_table_store.assert_called_once_with(
cassandra_host=['explicit-host'], cassandra_host=['explicit-host'],
cassandra_user='explicit-user', cassandra_username='explicit-user',
cassandra_password='explicit-pass', cassandra_password='explicit-pass',
keyspace='knowledge' keyspace='knowledge'
) )
@patch('trustgraph.storage.knowledge.store.KnowledgeTableStore') @patch('trustgraph.storage.knowledge.store.KnowledgeTableStore')
def test_backward_compatibility_cassandra_user(self, mock_table_store): def test_no_backward_compatibility_cassandra_user(self, mock_table_store):
"""Test backward compatibility with cassandra_user parameter.""" """Test that cassandra_user parameter is no longer supported."""
mock_store_instance = MagicMock() mock_store_instance = MagicMock()
mock_table_store.return_value = mock_store_instance mock_table_store.return_value = mock_store_instance
processor = KgStore( processor = KgStore(
taskgroup=MagicMock(), taskgroup=MagicMock(),
cassandra_host='compat-host', cassandra_host='compat-host',
cassandra_user='compat-user', # Old parameter name cassandra_user='compat-user', # Old parameter name - should be ignored
cassandra_password='compat-pass' cassandra_password='compat-pass'
) )
# Should still work with old parameter name # cassandra_user should be ignored
mock_table_store.assert_called_once_with( mock_table_store.assert_called_once_with(
cassandra_host=['compat-host'], cassandra_host=['compat-host'],
cassandra_user='compat-user', cassandra_username=None, # Should be None since cassandra_user is ignored
cassandra_password='compat-pass', cassandra_password='compat-pass',
keyspace='knowledge' keyspace='knowledge'
) )
@ -275,7 +275,7 @@ class TestKgStoreConfiguration:
# Should use defaults # Should use defaults
mock_table_store.assert_called_once_with( mock_table_store.assert_called_once_with(
cassandra_host=['cassandra'], cassandra_host=['cassandra'],
cassandra_user=None, cassandra_username=None,
cassandra_password=None, cassandra_password=None,
keyspace='knowledge' keyspace='knowledge'
) )
@ -423,7 +423,7 @@ class TestConfigurationPriorityIntegration:
# Verify correct priority resolution # Verify correct priority resolution
mock_table_store.assert_called_once_with( mock_table_store.assert_called_once_with(
cassandra_host=['param-host'], # From parameter cassandra_host=['param-host'], # From parameter
cassandra_user='env-user', # From environment cassandra_username='env-user', # From environment
cassandra_password='env-pass', # From environment cassandra_password='env-pass', # From environment
keyspace='knowledge' keyspace='knowledge'
) )