Skip to content
This repository was archived by the owner on Oct 23, 2023. It is now read-only.

Commit 9275ecb

Browse files
committed
test db load close db connection exception
1 parent 0f281d6 commit 9275ecb

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

tests/test_db_load.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -327,6 +327,16 @@ async def test_close(self, db_mock, mock_log):
327327
mock_log.info.mock_calls = ['Mark the database connection to be closed',
328328
'The database connection has been closed']
329329

330+
@asynctest.mock.patch('beacon_api.utils.db_load.LOG')
331+
@asynctest.mock.patch('beacon_api.utils.db_load.asyncpg.connect')
332+
async def test_close_error(self, db_mock, mock_log):
333+
"""Test database URL close error."""
334+
db_mock.return_value = ConnectionException()
335+
await self._db.connection()
336+
await self._db.close()
337+
log = "AN ERROR OCCURRED WHILE ATTEMPTING TO CLOSE DATABASE CONNECTION -> 'ConnectionException' object has no attribute 'close'"
338+
mock_log.error.assert_called_with(log)
339+
330340
@asynctest.mock.patch('beacon_api.utils.db_load.LOG')
331341
@asynctest.mock.patch('beacon_api.utils.db_load.asyncpg.connect')
332342
async def test_unpack(self, db_mock, mock_log):

0 commit comments

Comments
 (0)