2828)
2929
3030from opentelemetry .instrumentation .boto import BotoInstrumentor
31- from opentelemetry .semconv .trace import SpanAttributes
31+ from opentelemetry .semconv ._incubating . attributes . http_attributes import HTTP_METHOD , HTTP_STATUS_CODE
3232from opentelemetry .test .test_base import TestBase
3333
3434
3535def assert_span_http_status_code (span , code ):
3636 """Assert on the span's 'http.status_code' tag"""
37- tag = span .attributes [SpanAttributes . HTTP_STATUS_CODE ]
37+ tag = span .attributes [HTTP_STATUS_CODE ]
3838 assert tag == code , f"{ tag } != { code } "
3939
4040
@@ -60,7 +60,7 @@ def test_ec2_client(self):
6060 span = spans [0 ]
6161 self .assertEqual (span .attributes ["aws.operation" ], "DescribeInstances" )
6262 assert_span_http_status_code (span , 200 )
63- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "POST" )
63+ self .assertEqual (span .attributes [HTTP_METHOD ], "POST" )
6464 self .assertEqual (span .attributes ["aws.region" ], "us-west-2" )
6565
6666 # Create an instance
@@ -73,7 +73,7 @@ def test_ec2_client(self):
7373 assert_span_http_status_code (span , 200 )
7474 self .assertEqual (span .attributes ["endpoint" ], "ec2" )
7575 self .assertEqual (span .attributes ["http_method" ], "runinstances" )
76- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "POST" )
76+ self .assertEqual (span .attributes [HTTP_METHOD ], "POST" )
7777 self .assertEqual (span .attributes ["aws.region" ], "us-west-2" )
7878 self .assertEqual (span .name , "ec2.command" )
7979
@@ -120,7 +120,7 @@ def test_s3_client(self):
120120 self .assertEqual (len (spans ), 1 )
121121 span = spans [0 ]
122122 assert_span_http_status_code (span , 200 )
123- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "GET" )
123+ self .assertEqual (span .attributes [HTTP_METHOD ], "GET" )
124124 self .assertEqual (span .attributes ["aws.operation" ], "get_all_buckets" )
125125
126126 # Create a bucket command
@@ -130,7 +130,7 @@ def test_s3_client(self):
130130 self .assertEqual (len (spans ), 2 )
131131 span = spans [1 ]
132132 assert_span_http_status_code (span , 200 )
133- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "PUT" )
133+ self .assertEqual (span .attributes [HTTP_METHOD ], "PUT" )
134134 self .assertEqual (span .attributes ["path" ], "/" )
135135 self .assertEqual (span .attributes ["aws.operation" ], "create_bucket" )
136136
@@ -143,7 +143,7 @@ def test_s3_client(self):
143143 assert_span_http_status_code (span , 200 )
144144 self .assertEqual (span .attributes ["endpoint" ], "s3" )
145145 self .assertEqual (span .attributes ["http_method" ], "head" )
146- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "HEAD" )
146+ self .assertEqual (span .attributes [HTTP_METHOD ], "HEAD" )
147147 self .assertEqual (span .attributes ["aws.operation" ], "head_bucket" )
148148 self .assertEqual (span .name , "s3.command" )
149149
@@ -240,7 +240,7 @@ def test_lambda_client(self):
240240 assert_span_http_status_code (span , 200 )
241241 self .assertEqual (span .attributes ["endpoint" ], "lambda" )
242242 self .assertEqual (span .attributes ["http_method" ], "get" )
243- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "GET" )
243+ self .assertEqual (span .attributes [HTTP_METHOD ], "GET" )
244244 self .assertEqual (span .attributes ["aws.region" ], "us-east-2" )
245245 self .assertEqual (span .attributes ["aws.operation" ], "list_functions" )
246246
0 commit comments