explicitly check the peering attachment state

hashicorp · Mar 11, 2024 · f9e3812 · f9e3812
1 parent 5c64a39
commit f9e3812
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/internal/service/ec2/transitgateway_peering_attachment_test.go b/internal/service/ec2/transitgateway_peering_attachment_test.go
@@ -47,6 +47,7 @@ func testAccTransitGatewayPeeringAttachment_basic(t *testing.T) {
 					resource.TestCheckResourceAttr(resourceName, "tags.%", "0"),
 					resource.TestCheckResourceAttrPair(resourceName, "transit_gateway_id", transitGatewayResourceName, "id"),
 					resource.TestCheckResourceAttrSet(resourceName, "state"),
+					testAccCheckTransitGatewayPeeringAttachmentState(resourceName, &transitGatewayPeeringAttachment),
 				),
 			},
 			{
@@ -173,7 +174,7 @@ func testAccTransitGatewayPeeringAttachment_differentAccount(t *testing.T) {
 					resource.TestCheckResourceAttr(resourceName, "tags.%", "1"),
 					resource.TestCheckResourceAttr(resourceName, "tags.Name", rName),
 					resource.TestCheckResourceAttrPair(resourceName, "transit_gateway_id", transitGatewayResourceName, "id"),
-					resource.TestCheckResourceAttrSet(resourceName, "state"),
+					resource.TestCheckResourceAttrPtr(resourceName, "state", transitGatewayPeeringAttachment.State),
 				),
 			},
 			{
@@ -211,6 +212,12 @@ func testAccCheckTransitGatewayPeeringAttachmentExists(ctx context.Context, n st
 	}
 }
 
+func testAccCheckTransitGatewayPeeringAttachmentState(resourceName string, v *ec2.TransitGatewayPeeringAttachment) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+		return resource.TestCheckResourceAttr(resourceName, "state", *v.State)(s)
+	}
+}
+
 func testAccCheckTransitGatewayPeeringAttachmentDestroy(ctx context.Context) resource.TestCheckFunc {
 	return func(s *terraform.State) error {
 		conn := acctest.Provider.Meta().(*conns.AWSClient).EC2Conn(ctx)