Set up an extremely lightweight set of tests for the user infra implementation

This commit is contained in:
Campbell Alden 2026-09-03 22:58:29 +09:00
parent 0d270c9cfb
commit 32e3fe04a2
5 changed files with 47 additions and 0 deletions

8
src/infra/db_test.py Normal file
View file

@ -0,0 +1,8 @@
from src.infra import Base
from sqlalchemy import Engine, create_engine
def mock_db() -> Engine:
engine = create_engine('sqlite:///:memory:', echo=True)
Base.metadata.create_all(engine)
return engine

31
src/infra/users_test.py Normal file
View file

@ -0,0 +1,31 @@
from src.utils.secret import SecretBox
from src.services.users.data import UserDTO, User
from src.services.users.repo import UserRepo
from src.infra.users import UserRepoImpl
from src.infra.db_test import mock_db
import pytest
DB = mock_db()
@pytest.fixture
def db():
return DB
@pytest.fixture
def repo(db):
return UserRepoImpl(db)
@pytest.fixture
def users(repo: UserRepo):
return [
repo.create(UserDTO('example@example.com', SecretBox('hunter1')), None),
repo.create(UserDTO('example2@example.com', SecretBox('hunter2')), None),
]
def test_getting_a_user_by_email(users: list[User], repo: UserRepo):
assert repo.get_user_by_email('example@example.com') == users[0]
assert repo.get_user_by_email('fred@example.com') is None