def test_unwrap_create_pool(self):
async def check_connection(pool):
async with pool.acquire() as connection:
self.assertEqual(aiopg_mock.create_pool_call_count, 2)
self.assertIsInstance(connection, AiopgConnectionMock)
aiopg_mock = AiopgMock()
with mock.patch("aiopg.create_pool", aiopg_mock.create_pool):
wrappers.wrap_create_pool(self.tracer, "-")
pool = async_call(aiopg.create_pool())
self.assertEqual(aiopg_mock.create_pool_call_count, 1)
wrappers.unwrap_create_pool()
pool = async_call(aiopg.create_pool())
async_call(check_connection(pool))
def _uninstrument(self, **kwargs):
""""Disable aiopg instrumentation"""
wrappers.unwrap_connect()
wrappers.unwrap_create_pool()
