Skip to content

Commit b64f519

Browse files
committed
refactor: migrate http_attributes from SpanAttributes to new semantic conventions and refactor tests
1 parent 0db4d0b commit b64f519

File tree

2 files changed

+11
-11
lines changed

2 files changed

+11
-11
lines changed

instrumentation/opentelemetry-instrumentation-boto/src/opentelemetry/instrumentation/boto/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@
5252
from opentelemetry.instrumentation.boto.version import __version__
5353
from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
5454
from opentelemetry.instrumentation.utils import unwrap
55-
from opentelemetry.semconv.trace import SpanAttributes
55+
from opentelemetry.semconv._incubating.attributes.http_attributes import HTTP_METHOD, HTTP_STATUS_CODE
5656
from opentelemetry.trace import SpanKind, get_tracer
5757

5858
logger = logging.getLogger(__name__)
@@ -159,10 +159,10 @@ def _common_request( # pylint: disable=too-many-locals
159159
span.set_attribute(key, value)
160160

161161
span.set_attribute(
162-
SpanAttributes.HTTP_STATUS_CODE, getattr(result, "status")
162+
HTTP_STATUS_CODE, getattr(result, "status")
163163
)
164164
span.set_attribute(
165-
SpanAttributes.HTTP_METHOD, getattr(result, "_method")
165+
HTTP_METHOD, getattr(result, "_method")
166166
)
167167

168168
return result

instrumentation/opentelemetry-instrumentation-boto/tests/test_boto_instrumentation.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -28,13 +28,13 @@
2828
)
2929

3030
from 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
3232
from opentelemetry.test.test_base import TestBase
3333

3434

3535
def 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

Comments
 (0)