def test_wrap_create_pool(self): async def check_connection(pool): async with pool.acquire() as connection: self.assertEqual(aiopg_mock.create_pool_call_count, 1) self.assertIsInstance(connection.__wrapped__, 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()) async_call(check_connection(pool))
def _instrument(self, **kwargs): """Integrate with PostgreSQL aiopg library. aiopg: https://github.com/aio-libs/aiopg """ tracer_provider = kwargs.get("tracer_provider") wrappers.wrap_connect( __name__, self._DATABASE_SYSTEM, self._CONNECTION_ATTRIBUTES, version=__version__, tracer_provider=tracer_provider, ) wrappers.wrap_create_pool( __name__, self._DATABASE_SYSTEM, self._CONNECTION_ATTRIBUTES, version=__version__, tracer_provider=tracer_provider, )