28
28
)
29
29
30
30
from opentelemetry .instrumentation .boto import BotoInstrumentor
31
- from opentelemetry .semconv .trace import SpanAttributes
31
+ from opentelemetry .semconv ._incubating .attributes .http_attributes import (
32
+ HTTP_METHOD ,
33
+ HTTP_STATUS_CODE ,
34
+ )
32
35
from opentelemetry .test .test_base import TestBase
33
36
34
37
35
38
def assert_span_http_status_code (span , code ):
36
39
"""Assert on the span's 'http.status_code' tag"""
37
- tag = span .attributes [SpanAttributes . HTTP_STATUS_CODE ]
40
+ tag = span .attributes [HTTP_STATUS_CODE ]
38
41
assert tag == code , f"{ tag } != { code } "
39
42
40
43
@@ -60,7 +63,7 @@ def test_ec2_client(self):
60
63
span = spans [0 ]
61
64
self .assertEqual (span .attributes ["aws.operation" ], "DescribeInstances" )
62
65
assert_span_http_status_code (span , 200 )
63
- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "POST" )
66
+ self .assertEqual (span .attributes [HTTP_METHOD ], "POST" )
64
67
self .assertEqual (span .attributes ["aws.region" ], "us-west-2" )
65
68
66
69
# Create an instance
@@ -73,7 +76,7 @@ def test_ec2_client(self):
73
76
assert_span_http_status_code (span , 200 )
74
77
self .assertEqual (span .attributes ["endpoint" ], "ec2" )
75
78
self .assertEqual (span .attributes ["http_method" ], "runinstances" )
76
- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "POST" )
79
+ self .assertEqual (span .attributes [HTTP_METHOD ], "POST" )
77
80
self .assertEqual (span .attributes ["aws.region" ], "us-west-2" )
78
81
self .assertEqual (span .name , "ec2.command" )
79
82
@@ -120,7 +123,7 @@ def test_s3_client(self):
120
123
self .assertEqual (len (spans ), 1 )
121
124
span = spans [0 ]
122
125
assert_span_http_status_code (span , 200 )
123
- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "GET" )
126
+ self .assertEqual (span .attributes [HTTP_METHOD ], "GET" )
124
127
self .assertEqual (span .attributes ["aws.operation" ], "get_all_buckets" )
125
128
126
129
# Create a bucket command
@@ -130,7 +133,7 @@ def test_s3_client(self):
130
133
self .assertEqual (len (spans ), 2 )
131
134
span = spans [1 ]
132
135
assert_span_http_status_code (span , 200 )
133
- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "PUT" )
136
+ self .assertEqual (span .attributes [HTTP_METHOD ], "PUT" )
134
137
self .assertEqual (span .attributes ["path" ], "/" )
135
138
self .assertEqual (span .attributes ["aws.operation" ], "create_bucket" )
136
139
@@ -143,7 +146,7 @@ def test_s3_client(self):
143
146
assert_span_http_status_code (span , 200 )
144
147
self .assertEqual (span .attributes ["endpoint" ], "s3" )
145
148
self .assertEqual (span .attributes ["http_method" ], "head" )
146
- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "HEAD" )
149
+ self .assertEqual (span .attributes [HTTP_METHOD ], "HEAD" )
147
150
self .assertEqual (span .attributes ["aws.operation" ], "head_bucket" )
148
151
self .assertEqual (span .name , "s3.command" )
149
152
@@ -240,7 +243,7 @@ def test_lambda_client(self):
240
243
assert_span_http_status_code (span , 200 )
241
244
self .assertEqual (span .attributes ["endpoint" ], "lambda" )
242
245
self .assertEqual (span .attributes ["http_method" ], "get" )
243
- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "GET" )
246
+ self .assertEqual (span .attributes [HTTP_METHOD ], "GET" )
244
247
self .assertEqual (span .attributes ["aws.region" ], "us-east-2" )
245
248
self .assertEqual (span .attributes ["aws.operation" ], "list_functions" )
246
249
0 commit comments