opentelemetry_testbase_span_capture_and_verification_example.py

1from opentelemetry import trace
2from opentelemetry.test.test_base import TestBase
3
4class TestMyIntegration(TestBase):
5    def test_span_creation(self):
6        # The TestBase setup automatically configures a TracerProvider 
7        # with an InMemorySpanExporter.
8        tracer = trace.get_tracer(__name__)
9        
10        with tracer.start_as_current_span("test-span"):
11            print("Doing some work...")
12
13        # Get finished spans from the in-memory exporter
14        span_list = self.get_finished_spans()
15
16        # Assertions
17        self.assertEqual(len(span_list), 1)
18        span = span_list[0]
19        self.assertEqual(span.name, "test-span")
20        self.assertTrue(span.status.is_ok)
21
22if __name__ == "__main__":
23    import unittest
24    unittest.main()