diff --git a/test/conftest.py b/test/conftest.py index 6411b96..6821b9d 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -14,7 +14,7 @@ from src.iam.models import Group, Permission, OrgPermissions from src.auth.service import get_current_user, get_dev_user from src.auth.dependencies import empty_su_list, get_super_admin_list, testing_su_list from src.main import app # inited FastAPI app -from src.database import engine, get_db +from src.database import engine, get_db_session from src.models import CustomBase SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) @@ -35,51 +35,51 @@ def db_session(): @pytest.fixture async def default_client(db_session) -> AsyncGenerator[AsyncClient, None]: - def get_db_override(): - return db_session + def get_db_override(): + return db_session - app.dependency_overrides[get_db] = get_db_override - app.dependency_overrides[get_current_user] = get_dev_user - app.dependency_overrides[get_super_admin_list] = testing_su_list - transport = ASGITransport(app=app) - async with AsyncClient( - transport=transport, base_url="http://localhost:8000/api/v1" - ) as ac: - yield ac + app.dependency_overrides[get_db_session] = get_db_override + app.dependency_overrides[get_current_user] = get_dev_user + app.dependency_overrides[get_super_admin_list] = testing_su_list + transport = ASGITransport(app=app) + async with AsyncClient( + transport=transport, base_url="http://localhost:8000/api/v1" + ) as ac: + yield ac - app.dependency_overrides.clear() + app.dependency_overrides.clear() @pytest.fixture async def no_user_client(db_session) -> AsyncGenerator[AsyncClient, None]: - def get_db_override(): - return db_session + def get_db_override(): + return db_session - app.dependency_overrides[get_db] = get_db_override - transport = ASGITransport(app=app) - async with AsyncClient( - transport=transport, base_url="http://localhost:8000/api/v1" - ) as ac: - yield ac + app.dependency_overrides[get_db_session] = get_db_override + transport = ASGITransport(app=app) + async with AsyncClient( + transport=transport, base_url="http://localhost:8000/api/v1" + ) as ac: + yield ac - app.dependency_overrides.clear() + app.dependency_overrides.clear() @pytest.fixture async def no_su_client(db_session) -> AsyncGenerator[AsyncClient, None]: - def get_db_override(): - return db_session + def get_db_override(): + return db_session - app.dependency_overrides[get_db] = get_db_override - app.dependency_overrides[get_current_user] = get_dev_user - app.dependency_overrides[get_super_admin_list] = empty_su_list - transport = ASGITransport(app=app) - async with AsyncClient( - transport=transport, base_url="http://localhost:8000/api/v1" - ) as ac: - yield ac + app.dependency_overrides[get_db_session] = get_db_override + app.dependency_overrides[get_current_user] = get_dev_user + app.dependency_overrides[get_super_admin_list] = empty_su_list + transport = ASGITransport(app=app) + async with AsyncClient( + transport=transport, base_url="http://localhost:8000/api/v1" + ) as ac: + yield ac - app.dependency_overrides.clear() + app.dependency_overrides.clear() def _seed(db): @@ -256,27 +256,29 @@ def generate_body_and_status(params: dict[str, str]) -> list[tuple[dict, int]]: def get_testable_routes(): - routes = [] + routes = [] - for route in app.routes: - if not isinstance(route, APIRoute): - continue + for route in app.routes: + if not isinstance(route, APIRoute): + continue + if not route.methods: + continue - for method in route.methods: - if method in {"HEAD", "OPTIONS"}: - continue + for method in route.methods: + if method in {"HEAD", "OPTIONS"}: + continue - routes.append( - ( - method, - route.path, - route.status_code, - route.response_model, - route.summary, - ) - ) + routes.append( + ( + method, + route.path, + route.status_code, + route.response_model, + route.summary, + ) + ) - return routes + return routes # with open("endpoints.txt", "w") as f: