22
22
23
23
24
24
class TestResources (unittest .TestCase ):
25
+ def setUp (self ) -> None :
26
+ os .environ [resources .OTEL_RESOURCE_ATTRIBUTES ] = ""
27
+
28
+ def tearDown (self ) -> None :
29
+ os .environ .pop (resources .OTEL_RESOURCE_ATTRIBUTES )
30
+
25
31
def test_create (self ):
26
32
attributes = {
27
33
"service" : "ui" ,
@@ -44,14 +50,22 @@ def test_create(self):
44
50
self .assertIsInstance (resource , resources .Resource )
45
51
self .assertEqual (resource .attributes , expected_attributes )
46
52
53
+ os .environ [resources .OTEL_RESOURCE_ATTRIBUTES ] = "key=value"
54
+ resource = resources .Resource .create (attributes )
55
+ self .assertIsInstance (resource , resources .Resource )
56
+ expected_with_envar = expected_attributes .copy ()
57
+ expected_with_envar ["key" ] = "value"
58
+ self .assertEqual (resource .attributes , expected_with_envar )
59
+ os .environ [resources .OTEL_RESOURCE_ATTRIBUTES ] = ""
60
+
47
61
resource = resources .Resource .create_empty ()
48
- self .assertIs (resource , resources ._EMPTY_RESOURCE )
62
+ self .assertEqual (resource , resources ._EMPTY_RESOURCE )
49
63
50
64
resource = resources .Resource .create (None )
51
- self .assertIs (resource , resources ._DEFAULT_RESOURCE )
65
+ self .assertEqual (resource , resources ._DEFAULT_RESOURCE )
52
66
53
67
resource = resources .Resource .create ({})
54
- self .assertIs (resource , resources ._DEFAULT_RESOURCE )
68
+ self .assertEqual (resource , resources ._DEFAULT_RESOURCE )
55
69
56
70
def test_resource_merge (self ):
57
71
left = resources .Resource ({"service" : "ui" })
@@ -184,36 +198,38 @@ def test_resource_detector_raise_error(self):
184
198
185
199
class TestOTELResourceDetector (unittest .TestCase ):
186
200
def setUp (self ) -> None :
187
- os .environ [" OTEL_RESOURCE_ATTRIBUTES" ] = ""
201
+ os .environ [resources . OTEL_RESOURCE_ATTRIBUTES ] = ""
188
202
189
203
def tearDown (self ) -> None :
190
- os .environ .pop (" OTEL_RESOURCE_ATTRIBUTES" )
204
+ os .environ .pop (resources . OTEL_RESOURCE_ATTRIBUTES )
191
205
192
206
def test_empty (self ):
193
207
detector = resources .OTELResourceDetector ()
194
- os .environ [" OTEL_RESOURCE_ATTRIBUTES" ] = ""
208
+ os .environ [resources . OTEL_RESOURCE_ATTRIBUTES ] = ""
195
209
self .assertEqual (detector .detect (), resources .Resource .create_empty ())
196
210
197
211
def test_one (self ):
198
212
detector = resources .OTELResourceDetector ()
199
- os .environ [" OTEL_RESOURCE_ATTRIBUTES" ] = "k=v"
213
+ os .environ [resources . OTEL_RESOURCE_ATTRIBUTES ] = "k=v"
200
214
self .assertEqual (detector .detect (), resources .Resource ({"k" : "v" }))
201
215
202
216
def test_one_with_whitespace (self ):
203
217
detector = resources .OTELResourceDetector ()
204
- os .environ [" OTEL_RESOURCE_ATTRIBUTES" ] = " k = v "
218
+ os .environ [resources . OTEL_RESOURCE_ATTRIBUTES ] = " k = v "
205
219
self .assertEqual (detector .detect (), resources .Resource ({"k" : "v" }))
206
220
207
221
def test_multiple (self ):
208
222
detector = resources .OTELResourceDetector ()
209
- os .environ [" OTEL_RESOURCE_ATTRIBUTES" ] = "k=v,k2=v2"
223
+ os .environ [resources . OTEL_RESOURCE_ATTRIBUTES ] = "k=v,k2=v2"
210
224
self .assertEqual (
211
225
detector .detect (), resources .Resource ({"k" : "v" , "k2" : "v2" })
212
226
)
213
227
214
228
def test_multiple_with_whitespace (self ):
215
229
detector = resources .OTELResourceDetector ()
216
- os .environ ["OTEL_RESOURCE_ATTRIBUTES" ] = " k = v , k2 = v2 "
230
+ os .environ [
231
+ resources .OTEL_RESOURCE_ATTRIBUTES
232
+ ] = " k = v , k2 = v2 "
217
233
self .assertEqual (
218
234
detector .detect (), resources .Resource ({"k" : "v" , "k2" : "v2" })
219
235
)
0 commit comments