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_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_event_loop_unpatch(self):
# ensures that the event loop can be unpatched
unpatch()
assert isinstance(self.tracer.context_provider, DefaultContextProvider)
assert asyncio.BaseEventLoop.create_task == _orig_create_task
def tearDown(self):
# unpatch asyncio event loop
super(TestAsyncioPropagation, self).tearDown()
unpatch()
def tearDown(self):
unpatch()
super(TestAsyncioPatch, self).tearDown()
def test_event_loop_unpatch(self):
# ensures that the event loop can be unpatched
unpatch()
ok_(isinstance(self.tracer._context_provider, DefaultContextProvider))
ok_(BaseEventLoop.create_task == _orig_create_task)
def tearDown(self):
# unpatch asyncio event loop
super(TestAsyncioPropagation, self).tearDown()
unpatch()
