diff options
Diffstat (limited to 'shared/tests/test_db.py')
| -rw-r--r-- | shared/tests/test_db.py | 1 |
1 files changed, 1 insertions, 0 deletions
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) |