feat: add working tests

This commit is contained in:
sam 2024-03-15 03:46:25 +01:00
parent cd7a5431e9
commit 3d7217ec69
Signed by: sam
GPG key ID: B4EF20DDE721CAA1
10 changed files with 80 additions and 21 deletions

View file

@ -1,8 +1,61 @@
import pytest
import pytest_asyncio
from sqlalchemy import text, delete
from foxnouns.db import Base
from foxnouns.settings import DATABASE
# Override the database name to the testing database
DATABASE["NAME"] = f"{DATABASE['NAME']}_test"
def pytest_collection_modifyitems(items):
"""Ensure that all async tests use the same event loop."""
pytest_asyncio_tests = (
item for item in items if pytest_asyncio.is_async_test(item)
)
session_scope_marker = pytest.mark.asyncio(scope="session")
for async_test in pytest_asyncio_tests:
async_test.add_marker(session_scope_marker, append=False)
@pytest.fixture(scope="session", autouse=True)
def setup():
"""Migrate the testing database to the latest migration, and once the tests complete, clear the database again."""
from foxnouns.db.sync import engine
from alembic import command, config
cfg = config.Config("alembic.ini")
cfg.attributes["connection"] = engine.connect()
command.upgrade(cfg, "head")
yield
with engine.begin() as session:
Base.metadata.drop_all(session)
session.execute(text("DROP TABLE alembic_version"))
session.commit()
@pytest.fixture(scope="function", autouse=True)
def clean_tables_after_tests():
"""Clean tables after every test."""
yield
from foxnouns.db.sync import engine
with engine.begin() as session:
for table in reversed(Base.metadata.sorted_tables):
session.execute(delete(table))
session.commit()
@pytest_asyncio.fixture(scope="session", autouse=True)
async def setup():
print("hello from setup!")
yield
print("bye from setup!")
async def app():
from foxnouns.app import app
return app

View file

@ -1,8 +0,0 @@
import pytest
from foxnouns import hello
@pytest.mark.asyncio
async def test_hello():
assert (await hello()) == "Hello world!"

13
tests/test_users.py Normal file
View file

@ -0,0 +1,13 @@
import pytest
from quart import Quart
@pytest.mark.asyncio
class TestUsers:
async def test_get_me_returns_403_if_unauthenticated(self, app: Quart):
resp = await app.test_client().get("/api/v2/users/@me")
assert resp.status_code == 403
async def test_get_users_returns_404_if_user_not_found(self, app: Quart):
resp = await app.test_client().get("/api/v2/users/unknown_user")
assert resp.status_code == 404