feat: enhance test infrastructure and utility functions

Author: VuQuoc18

backend/app/api/routes/utils.py

@@ -27,5 +27,28 @@ def test_email(email_to: EmailStr) -> Message:
 
 
 @router.get("/health-check/")
-async def health_check() -> bool:
-    return True
+async def health_check() -> dict[str, bool]:
+    """
+    Health check endpoint including database and Redis status.
+    """
+    from app.api.deps import get_db
+    from sqlalchemy import text
+    from app.core.redis import redis_client
+
+    # Check database connection
+    db_status = False
+    try:
+        db = next(get_db())
+        db.execute(text("SELECT 1"))
+        db_status = True
+    except Exception:
+        pass
+
+    # Check Redis connection
+    redis_status = redis_client.ping()
+
+    return {
+        "database": db_status,
+        "redis": redis_status,
+        "overall": db_status and redis_status,
+    }

backend/scripts/prestart.sh

@@ -4,10 +4,10 @@ set -e
 set -x
 
 # Let the DB start
-python app/backend_pre_start.py
+uv run python app/backend_pre_start.py
 
 # Run migrations
-alembic upgrade head
+uv run alembic upgrade head
 
 # Create initial data in DB
-python app/initial_data.py
+uv run python app/initial_data.py

backend/scripts/tests-start.sh

@@ -1,7 +1,65 @@
-#! /usr/bin/env bash
+#!/bin/bash
+
+# Development test setup script for backend project
+# This script starts necessary services for testing
+
 set -e
-set -x
 
-python app/tests_pre_start.py
+echo "🚀 Starting development environment for testing..."
+
+# Check if virtual environment is active
+if [[ "$VIRTUAL_ENV" == "" ]]; then
+    echo "❌ ERROR: Virtual environment not activated!"
+    echo "Please run: source .venv/bin/activate"
+    exit 1
+fi
+
+# Check if uv is installed
+if ! command -v uv &> /dev/null; then
+    echo "❌ ERROR: uv is not installed!"
+    echo "Please install uv: pip install uv"
+    exit 1
+fi
+
+echo "✅ Environment checks passed"
+
+# Sync dependencies if needed
+echo "📦 Syncing dependencies..."
+uv sync
+
+# Validate critical versions
+echo "🔍 Validating critical dependency versions..."
+
+# Check httpx version
+HTTPX_VERSION=$(uv run python -c "import httpx; print(httpx.__version__)" 2>/dev/null || echo "not installed")
+if [[ "$HTTPX_VERSION" == "0.24.1" ]]; then
+    echo "✅ httpx version: $HTTPX_VERSION (correct)"
+else
+    echo "❌ httpx version: $HTTPX_VERSION (expected: 0.24.1)"
+    echo "Please run: uv sync"
+    exit 1
+fi
+
+# Check fastapi version
+FASTAPI_VERSION=$(uv run python -c "import fastapi; print(fastapi.__version__)" 2>/dev/null || echo "not installed")
+if [[ "$FASTAPI_VERSION" == "0.109.2" ]]; then
+    echo "✅ fastapi version: $FASTAPI_VERSION (correct)"
+else
+    echo "❌ fastapi version: $FASTAPI_VERSION (expected: 0.109.2)"
+    echo "Please run: uv sync"
+    exit 1
+fi
+
+# Run pre-start script if it exists
+if [[ -f "app/tests_pre_start.py" ]]; then
+    echo "🔧 Running test pre-start script..."
+    python app/tests_pre_start.py
+fi
 
-bash scripts/test.sh "$@"
+echo "✅ All validations passed!"
+echo "🎯 Development environment ready for testing!"
+echo ""
+echo "Usage:"
+echo "  ./scripts/run-tests.sh                    # Run all tests"
+echo "  ./scripts/run-tests.sh test_file.py       # Run specific test"
+echo "  ./scripts/run-tests.sh -v                 # Verbose output"

backend/tests/conftest.py

@@ -7,16 +7,22 @@
 from app.core.config import settings
 from app.core.db import engine, init_db
 from app.main import app
-from app.models import Item, User
+from app.models import Item, User, Image, ImageVariant, ImageProcessingJob
 from tests.utils.user import authentication_token_from_email
 from tests.utils.utils import get_superuser_token_headers
 
 
-@pytest.fixture(scope="session", autouse=True)
+@pytest.fixture(scope="session")
 def db() -> Generator[Session, None, None]:
     with Session(engine) as session:
         init_db(session)
         yield session
+        statement = delete(ImageProcessingJob)
+        session.execute(statement)
+        statement = delete(ImageVariant)
+        session.execute(statement)
+        statement = delete(Image)
+        session.execute(statement)
         statement = delete(Item)
         session.execute(statement)
         statement = delete(User)
@@ -25,7 +31,7 @@ def db() -> Generator[Session, None, None]:
 
 
 @pytest.fixture(scope="module")
-def client() -> Generator[TestClient, None, None]:
+def client(db: Session) -> Generator[TestClient, None, None]:
     with TestClient(app) as c:
         yield c