def test_acquire_fails_if_pool_closed(self):
pool = DBConnectionPool(Mock())
pool.close()
with self.assertRaises(RuntimeError):
with pool.acquire() as p:
pass
def test_conn_close_called_on_close(self):
mock_conn = Mock()
create_func = Mock(return_value=mock_conn)
pool = DBConnectionPool(create_func, min_size=2, max_size=2)
pool.close()
mock_conn.assert_has_calls([call.close(), call.close()])