opensource/opentelemetry-python-contrib
1
0
Fork 0
mirror of https://github.com/open-telemetry/opentelemetry-python-contrib.git synced 2025-07-28 20:52:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

add uninstrument test for Celery (#1490)

Browse Source
This commit is contained in:
Nimrod Shlagman
2022-12-18 16:46:26 +02:00
committed by GitHub
parent 41592015a3
commit 23f63a5328
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
instrumentation/opentelemetry-instrumentation-celery/tests/test_tasks.py
View File

@ -33,6 +33,7 @@ class TestCeleryInstrumentation(TestBase):
def tearDown(self): def tearDown(self):
super().tearDown() super().tearDown()
CeleryInstrumentor().uninstrument()
self._worker.stop() self._worker.stop()
self._thread.join() self._thread.join()
@ -40,7 +41,11 @@ class TestCeleryInstrumentation(TestBase):
CeleryInstrumentor().instrument() CeleryInstrumentor().instrument()
result = task_add.delay(1, 2) result = task_add.delay(1, 2)
timeout = time.time() + 60 * 1 # 1 minutes from now
while not result.ready(): while not result.ready():
if time.time() > timeout:
break
time.sleep(0.05) time.sleep(0.05)
spans = self.sorted_spans(self.memory_exporter.get_finished_spans()) spans = self.sorted_spans(self.memory_exporter.get_finished_spans())
@ -77,3 +82,18 @@ class TestCeleryInstrumentation(TestBase):
self.assertNotEqual(consumer.parent, producer.context) self.assertNotEqual(consumer.parent, producer.context)
self.assertEqual(consumer.parent.span_id, producer.context.span_id) self.assertEqual(consumer.parent.span_id, producer.context.span_id)
self.assertEqual(consumer.context.trace_id, producer.context.trace_id) self.assertEqual(consumer.context.trace_id, producer.context.trace_id)
def test_uninstrument(self):
CeleryInstrumentor().instrument()
CeleryInstrumentor().uninstrument()
result = task_add.delay(1, 2)
timeout = time.time() + 60 * 1 # 1 minutes from now
while not result.ready():
if time.time() > timeout:
break
time.sleep(0.05)
spans = self.memory_exporter.get_finished_spans()
self.assertEqual(len(spans), 0)