Step 12 - completed

tests/test_middleware.py

@@ -18,13 +18,30 @@ from fastapi.testclient import TestClient
 from jose import jwt
 
 from app.config.settings import settings
+from app.db import get_session
 from app.main import app
 from app.schemas import ChatResponse
+from tests.conftest import TEST_USER_IDS
 
 # ---------------------------------------------------------------------------
 # Helpers
 # ---------------------------------------------------------------------------
 
+# ---------------------------------------------------------------------------
+# Autouse: redirect all DB access to the in-memory SQLite test engine.
+# ---------------------------------------------------------------------------
+
+@pytest.fixture(autouse=True)
+def _override_db(db_session):
+    """Route all get_session calls to the test SQLite session."""
+    async def _gen():
+        yield db_session
+
+    app.dependency_overrides[get_session] = _gen
+    yield
+    app.dependency_overrides.pop(get_session, None)
+
+
 _CHAT_BODY = {
     "message": "hello",
     "context": {
@@ -74,14 +91,15 @@ class TestAuthMiddleware:
     """Tests exercised via GET /api/v1/auth/me."""
 
     def test_valid_token_returns_profile(self) -> None:
-        uid = str(uuid.uuid4())
-        token = _make_jwt(user_id=uid, email="alice@example.com", tier="pro")
+        # Use the seeded pro user so the subscription lookup returns 'pro'.
+        uid = TEST_USER_IDS["pro"]
+        token = _make_jwt(user_id=uid, email="pro@test.com", tier="pro")
         with TestClient(app) as client:
             resp = client.get("/api/v1/auth/me", headers=_auth_header(token))
         assert resp.status_code == 200
         data = resp.json()
         assert data["id"] == uid
-        assert data["email"] == "alice@example.com"
+        assert data["email"] == "pro@test.com"
         assert data["tier"] == "pro"
 
     def test_missing_token_returns_401(self) -> None: