def test_contextmanager(self):
mock_dbapi2 = mock.MagicMock()
pool = ConnectionPool(mock_dbapi2, {}, maxsize=10)
with pool.connection() as connection:
mock_dbapi2.assert_has_calls([mock.call.connect()])
self.assertEqual(id(connection),
id(mock_dbapi2.connect.return_value))
mock_dbapi2.assert_has_calls([
mock.call.connect(),
mock.call.connect().rollback(),
])
def test_contextmanager_error(self): # pylint: disable=no-self-use
class MockDBAPI2Error(Exception):
pass
mock_dbapi2 = mock.MagicMock()
pool = ConnectionPool(mock_dbapi2, {}, maxsize=10)
try:
with pool.connection():
raise MockDBAPI2Error
except MockDBAPI2Error:
pass
mock_dbapi2.assert_has_calls([
mock.call.connect(),
mock.call.connect().rollback(),
])
def test_contextmanager(self):
mock_dbapi2 = mock.MagicMock()
pool = ConnectionPool(mock_dbapi2, {}, maxsize=10)
with pool.connection() as connection:
mock_dbapi2.assert_has_calls(
[
mock.call.connect()
]
)
self.assertEqual(
id(connection),
id(mock_dbapi2.connect.return_value)
)
mock_dbapi2.assert_has_calls(
[
mock.call.connect(),
mock.call.connect().rollback(),
]
)