def test_custom_event_loop(self):
loop = CustomEventLoop()
self.assertFalse(isinstance(loop.create_task, wrapt.ObjectProxy))
asyncio.set_event_loop(loop)
patch()
self.assertTrue(isinstance(loop.create_task, wrapt.ObjectProxy))
unpatch()
self.assertFalse(isinstance(loop.create_task, wrapt.ObjectProxy))
def test_after_set_event_loop(self):
loop = asyncio.new_event_loop()
self.assertNotIsInstance(loop.create_task, wrapt.ObjectProxy)
asyncio.set_event_loop(loop)
patch()
self.assertIsInstance(loop.create_task, wrapt.ObjectProxy)
unpatch()
self.assertNotIsInstance(loop.create_task, wrapt.ObjectProxy)
def test_event_loop_double_patch(self):
# ensures that double patching will not double instrument
# the event loop
patch()
self.test_tasks_chaining()
def setUp(self):
# patch asyncio event loop
super(TestAsyncioPropagation, self).setUp()
patch()
def test_new_loop(self):
patch()
loop = asyncio.new_event_loop()
self.assertIsInstance(loop.create_task, wrapt.ObjectProxy)
def test_event_loop_double_patch(self):
# ensures that double patching will not double instrument
# the event loop
patch()
self.test_tasks_chaining()
def setUp(self):
# patch asyncio event loop
super(TestAsyncioPropagation, self).setUp()
patch()
async def test_event_loop_double_patch(tracer):
# ensures that double patching will not double instrument
# the event loop
patch()
patch()
await test_tasks_chaining(tracer)
def test_event_loop_unpatch(tracer):
patch()
# ensures that the event loop can be unpatched
unpatch()
assert isinstance(tracer.context_provider, DefaultContextProvider)
assert asyncio.BaseEventLoop.create_task == _orig_create_task
