chore: lint unit tests, setup pip compiler for requirements

Author: lpm0073

.pre-commit-config.yaml

@@ -7,7 +7,7 @@ repos:
     hooks:
       - id: codespell
         args: ["--ignore-words=codespell.txt"]
-        exclude: '(\.svg$|data/wfu-1-metaai\.yaml)'
+        exclude: '(\.svg$|data/wfu-1-metaai\.yaml|^requirements/)'
   - repo: https://github.com/pre-commit/mirrors-prettier
     rev: v4.0.0-alpha.8
     hooks:

Dockerfile

@@ -25,13 +25,14 @@ FROM base AS requirements
 WORKDIR /dist
 
 # Copy the current directory contents into the container at /app
-COPY requirements/prod.txt requirements.txt
+COPY requirements/base.txt base.txt
+COPY requirements/local.txt local.txt
 
 # Set environment variables
 ENV PYTHONPATH=/dist
 
 # Install any needed packages specified in requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r base.txt
 
 # Install Python dependencies for the local environment for cases where
 # we're going to run python unit tests in the Docker container.

app/database.py

@@ -2,7 +2,7 @@
 """Database connection and utilities for MySQL."""
 
 from contextlib import contextmanager
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, Iterator, List, Optional
 
 import pymysql
 
@@ -73,7 +73,7 @@ def get_connection(self) -> pymysql.Connection:
             raise pymysql.Error(f"Failed to connect to MySQL database: {e}")
 
     @contextmanager
-    def get_cursor(self):
+    def get_cursor(self) -> Iterator[pymysql.cursors.DictCursor]:
         """
         Context manager for database operations with automatic connection handling.
 
@@ -113,7 +113,7 @@ def execute_query(self, query: str, params: Optional[tuple] = None) -> List[Dict
         logger.debug("Executing query: %s with params: %s", query, params)
         with self.get_cursor() as cursor:
             cursor.execute(query, params or ())
-            return cursor.fetchall()
+            return list(cursor.fetchall())
 
     def execute_update(self, query: str, params: Optional[tuple] = None) -> int:
         """

app/tests/const.py

@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+"""
+Common constants for untit tests.
+"""
+
+import os
+import sys
+from pathlib import Path
+
+
+HERE = os.path.abspath(os.path.dirname(__file__))
+PROJECT_ROOT = str(Path(HERE).parent.parent)
+PYTHON_ROOT = str(Path(PROJECT_ROOT).parent)
+if PYTHON_ROOT not in sys.path:
+    sys.path.append(PYTHON_ROOT)  # noqa: E402
+
+__all__ = ["HERE", "PROJECT_ROOT", "PYTHON_ROOT"]

app/tests/test_agent.py

@@ -4,23 +4,13 @@
 """Test application."""
 
 # python stuff
-import os
-import sys
 import unittest
-from pathlib import Path
 from unittest.mock import patch
 
 from app import agent
 from app.agent import main  # noqa: E402
 
 
-HERE = os.path.abspath(os.path.dirname(__file__))
-PROJECT_ROOT = str(Path(HERE).parent.parent)
-PYTHON_ROOT = str(Path(PROJECT_ROOT).parent)
-if PYTHON_ROOT not in sys.path:
-    sys.path.append(PYTHON_ROOT)  # noqa: E402
-
-
 class TestApplication(unittest.TestCase):
     """Test application."""
 

app/tests/test_database.py

@@ -4,10 +4,7 @@
 """Test database connectivity."""
 
 # python stuff
-import os
-import sys
 import unittest
-from pathlib import Path
 from unittest.mock import MagicMock, patch
 
 import pymysql
@@ -16,13 +13,6 @@
 from app.logging_config import get_logger
 
 
-HERE = os.path.abspath(os.path.dirname(__file__))
-PROJECT_ROOT = str(Path(HERE).parent.parent)
-PYTHON_ROOT = str(Path(PROJECT_ROOT).parent)
-if PYTHON_ROOT not in sys.path:
-    sys.path.append(PYTHON_ROOT)  # noqa: E402
-
-
 logger = get_logger(__name__)
 
 

app/tests/test_prompt.py

@@ -4,10 +4,7 @@
 """Test prompt."""
 
 # python stuff
-import os
-import sys
 import unittest
-from pathlib import Path
 from unittest.mock import MagicMock, patch
 
 import httpx
@@ -16,13 +13,6 @@
 from app import prompt
 
 
-HERE = os.path.abspath(os.path.dirname(__file__))
-PROJECT_ROOT = str(Path(HERE).parent.parent)
-PYTHON_ROOT = str(Path(PROJECT_ROOT).parent)
-if PYTHON_ROOT not in sys.path:
-    sys.path.append(PYTHON_ROOT)  # noqa: E402
-
-
 class DummyResponse:
     def __init__(self):
         self.request = None

app/tests/test_stackademy.py

@@ -4,24 +4,14 @@
 """Test Stackademy application."""
 
 # python stuff
-import os
-import sys
 import unittest
-from pathlib import Path
 from unittest.mock import Mock, patch
 
 from app.exceptions import ConfigurationException
 from app.logging_config import get_logger
 from app.stackademy import Stackademy
 
 
-HERE = os.path.abspath(os.path.dirname(__file__))
-PROJECT_ROOT = str(Path(HERE).parent.parent)
-PYTHON_ROOT = str(Path(PROJECT_ROOT).parent)
-if PYTHON_ROOT not in sys.path:
-    sys.path.append(PYTHON_ROOT)  # noqa: E402
-
-
 logger = get_logger(__name__)
 
 

app/tests/test_structured_outputs.py

@@ -4,22 +4,12 @@
 """Test structured outputs."""
 
 # python stuff
-import os
-import sys
 import unittest
-from pathlib import Path
 from unittest.mock import PropertyMock, patch
 
 import app.structured_outputs as so
 
 
-HERE = os.path.abspath(os.path.dirname(__file__))
-PROJECT_ROOT = str(Path(HERE).parent.parent)
-PYTHON_ROOT = str(Path(PROJECT_ROOT).parent)
-if PYTHON_ROOT not in sys.path:
-    sys.path.append(PYTHON_ROOT)  # noqa: E402
-
-
 class TestStructuredOutputs(unittest.TestCase):
     """Test structured outputs."""
 

app/tests/test_utils.py

@@ -4,22 +4,12 @@
 """Test utils."""
 
 # python stuff
-import os
-import sys
 import unittest
-from pathlib import Path
 from unittest.mock import patch
 
 from app import utils
 
 
-HERE = os.path.abspath(os.path.dirname(__file__))
-PROJECT_ROOT = str(Path(HERE).parent.parent)
-PYTHON_ROOT = str(Path(PROJECT_ROOT).parent)
-if PYTHON_ROOT not in sys.path:
-    sys.path.append(PYTHON_ROOT)  # noqa: E402
-
-
 class TestUtils(unittest.TestCase):
     """Test utils."""