datadog: fix test_memory_trace_client_test2.py

automon/integrations/openTelemetryWrapper/client.py

@@ -82,6 +82,18 @@ async def test(self):
 
         return True
 
+    async def test2(self):
+        with self.config.tracer.start_as_current_span(name='rootSpan') as trace_root:
+            self.config.opentelemetry.trace.get_current_span('AAAAAAAA')
+
+            with self.config.tracer.start_as_current_span(name='childSpan') as trace_child:
+                self.config.opentelemetry.trace.get_current_span('AAAAAAAA')
+                self.config.opentelemetry.trace.get_current_span('BBBBBBBB')
+
+            self.config.opentelemetry.trace.get_current_span('BBBBBBBB')
+
+        return True
+
     async def to_dict(self):
         return [
             json.loads(span.to_json())

automon/integrations/openTelemetryWrapper/config.py

@@ -13,6 +13,7 @@
 
 class OpenTelemetryConfig(object):
     def __init__(self):
+        self.opentelemetry = opentelemetry
         self.provider = TracerProvider()
         self.memory_processor = InMemorySpanExporter()
         self.processor = SimpleSpanProcessor(self.memory_processor)

automon/integrations/openTelemetryWrapper/test/test_memory_trace_client_test2.py

@@ -0,0 +1,21 @@
+import unittest
+import asyncio
+
+from automon.integrations.openTelemetryWrapper import OpenTelemetryClient
+
+
+class MyTestCase(unittest.TestCase):
+    client = OpenTelemetryClient()
+
+    def test_something(self):
+        self.assertTrue(asyncio.run(
+            self.client.test2()))
+
+        spans = asyncio.run(
+            self.client.get_finished_spans())
+
+        pass
+
+
+if __name__ == '__main__':
+    unittest.main()