External/securelens-backend
1
0
Fork 0
mirror of https://github.com/Rarebuffalo/securelens-backend.git synced 2026-06-19 07:00:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

switch pytest fixtures to use pytest_asyncio decorators

Browse Source
This commit is contained in:
rarebuffalo
2026-06-12 19:10:58 +05:30
parent 6778a2fc65
commit 8e528b149a
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/conftest.py
View File

@@ -1,4 +1,5 @@
import pytest import pytest
import pytest_asyncio
from httpx import ASGITransport, AsyncClient from httpx import ASGITransport, AsyncClient
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
@@ -28,7 +29,7 @@ async def override_get_db():
app.dependency_overrides[get_db] = override_get_db app.dependency_overrides[get_db] = override_get_db
@pytest.fixture(autouse=True) @pytest_asyncio.fixture(autouse=True)
async def setup_db(): async def setup_db():
async with test_engine.begin() as conn: async with test_engine.begin() as conn:
await conn.run_sync(Base.metadata.create_all) await conn.run_sync(Base.metadata.create_all)
@@ -43,14 +44,14 @@ def client():
return TestClient(app) return TestClient(app)
@pytest.fixture @pytest_asyncio.fixture
async def async_client(): async def async_client():
transport = ASGITransport(app=app) transport = ASGITransport(app=app)
async with AsyncClient(transport=transport, base_url="http://test") as ac: async with AsyncClient(transport=transport, base_url="http://test") as ac:
yield ac yield ac
@pytest.fixture @pytest_asyncio.fixture
async def test_user(): async def test_user():
async with TestSessionLocal() as session: async with TestSessionLocal() as session:
user = User( user = User(
@@ -64,7 +65,7 @@ async def test_user():
return user return user
@pytest.fixture @pytest_asyncio.fixture
async def auth_headers(test_user): async def auth_headers(test_user):
token = create_access_token(test_user.id) token = create_access_token(test_user.id)
return {"Authorization": f"Bearer {token}"} return {"Authorization": f"Bearer {token}"}