|
1 | 1 | # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
|
2 | 2 | # SPDX-License-Identifier: Apache-2.0
|
| 3 | + |
| 4 | +# pylint: disable=too-many-lines |
| 5 | + |
3 | 6 | from typing import Dict, List, Optional
|
4 | 7 | from unittest import TestCase
|
5 | 8 | from unittest.mock import MagicMock
|
@@ -303,6 +306,101 @@ def test_server_span_with_span_name_with_http_target(self):
|
303 | 306 | self._mock_attribute(SpanAttributes.HTTP_METHOD, None)
|
304 | 307 | self._mock_attribute(SpanAttributes.HTTP_TARGET, None)
|
305 | 308 |
|
| 309 | + def test_server_span_with_span_name_with_target_and_url(self): |
| 310 | + # when http.target & http.url are present, the local operation should be derived from the http.target |
| 311 | + self._update_resource_with_service_name() |
| 312 | + self.span_mock.name = "POST" |
| 313 | + self._mock_attribute( |
| 314 | + [SpanAttributes.HTTP_METHOD, SpanAttributes.HTTP_TARGET, SpanAttributes.HTTP_URL], |
| 315 | + ["POST", "/my-target/09876", "http://127.0.0.1:8000/payment/123"], |
| 316 | + ) |
| 317 | + |
| 318 | + expected_attributes: Attributes = { |
| 319 | + AWS_SPAN_KIND: SpanKind.SERVER.name, |
| 320 | + AWS_LOCAL_SERVICE: _SERVICE_NAME_VALUE, |
| 321 | + AWS_LOCAL_OPERATION: "POST /my-target", |
| 322 | + } |
| 323 | + self._validate_attributes_produced_for_non_local_root_span_of_kind(expected_attributes, SpanKind.SERVER) |
| 324 | + self._mock_attribute(SpanAttributes.HTTP_METHOD, None) |
| 325 | + self._mock_attribute(SpanAttributes.HTTP_TARGET, None) |
| 326 | + self._mock_attribute(SpanAttributes.HTTP_URL, None) |
| 327 | + |
| 328 | + def test_server_span_with_span_name_with_http_url(self): |
| 329 | + self._update_resource_with_service_name() |
| 330 | + self.span_mock.name = "POST" |
| 331 | + self._mock_attribute( |
| 332 | + [SpanAttributes.HTTP_METHOD, SpanAttributes.HTTP_URL], ["POST", "http://127.0.0.1:8000/payment/123"] |
| 333 | + ) |
| 334 | + |
| 335 | + expected_attributes: Attributes = { |
| 336 | + AWS_SPAN_KIND: SpanKind.SERVER.name, |
| 337 | + AWS_LOCAL_SERVICE: _SERVICE_NAME_VALUE, |
| 338 | + AWS_LOCAL_OPERATION: "POST /payment", |
| 339 | + } |
| 340 | + self._validate_attributes_produced_for_non_local_root_span_of_kind(expected_attributes, SpanKind.SERVER) |
| 341 | + self._mock_attribute(SpanAttributes.HTTP_METHOD, None) |
| 342 | + self._mock_attribute(SpanAttributes.HTTP_URL, None) |
| 343 | + |
| 344 | + def test_server_span_with_http_url_with_no_path(self): |
| 345 | + # http.url with no path should result in local operation to be "POST /" |
| 346 | + self._update_resource_with_service_name() |
| 347 | + self.span_mock.name = "POST" |
| 348 | + self._mock_attribute([SpanAttributes.HTTP_METHOD, SpanAttributes.HTTP_URL], ["POST", "http://www.example.com"]) |
| 349 | + |
| 350 | + expected_attributes: Attributes = { |
| 351 | + AWS_SPAN_KIND: SpanKind.SERVER.name, |
| 352 | + AWS_LOCAL_SERVICE: _SERVICE_NAME_VALUE, |
| 353 | + AWS_LOCAL_OPERATION: "POST /", |
| 354 | + } |
| 355 | + self._validate_attributes_produced_for_non_local_root_span_of_kind(expected_attributes, SpanKind.SERVER) |
| 356 | + self._mock_attribute(SpanAttributes.HTTP_METHOD, None) |
| 357 | + self._mock_attribute(SpanAttributes.HTTP_URL, None) |
| 358 | + |
| 359 | + def test_server_span_with_http_url_as_none(self): |
| 360 | + # if http.url is none, local operation should default to UnknownOperation |
| 361 | + self._update_resource_with_service_name() |
| 362 | + self.span_mock.name = "POST" |
| 363 | + self._mock_attribute([SpanAttributes.HTTP_METHOD, SpanAttributes.HTTP_URL], ["POST", None]) |
| 364 | + |
| 365 | + expected_attributes: Attributes = { |
| 366 | + AWS_SPAN_KIND: SpanKind.SERVER.name, |
| 367 | + AWS_LOCAL_SERVICE: _SERVICE_NAME_VALUE, |
| 368 | + AWS_LOCAL_OPERATION: _UNKNOWN_OPERATION, |
| 369 | + } |
| 370 | + self._validate_attributes_produced_for_non_local_root_span_of_kind(expected_attributes, SpanKind.SERVER) |
| 371 | + self._mock_attribute(SpanAttributes.HTTP_METHOD, None) |
| 372 | + self._mock_attribute(SpanAttributes.HTTP_URL, None) |
| 373 | + |
| 374 | + def test_server_span_with_http_url_as_empty(self): |
| 375 | + # if http.url is empty, local operation should default to "POST /" |
| 376 | + self._update_resource_with_service_name() |
| 377 | + self.span_mock.name = "POST" |
| 378 | + self._mock_attribute([SpanAttributes.HTTP_METHOD, SpanAttributes.HTTP_URL], ["POST", ""]) |
| 379 | + |
| 380 | + expected_attributes: Attributes = { |
| 381 | + AWS_SPAN_KIND: SpanKind.SERVER.name, |
| 382 | + AWS_LOCAL_SERVICE: _SERVICE_NAME_VALUE, |
| 383 | + AWS_LOCAL_OPERATION: "POST /", |
| 384 | + } |
| 385 | + self._validate_attributes_produced_for_non_local_root_span_of_kind(expected_attributes, SpanKind.SERVER) |
| 386 | + self._mock_attribute(SpanAttributes.HTTP_METHOD, None) |
| 387 | + self._mock_attribute(SpanAttributes.HTTP_URL, None) |
| 388 | + |
| 389 | + def test_server_span_with_http_url_as_invalid(self): |
| 390 | + # if http.url is invalid, local operation should default to "POST /" |
| 391 | + self._update_resource_with_service_name() |
| 392 | + self.span_mock.name = "POST" |
| 393 | + self._mock_attribute([SpanAttributes.HTTP_METHOD, SpanAttributes.HTTP_URL], ["POST", "invalid_url"]) |
| 394 | + |
| 395 | + expected_attributes: Attributes = { |
| 396 | + AWS_SPAN_KIND: SpanKind.SERVER.name, |
| 397 | + AWS_LOCAL_SERVICE: _SERVICE_NAME_VALUE, |
| 398 | + AWS_LOCAL_OPERATION: "POST /", |
| 399 | + } |
| 400 | + self._validate_attributes_produced_for_non_local_root_span_of_kind(expected_attributes, SpanKind.SERVER) |
| 401 | + self._mock_attribute(SpanAttributes.HTTP_METHOD, None) |
| 402 | + self._mock_attribute(SpanAttributes.HTTP_URL, None) |
| 403 | + |
306 | 404 | def test_producer_span_with_attributes(self):
|
307 | 405 | self._update_resource_with_service_name()
|
308 | 406 | self._mock_attribute(
|
|
0 commit comments