2828)
2929
3030from 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+ )
3235from opentelemetry .test .test_base import TestBase
3336
3437
3538def assert_span_http_status_code (span , code ):
3639 """Assert on the span's 'http.status_code' tag"""
37- tag = span .attributes [SpanAttributes . HTTP_STATUS_CODE ]
40+ tag = span .attributes [HTTP_STATUS_CODE ]
3841 assert tag == code , f"{ tag } != { code } "
3942
4043
@@ -60,7 +63,7 @@ def test_ec2_client(self):
6063 span = spans [0 ]
6164 self .assertEqual (span .attributes ["aws.operation" ], "DescribeInstances" )
6265 assert_span_http_status_code (span , 200 )
63- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "POST" )
66+ self .assertEqual (span .attributes [HTTP_METHOD ], "POST" )
6467 self .assertEqual (span .attributes ["aws.region" ], "us-west-2" )
6568
6669 # Create an instance
@@ -73,7 +76,7 @@ def test_ec2_client(self):
7376 assert_span_http_status_code (span , 200 )
7477 self .assertEqual (span .attributes ["endpoint" ], "ec2" )
7578 self .assertEqual (span .attributes ["http_method" ], "runinstances" )
76- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "POST" )
79+ self .assertEqual (span .attributes [HTTP_METHOD ], "POST" )
7780 self .assertEqual (span .attributes ["aws.region" ], "us-west-2" )
7881 self .assertEqual (span .name , "ec2.command" )
7982
@@ -120,7 +123,7 @@ def test_s3_client(self):
120123 self .assertEqual (len (spans ), 1 )
121124 span = spans [0 ]
122125 assert_span_http_status_code (span , 200 )
123- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "GET" )
126+ self .assertEqual (span .attributes [HTTP_METHOD ], "GET" )
124127 self .assertEqual (span .attributes ["aws.operation" ], "get_all_buckets" )
125128
126129 # Create a bucket command
@@ -130,7 +133,7 @@ def test_s3_client(self):
130133 self .assertEqual (len (spans ), 2 )
131134 span = spans [1 ]
132135 assert_span_http_status_code (span , 200 )
133- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "PUT" )
136+ self .assertEqual (span .attributes [HTTP_METHOD ], "PUT" )
134137 self .assertEqual (span .attributes ["path" ], "/" )
135138 self .assertEqual (span .attributes ["aws.operation" ], "create_bucket" )
136139
@@ -143,7 +146,7 @@ def test_s3_client(self):
143146 assert_span_http_status_code (span , 200 )
144147 self .assertEqual (span .attributes ["endpoint" ], "s3" )
145148 self .assertEqual (span .attributes ["http_method" ], "head" )
146- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "HEAD" )
149+ self .assertEqual (span .attributes [HTTP_METHOD ], "HEAD" )
147150 self .assertEqual (span .attributes ["aws.operation" ], "head_bucket" )
148151 self .assertEqual (span .name , "s3.command" )
149152
@@ -240,7 +243,7 @@ def test_lambda_client(self):
240243 assert_span_http_status_code (span , 200 )
241244 self .assertEqual (span .attributes ["endpoint" ], "lambda" )
242245 self .assertEqual (span .attributes ["http_method" ], "get" )
243- self .assertEqual (span .attributes [SpanAttributes . HTTP_METHOD ], "GET" )
246+ self .assertEqual (span .attributes [HTTP_METHOD ], "GET" )
244247 self .assertEqual (span .attributes ["aws.region" ], "us-east-2" )
245248 self .assertEqual (span .attributes ["aws.operation" ], "list_functions" )
246249
0 commit comments