Merge pull request #2 from EmiM/chore/organize-tests

Reorganize vector mode tests into proper unit test structure with typ…

Author: fluffypony

.github/workflows/unit-tests.yml

@@ -0,0 +1,39 @@
+name: Unit Tests
+
+on:
+  push:
+    branches: [ main, develop ]
+  pull_request:
+    branches: [ main, develop ]
+
+permissions:
+  contents: read
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: ${{ github.ref != 'refs/heads/main' }}
+
+jobs:
+  unit-tests:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@8e5e7e5ab8b370d6c329ec480221332ada57f0ab # v4
+    - uses: actions/setup-python@82c7e631bb3cdc910f68e0081d67478d79c6982d # v5
+      with:
+        python-version: '3.12'
+    - uses: snok/install-poetry@v1
+      with:
+        version: latest
+        virtualenvs-create: true
+        virtualenvs-in-project: true
+        virtualenvs-path: .venv
+        installer-parallel: true
+    - uses: actions/cache@v4
+      with:
+        path: .venv
+        key: ${{ runner.os }}-poetry-${{ hashFiles('**/poetry.lock') }}
+        restore-keys: |
+          ${{ runner.os }}-poetry-
+    - run: poetry install --with dev
+    - run: poetry run pip install pytest-xdist pytest-timeout
+    - run: poetry run pytest tests/unit -n 32 --timeout=90 --timeout-method=thread

tests/conftest.py

@@ -12,6 +12,7 @@
 
 import pytest
 import pytest_asyncio
+from _pytest.fixtures import FixtureRequest
 
 from mcp_code_indexer.database.database import DatabaseManager
 from mcp_code_indexer.database.models import FileDescription, Project
@@ -41,6 +42,26 @@ async def temp_db() -> AsyncGenerator[Path, None]:
         db_path.unlink()
 
 
+@pytest_asyncio.fixture
+async def temp_db_with_test_table(temp_db: Path) -> AsyncGenerator[Path, None]:
+    """Create a temporary database with a test table for retry testing."""
+    import aiosqlite
+
+    # Create database with tables
+    async with aiosqlite.connect(temp_db) as db:
+        await db.execute(
+            """
+            CREATE TABLE test_table (
+                id INTEGER PRIMARY KEY,
+                data TEXT NOT NULL
+            )
+        """
+        )
+        await db.commit()
+
+    yield temp_db
+
+
 @pytest_asyncio.fixture
 async def db_manager(temp_db: Path) -> AsyncGenerator[DatabaseManager, None]:
     """Create and initialize a database manager for testing."""
@@ -53,6 +74,24 @@ async def db_manager(temp_db: Path) -> AsyncGenerator[DatabaseManager, None]:
     await manager.close_pool()
 
 
+@pytest_asyncio.fixture
+async def temp_db_manager_pool(
+    temp_db: Path, request: FixtureRequest
+) -> AsyncGenerator[DatabaseManager, None]:
+    """Create and initialize a database manager with specified pool size for testing.
+
+    Use with @pytest.mark.parametrize("temp_db_manager_pool", [pool_size], indirect=True)
+    """
+    pool_size = request.param
+    manager = DatabaseManager(temp_db, pool_size=pool_size)
+    await manager.initialize()
+
+    yield manager
+
+    # Cleanup
+    await manager.close_pool()
+
+
 @pytest_asyncio.fixture
 async def sample_project(db_manager: DatabaseManager) -> Project:
     """Create a sample project for testing."""
@@ -293,10 +332,8 @@ async def mcp_server(tmp_path: Path):
 
     # Import here to avoid circular imports
     from mcp_code_indexer.server.mcp_server import MCPCodeIndexServer
-    
-    server = MCPCodeIndexServer(
-        token_limit=1000, db_path=db_path, cache_dir=cache_dir
-    )
+
+    server = MCPCodeIndexServer(token_limit=1000, db_path=db_path, cache_dir=cache_dir)
     await server.initialize()
 
     yield server