diff --git a/internal/service/ec2/transitgateway_peering_attachment_test.go b/internal/service/ec2/transitgateway_peering_attachment_test.go index c48e0af26bc5..4f0a62242a1b 100644 --- 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)