diff --git a/src/infra/db_test.py b/src/infra/conftest.py similarity index 77% rename from src/infra/db_test.py rename to src/infra/conftest.py index e362239..0b8f2a9 100644 --- a/src/infra/db_test.py +++ b/src/infra/conftest.py @@ -1,3 +1,4 @@ +import pytest from src.infra import Base from sqlalchemy import Engine, create_engine @@ -6,3 +7,8 @@ def mock_db() -> Engine: engine = create_engine('sqlite:///:memory:', echo=True) Base.metadata.create_all(engine) return engine + + +@pytest.fixture +def db(): + return mock_db() diff --git a/src/infra/users_test.py b/src/infra/users_test.py index a7f8da7..80ce6bc 100644 --- a/src/infra/users_test.py +++ b/src/infra/users_test.py @@ -2,15 +2,10 @@ 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 -@pytest.fixture -def db(): - return mock_db() - - @pytest.fixture def repo(db): return UserRepoImpl(db)