diff options
Diffstat (limited to 'shared')
| -rw-r--r-- | shared/alembic.ini | 2 | ||||
| -rw-r--r-- | shared/tests/test_db.py | 1 |
2 files changed, 2 insertions, 1 deletions
diff --git a/shared/alembic.ini b/shared/alembic.ini index 2c4fd1f..7206a9b 100644 --- a/shared/alembic.ini +++ b/shared/alembic.ini @@ -1,6 +1,6 @@ [alembic] script_location = alembic -sqlalchemy.url = postgresql+asyncpg://postgres:postgres@localhost:5432/trading +sqlalchemy.url = postgresql+asyncpg://trading:trading@localhost:5432/trading [loggers] keys = root,sqlalchemy,alembic diff --git a/shared/tests/test_db.py b/shared/tests/test_db.py index f4cabfd..d33dfe1 100644 --- a/shared/tests/test_db.py +++ b/shared/tests/test_db.py @@ -306,6 +306,7 @@ class TestTransactionContextManager: db = Database("postgresql+asyncpg://host/db") mock_session = AsyncMock() + mock_session.add = MagicMock() mock_session.__aenter__ = AsyncMock(return_value=mock_session) mock_session.__aexit__ = AsyncMock(return_value=False) |