working auth + users systems

backend/tests/conftest.py

@@ -0,0 +1,58 @@
+# conftest.py
+from typing import Generator, Callable, Any
+import pytest
+from testcontainers.postgres import PostgresContainer
+from fastapi.testclient import TestClient
+from core.config import settings
+from faker import Faker
+from sqlalchemy.orm import Session
+
+from core.database import get_db, get_sessionmaker
+
+
+fake = Faker()
+
+@pytest.fixture(scope="session")
+def postgres_container() -> Generator[PostgresContainer, None, None]:
+    """Fixture to create a PostgreSQL container for testing."""
+    print("Starting Postgres container...")
+    with PostgresContainer("postgres:latest") as postgres:
+        settings.DB_URL = postgres.get_connection_url()
+        print(f"Postgres container started at {settings.DB_URL}")
+        yield postgres
+    print("Postgres container stopped.")
+    
+@pytest.fixture(scope="function")
+def db(postgres_container) -> Generator[Session, None, None]:
+    """Function-scoped database session with rollback"""
+    SessionLocal = get_sessionmaker()
+    session = SessionLocal()
+    session.begin_nested()  # Enable nested transaction
+    try:
+        yield session
+    finally:
+        session.rollback()
+        session.close()
+
+@pytest.fixture(scope="function")
+def client(db: Session) -> Generator[TestClient, None, None]:
+    """Function-scoped test client with dependency override"""
+    from main import app
+    
+    # Override the database dependency
+    def override_get_db():
+        try:
+            yield db
+        finally:
+            pass  # Don't close session here
+    
+    app.dependency_overrides[get_db] = override_get_db
+    
+    with TestClient(app) as test_client:
+        yield test_client
+    
+    app.dependency_overrides.clear()
+
+def override_dependency(dependency: Callable[..., Any], mocked_response: Any) -> None:
+    from main import app
+    app.dependency_overrides[dependency] = lambda: mocked_response