gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test: verify logger warning for invalid JSON in SQLAlchemyAdapter

Browse source 
Signed-off-by: ketanjain7981 <ketan.jain@think41.com>
This commit is contained in:
ketanjain7981 2025-12-02 23:35:35 +05:30
parent 4f3a1bcf01
commit a7da9c7d65
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/tests/unit/infrastructure/databases/relational/sqlalchemy/test_SqlAlchemyAdapter.py
View file

@ -47,11 +47,17 @@ class TestSqlAlchemyAdapter:
@patch( @patch(
"cognee.infrastructure.databases.relational.sqlalchemy.SqlAlchemyAdapter.create_async_engine" "cognee.infrastructure.databases.relational.sqlalchemy.SqlAlchemyAdapter.create_async_engine"
) )
@patch("cognee.infrastructure.databases.relational.sqlalchemy.SqlAlchemyAdapter.logger")
@patch("os.getenv") @patch("os.getenv")
def test_sqlite_with_invalid_json_env_var(self, mock_getenv, mock_create_engine): def test_sqlite_with_invalid_json_env_var(self, mock_getenv, mock_logger, mock_create_engine):
"""Test that SQLite connection uses default timeout when env var has invalid JSON.""" """Test that SQLite connection uses default timeout when env var has invalid JSON."""
mock_getenv.return_value = '{"timeout": 60' # Invalid JSON mock_getenv.return_value = '{"timeout": 60' # Invalid JSON
SQLAlchemyAdapter("sqlite:///test.db") SQLAlchemyAdapter("sqlite:///test.db")
mock_logger.warning.assert_called_with(
"Failed to parse DATABASE_CONNECT_ARGS as JSON, ignoring"
)
mock_create_engine.assert_called_once() mock_create_engine.assert_called_once()
_, kwargs = mock_create_engine.call_args _, kwargs = mock_create_engine.call_args
assert "connect_args" in kwargs assert "connect_args" in kwargs