25
25
26
26
from opentelemetry import trace
27
27
from opentelemetry .instrumentation .redis import RedisInstrumentor
28
- from opentelemetry .semconv .trace import (
28
+ from opentelemetry .semconv ._incubating .attributes .db_attributes import (
29
+ DB_REDIS_DATABASE_INDEX ,
30
+ DB_SYSTEM ,
29
31
DbSystemValues ,
32
+ )
33
+ from opentelemetry .semconv ._incubating .attributes .net_attributes import (
34
+ NET_PEER_NAME ,
35
+ NET_PEER_PORT ,
36
+ NET_TRANSPORT ,
30
37
NetTransportValues ,
31
- SpanAttributes ,
32
38
)
33
39
from opentelemetry .test .test_base import TestBase
34
40
from opentelemetry .trace import SpanKind
@@ -257,18 +263,14 @@ def test_attributes_default(self):
257
263
258
264
span = spans [0 ]
259
265
self .assertEqual (
260
- span .attributes [SpanAttributes . DB_SYSTEM ],
266
+ span .attributes [DB_SYSTEM ],
261
267
DbSystemValues .REDIS .value ,
262
268
)
269
+ self .assertEqual (span .attributes [DB_REDIS_DATABASE_INDEX ], 0 )
270
+ self .assertEqual (span .attributes [NET_PEER_NAME ], "localhost" )
271
+ self .assertEqual (span .attributes [NET_PEER_PORT ], 6379 )
263
272
self .assertEqual (
264
- span .attributes [SpanAttributes .DB_REDIS_DATABASE_INDEX ], 0
265
- )
266
- self .assertEqual (
267
- span .attributes [SpanAttributes .NET_PEER_NAME ], "localhost"
268
- )
269
- self .assertEqual (span .attributes [SpanAttributes .NET_PEER_PORT ], 6379 )
270
- self .assertEqual (
271
- span .attributes [SpanAttributes .NET_TRANSPORT ],
273
+ span .attributes [NET_TRANSPORT ],
272
274
NetTransportValues .IP_TCP .value ,
273
275
)
274
276
@@ -283,18 +285,14 @@ def test_attributes_tcp(self):
283
285
284
286
span = spans [0 ]
285
287
self .assertEqual (
286
- span .attributes [SpanAttributes . DB_SYSTEM ],
288
+ span .attributes [DB_SYSTEM ],
287
289
DbSystemValues .REDIS .value ,
288
290
)
291
+ self .assertEqual (span .attributes [DB_REDIS_DATABASE_INDEX ], 1 )
292
+ self .assertEqual (span .attributes [NET_PEER_NAME ], "1.1.1.1" )
293
+ self .assertEqual (span .attributes [NET_PEER_PORT ], 6380 )
289
294
self .assertEqual (
290
- span .attributes [SpanAttributes .DB_REDIS_DATABASE_INDEX ], 1
291
- )
292
- self .assertEqual (
293
- span .attributes [SpanAttributes .NET_PEER_NAME ], "1.1.1.1"
294
- )
295
- self .assertEqual (span .attributes [SpanAttributes .NET_PEER_PORT ], 6380 )
296
- self .assertEqual (
297
- span .attributes [SpanAttributes .NET_TRANSPORT ],
295
+ span .attributes [NET_TRANSPORT ],
298
296
NetTransportValues .IP_TCP .value ,
299
297
)
300
298
@@ -311,18 +309,16 @@ def test_attributes_unix_socket(self):
311
309
312
310
span = spans [0 ]
313
311
self .assertEqual (
314
- span .attributes [SpanAttributes . DB_SYSTEM ],
312
+ span .attributes [DB_SYSTEM ],
315
313
DbSystemValues .REDIS .value ,
316
314
)
315
+ self .assertEqual (span .attributes [DB_REDIS_DATABASE_INDEX ], 3 )
317
316
self .assertEqual (
318
- span .attributes [SpanAttributes .DB_REDIS_DATABASE_INDEX ], 3
319
- )
320
- self .assertEqual (
321
- span .attributes [SpanAttributes .NET_PEER_NAME ],
317
+ span .attributes [NET_PEER_NAME ],
322
318
"/path/to/socket.sock" ,
323
319
)
324
320
self .assertEqual (
325
- span .attributes [SpanAttributes . NET_TRANSPORT ],
321
+ span .attributes [NET_TRANSPORT ],
326
322
NetTransportValues .OTHER .value ,
327
323
)
328
324
0 commit comments