feat: update contract tests for cfn primary id

Author: yiyuan-he

contract-tests/images/applications/botocore/botocore_server.py

@@ -246,7 +246,7 @@ def _handle_bedrock_request(self) -> None:
             set_main_status(200)
             bedrock_client.meta.events.register(
                 "before-call.bedrock.GetGuardrail",
-                lambda **kwargs: inject_200_success(guardrailId="bt4o77i015cu", **kwargs),
+                lambda **kwargs: inject_200_success(guardrailId="bt4o77i015cu", guardrailArn="arn:aws:bedrock:us-east-1:000000000000:guardrail/bt4o77i015cu", **kwargs),
             )
             bedrock_client.get_guardrail(
                 guardrailIdentifier="arn:aws:bedrock:us-east-1:000000000000:guardrail/bt4o77i015cu"
@@ -363,6 +363,9 @@ def inject_200_success(**kwargs):
     guardrail_id = kwargs.get("guardrailId")
     if guardrail_id is not None:
         response_body["guardrailId"] = guardrail_id
+    guardrail_arn = kwargs.get("guardrailArn")
+    if guardrail_arn is not None:
+        response_body["guardrailArn"] = guardrail_arn
 
     HTTPResponse = namedtuple("HTTPResponse", ["status_code", "headers", "body"])
     headers = kwargs.get("headers", {})

contract-tests/tests/test/amazon/botocore/botocore_test.py

@@ -10,6 +10,7 @@
 
 from amazon.base.contract_test_base import NETWORK_NAME, ContractTestBase
 from amazon.utils.application_signals_constants import (
+    AWS_CLOUDFORMATION_PRIMARY_IDENTIFIER,
     AWS_LOCAL_OPERATION,
     AWS_LOCAL_SERVICE,
     AWS_REMOTE_OPERATION,
@@ -99,6 +100,7 @@ def test_s3_create_bucket(self):
             remote_operation="CreateBucket",
             remote_resource_type="AWS::S3::Bucket",
             remote_resource_identifier="test-bucket-name",
+            cloudformation_primary_identifier="test-bucket-name",
             request_specific_attributes={
                 SpanAttributes.AWS_S3_BUCKET: "test-bucket-name",
             },
@@ -116,6 +118,7 @@ def test_s3_create_object(self):
             remote_operation="PutObject",
             remote_resource_type="AWS::S3::Bucket",
             remote_resource_identifier="test-put-object-bucket-name",
+            cloudformation_primary_identifier="test-put-object-bucket-name",
             request_specific_attributes={
                 SpanAttributes.AWS_S3_BUCKET: "test-put-object-bucket-name",
             },
@@ -133,6 +136,7 @@ def test_s3_get_object(self):
             remote_operation="GetObject",
             remote_resource_type="AWS::S3::Bucket",
             remote_resource_identifier="test-get-object-bucket-name",
+            cloudformation_primary_identifier="test-get-object-bucket-name",
             request_specific_attributes={
                 SpanAttributes.AWS_S3_BUCKET: "test-get-object-bucket-name",
             },
@@ -150,6 +154,7 @@ def test_s3_error(self):
             remote_operation="CreateBucket",
             remote_resource_type="AWS::S3::Bucket",
             remote_resource_identifier="-",
+            cloudformation_primary_identifier="-",
             request_specific_attributes={
                 SpanAttributes.AWS_S3_BUCKET: "-",
             },
@@ -167,6 +172,7 @@ def test_s3_fault(self):
             remote_operation="CreateBucket",
             remote_resource_type="AWS::S3::Bucket",
             remote_resource_identifier="valid-bucket-name",
+            cloudformation_primary_identifier="valid-bucket-name",
             request_specific_attributes={
                 SpanAttributes.AWS_S3_BUCKET: "valid-bucket-name",
             },
@@ -184,6 +190,7 @@ def test_dynamodb_create_table(self):
             remote_operation="CreateTable",
             remote_resource_type="AWS::DynamoDB::Table",
             remote_resource_identifier="test_table",
+            cloudformation_primary_identifier="test_table",
             request_specific_attributes={
                 SpanAttributes.AWS_DYNAMODB_TABLE_NAMES: ["test_table"],
             },
@@ -201,6 +208,7 @@ def test_dynamodb_put_item(self):
             remote_operation="PutItem",
             remote_resource_type="AWS::DynamoDB::Table",
             remote_resource_identifier="put_test_table",
+            cloudformation_primary_identifier="put_test_table",
             request_specific_attributes={
                 SpanAttributes.AWS_DYNAMODB_TABLE_NAMES: ["put_test_table"],
             },
@@ -218,6 +226,7 @@ def test_dynamodb_error(self):
             remote_operation="PutItem",
             remote_resource_type="AWS::DynamoDB::Table",
             remote_resource_identifier="invalid_table",
+            cloudformation_primary_identifier="invalid_table",
             request_specific_attributes={
                 SpanAttributes.AWS_DYNAMODB_TABLE_NAMES: ["invalid_table"],
             },
@@ -235,6 +244,7 @@ def test_dynamodb_fault(self):
             remote_operation="PutItem",
             remote_resource_type="AWS::DynamoDB::Table",
             remote_resource_identifier="invalid_table",
+            cloudformation_primary_identifier="invalid_table",
             request_specific_attributes={
                 SpanAttributes.AWS_DYNAMODB_TABLE_NAMES: ["invalid_table"],
             },
@@ -407,6 +417,7 @@ def test_bedrock_get_guardrail(self):
             remote_operation="GetGuardrail",
             remote_resource_type="AWS::Bedrock::Guardrail",
             remote_resource_identifier="bt4o77i015cu",
+            cloudformation_primary_identifier="arn:aws:bedrock:us-east-1:000000000000:guardrail/bt4o77i015cu",
             request_specific_attributes={
                 _AWS_BEDROCK_GUARDRAIL_ID: "bt4o77i015cu",
             },
@@ -519,6 +530,7 @@ def _assert_aws_span_attributes(self, resource_scope_spans: List[ResourceScopeSp
             "LOCAL_ROOT",
             kwargs.get("remote_resource_type", "None"),
             kwargs.get("remote_resource_identifier", "None"),
+            kwargs.get("cloudformation_primary_identifier", "None"),
         )
 
     def _assert_aws_attributes(
@@ -529,6 +541,7 @@ def _assert_aws_attributes(
         span_kind: str,
         remote_resource_type: str,
         remote_resource_identifier: str,
+        cloudformation_primary_identifier: str,
     ) -> None:
         attributes_dict: Dict[str, AnyValue] = self._get_attributes_dict(attributes_list)
         self._assert_str_attribute(attributes_dict, AWS_LOCAL_SERVICE, self.get_application_otel_service_name())
@@ -541,6 +554,8 @@ def _assert_aws_attributes(
             self._assert_str_attribute(attributes_dict, AWS_REMOTE_RESOURCE_TYPE, remote_resource_type)
         if remote_resource_identifier != "None":
             self._assert_str_attribute(attributes_dict, AWS_REMOTE_RESOURCE_IDENTIFIER, remote_resource_identifier)
+        if cloudformation_primary_identifier != "None":
+            self._assert_str_attribute(attributes_dict, AWS_CLOUDFORMATION_PRIMARY_IDENTIFIER, cloudformation_primary_identifier)
         # See comment above AWS_LOCAL_OPERATION
         self._assert_str_attribute(attributes_dict, AWS_SPAN_KIND, span_kind)
 

contract-tests/tests/test/amazon/utils/application_signals_constants.py

@@ -10,6 +10,7 @@
 FAULT_METRIC: str = "fault"
 
 # Attribute names
+AWS_CLOUDFORMATION_PRIMARY_IDENTIFIER: str = "aws.remote.resource.cfn.primary.identifier"
 AWS_LOCAL_SERVICE: str = "aws.local.service"
 AWS_LOCAL_OPERATION: str = "aws.local.operation"
 AWS_REMOTE_DB_USER: str = "aws.remote.db.user"