def test_trace_init_custom_id_generator(self, mock_iter_entry_points):
mock_iter_entry_points.configure_mock(return_value=[
IterEntryPoint("custom_id_generator", CustomIdGenerator)
])
id_generator_name = _get_id_generator()
id_generator = _import_id_generator(id_generator_name)
_init_tracing({}, id_generator)
provider = self.set_provider_mock.call_args[0][0]
self.assertIsInstance(provider.id_generator, CustomIdGenerator)
def test_trace_init_default(self):
_init_tracing({"zipkin": Exporter}, RandomIdGenerator)
self.assertEqual(self.set_provider_mock.call_count, 1)
provider = self.set_provider_mock.call_args[0][0]
self.assertIsInstance(provider, Provider)
self.assertIsInstance(provider.id_generator, RandomIdGenerator)
self.assertIsInstance(provider.processor, Processor)
self.assertIsInstance(provider.processor.exporter, Exporter)
self.assertEqual(provider.processor.exporter.service_name,
"my-test-service")
def test_trace_init_default(self):
environ[OTEL_PYTHON_SERVICE_NAME] = "my-test-service"
_init_tracing({"zipkin": Exporter}, RandomIdsGenerator)
self.assertEqual(self.set_provider_mock.call_count, 1)
provider = self.set_provider_mock.call_args[0][0]
self.assertIsInstance(provider, Provider)
self.assertIsInstance(provider.ids_generator, RandomIdsGenerator)
self.assertIsInstance(provider.processor, Processor)
self.assertIsInstance(provider.processor.exporter, Exporter)
self.assertEqual(
provider.processor.exporter.service_name, "my-test-service"
)
def test_trace_init_otlp(self):
_init_tracing({"otlp": OTLPExporter}, RandomIdGenerator)
self.assertEqual(self.set_provider_mock.call_count, 1)
provider = self.set_provider_mock.call_args[0][0]
self.assertIsInstance(provider, Provider)
self.assertIsInstance(provider.id_generator, RandomIdGenerator)
self.assertIsInstance(provider.processor, Processor)
self.assertIsInstance(provider.processor.exporter, OTLPExporter)
self.assertIsInstance(provider.resource, Resource)
self.assertEqual(
provider.resource.attributes.get("service.name"),
"my-otlp-test-service",
)
def test_trace_init_otlp(self):
environ["OTEL_SERVICE_NAME"] = "my-otlp-test-service"
Configuration._reset()
_init_tracing({"otlp": OTLPExporter}, RandomIdsGenerator)
self.assertEqual(self.set_provider_mock.call_count, 1)
provider = self.set_provider_mock.call_args[0][0]
self.assertIsInstance(provider, Provider)
self.assertIsInstance(provider.ids_generator, RandomIdsGenerator)
self.assertIsInstance(provider.processor, Processor)
self.assertIsInstance(provider.processor.exporter, OTLPExporter)
self.assertIsInstance(provider.resource, Resource)
self.assertEqual(
provider.resource.attributes.get("service.name"),
"my-otlp-test-service",
)
del environ["OTEL_SERVICE_NAME"]
