18
18
from unittest .mock import patch
19
19
20
20
from opentelemetry .distro import (
21
- _get_ids_generator ,
22
- _import_ids_generator ,
21
+ _get_id_generator ,
22
+ _import_id_generator ,
23
23
_init_tracing ,
24
24
)
25
25
from opentelemetry .environment_variables import (
26
- OTEL_PYTHON_IDS_GENERATOR ,
26
+ OTEL_PYTHON_ID_GENERATOR ,
27
27
OTEL_PYTHON_SERVICE_NAME ,
28
28
)
29
29
from opentelemetry .sdk .resources import Resource
30
- from opentelemetry .sdk .trace .ids_generator import (
31
- IdsGenerator ,
32
- RandomIdsGenerator ,
33
- )
30
+ from opentelemetry .sdk .trace .id_generator import IdGenerator , RandomIdGenerator
34
31
35
32
36
33
class Provider :
37
- def __init__ (self , resource = None , ids_generator = None ):
38
- self .ids_generator = ids_generator
34
+ def __init__ (self , resource = None , id_generator = None ):
35
+ self .id_generator = id_generator
39
36
self .processor = None
40
37
self .resource = resource
41
38
@@ -60,7 +57,7 @@ class OTLPExporter:
60
57
pass
61
58
62
59
63
- class CustomIdsGenerator ( IdsGenerator ):
60
+ class CustomIdGenerator ( IdGenerator ):
64
61
def generate_span_id (self ):
65
62
pass
66
63
@@ -103,12 +100,12 @@ def tearDown(self):
103
100
# pylint: disable=protected-access
104
101
def test_trace_init_default (self ):
105
102
environ [OTEL_PYTHON_SERVICE_NAME ] = "my-test-service"
106
- _init_tracing ({"zipkin" : Exporter }, RandomIdsGenerator )
103
+ _init_tracing ({"zipkin" : Exporter }, RandomIdGenerator )
107
104
108
105
self .assertEqual (self .set_provider_mock .call_count , 1 )
109
106
provider = self .set_provider_mock .call_args [0 ][0 ]
110
107
self .assertIsInstance (provider , Provider )
111
- self .assertIsInstance (provider .ids_generator , RandomIdsGenerator )
108
+ self .assertIsInstance (provider .id_generator , RandomIdGenerator )
112
109
self .assertIsInstance (provider .processor , Processor )
113
110
self .assertIsInstance (provider .processor .exporter , Exporter )
114
111
self .assertEqual (
@@ -117,12 +114,12 @@ def test_trace_init_default(self):
117
114
118
115
def test_trace_init_otlp (self ):
119
116
environ [OTEL_PYTHON_SERVICE_NAME ] = "my-otlp-test-service"
120
- _init_tracing ({"otlp" : OTLPExporter }, RandomIdsGenerator )
117
+ _init_tracing ({"otlp" : OTLPExporter }, RandomIdGenerator )
121
118
122
119
self .assertEqual (self .set_provider_mock .call_count , 1 )
123
120
provider = self .set_provider_mock .call_args [0 ][0 ]
124
121
self .assertIsInstance (provider , Provider )
125
- self .assertIsInstance (provider .ids_generator , RandomIdsGenerator )
122
+ self .assertIsInstance (provider .id_generator , RandomIdGenerator )
126
123
self .assertIsInstance (provider .processor , Processor )
127
124
self .assertIsInstance (provider .processor .exporter , OTLPExporter )
128
125
self .assertIsInstance (provider .resource , Resource )
@@ -132,17 +129,17 @@ def test_trace_init_otlp(self):
132
129
)
133
130
del environ [OTEL_PYTHON_SERVICE_NAME ]
134
131
135
- @patch .dict (environ , {OTEL_PYTHON_IDS_GENERATOR : "custom_ids_generator " })
136
- @patch ("opentelemetry.distro.IdsGenerator " , new = IdsGenerator )
132
+ @patch .dict (environ , {OTEL_PYTHON_ID_GENERATOR : "custom_id_generator " })
133
+ @patch ("opentelemetry.distro.IdGenerator " , new = IdGenerator )
137
134
@patch ("opentelemetry.distro.iter_entry_points" )
138
- def test_trace_init_custom_ids_generator (self , mock_iter_entry_points ):
135
+ def test_trace_init_custom_id_generator (self , mock_iter_entry_points ):
139
136
mock_iter_entry_points .configure_mock (
140
137
return_value = [
141
- IterEntryPoint ("custom_ids_generator " , CustomIdsGenerator )
138
+ IterEntryPoint ("custom_id_generator " , CustomIdGenerator )
142
139
]
143
140
)
144
- ids_generator_name = _get_ids_generator ()
145
- ids_generator = _import_ids_generator ( ids_generator_name )
146
- _init_tracing ({}, ids_generator )
141
+ id_generator_name = _get_id_generator ()
142
+ id_generator = _import_id_generator ( id_generator_name )
143
+ _init_tracing ({}, id_generator )
147
144
provider = self .set_provider_mock .call_args [0 ][0 ]
148
- self .assertIsInstance (provider .ids_generator , CustomIdsGenerator )
145
+ self .assertIsInstance (provider .id_generator , CustomIdGenerator )
0 commit comments