Use constants

hashicorp · Oct 2, 2024 · b4f4891 · b4f4891
1 parent 538eb71
commit b4f4891
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/service/bedrockagent/agent_test.go b/internal/service/bedrockagent/agent_test.go
@@ -406,7 +406,7 @@ func TestAccBedrockAgentAgent_kms(t *testing.T) {
 					resource.TestCheckResourceAttr(resourceName, "prompt_override_configuration.#", acctest.Ct1),
 					resource.TestCheckResourceAttr(resourceName, names.AttrDescription, "basic claude"),
 					resource.TestCheckResourceAttr(resourceName, "prepare_agent", acctest.CtTrue),
-					resource.TestCheckResourceAttrPair(resourceName, "customer_encryption_key_arn", "aws_kms_key.test_agent", "arn"),
+					resource.TestCheckResourceAttrPair(resourceName, "customer_encryption_key_arn", "aws_kms_key.test_agent", names.AttrARN),
 				),
 			},
 			{
@@ -424,7 +424,7 @@ func TestAccBedrockAgentAgent_kms(t *testing.T) {
 					resource.TestCheckResourceAttr(resourceName, "prompt_override_configuration.#", acctest.Ct1),
 					resource.TestCheckResourceAttr(resourceName, names.AttrDescription, "basic claude"),
 					resource.TestCheckResourceAttr(resourceName, "prepare_agent", acctest.CtTrue),
-					resource.TestCheckResourceAttrPair(resourceName, "customer_encryption_key_arn", "aws_kms_key.test_agent", "arn"),
+					resource.TestCheckResourceAttrPair(resourceName, "customer_encryption_key_arn", "aws_kms_key.test_agent", names.AttrARN),
 				),
 			},
 		},