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