From a7da9c7d655e705beceaf376a6cc6f3587be41d7 Mon Sep 17 00:00:00 2001 From: ketanjain7981 Date: Tue, 2 Dec 2025 23:35:35 +0530 Subject: [PATCH] test: verify logger warning for invalid JSON in SQLAlchemyAdapter Signed-off-by: ketanjain7981 --- .../relational/sqlalchemy/test_SqlAlchemyAdapter.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/cognee/tests/unit/infrastructure/databases/relational/sqlalchemy/test_SqlAlchemyAdapter.py b/cognee/tests/unit/infrastructure/databases/relational/sqlalchemy/test_SqlAlchemyAdapter.py index bde5b9855..abff77660 100644 --- a/cognee/tests/unit/infrastructure/databases/relational/sqlalchemy/test_SqlAlchemyAdapter.py +++ b/cognee/tests/unit/infrastructure/databases/relational/sqlalchemy/test_SqlAlchemyAdapter.py @@ -47,11 +47,17 @@ class TestSqlAlchemyAdapter: @patch( "cognee.infrastructure.databases.relational.sqlalchemy.SqlAlchemyAdapter.create_async_engine" ) + @patch("cognee.infrastructure.databases.relational.sqlalchemy.SqlAlchemyAdapter.logger") @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.""" mock_getenv.return_value = '{"timeout": 60' # Invalid JSON 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() _, kwargs = mock_create_engine.call_args assert "connect_args" in kwargs