fix unit tests

Author: aponcedeleonch

src/codegate/db/connection.py

@@ -31,7 +31,7 @@ def __init__(self, sqlite_path: Optional[str] = None):
             )
         self._db_path = Path(sqlite_path).absolute()
         self._db_path.parent.mkdir(parents=True, exist_ok=True)
-        logger.info(f"Initializing DB from path: {self._db_path}")
+        logger.debug(f"Initializing DB from path: {self._db_path}")
         engine_dict = {
             "url": f"sqlite+aiosqlite:///{self._db_path}",
             "echo": False,  # Set to False in production

tests/dashboard/test_post_processing.py

@@ -150,7 +150,7 @@ async def test_parse_output(output_dict, expected_str, expected_chat_id):
 @pytest.mark.asyncio
 @pytest.mark.parametrize("request_msg_str", ["Hello", None])
 @pytest.mark.parametrize("output_msg_str", ["Hello, how can I help you?", None])
-@pytest.mark.parametrize("chat_id", ["chatcmpl-AaQw9O1O2u360mhba5UbMoPwFgqEl", None])
+@pytest.mark.parametrize("chat_id", ["chatcmpl-AaQw9O1O2u360mhba5UbMoPwFgqEl"])
 @pytest.mark.parametrize(
     "row",
     [
@@ -181,12 +181,13 @@ async def test_parse_get_prompt_with_output(request_msg_str, output_msg_str, cha
             mock_parse_request.assert_called_once()
             mock_parse_output.assert_called_once()
 
-            if any([request_msg_str is None, output_msg_str is None, chat_id is None]):
+            if request_msg_str is None:
                 assert result is None
             else:
                 assert result.question_answer.question.message == request_msg_str
-                assert result.question_answer.answer.message == output_msg_str
-                assert result.chat_id == chat_id
+                if output_msg_str is not None:
+                    assert result.question_answer.answer.message == output_msg_str
+                    assert result.chat_id == chat_id
                 assert result.provider == "provider"
                 assert result.type == "chat"
                 assert result.request_timestamp == timestamp_now