Add acceptance tests for regional NAT gateway data source

Author: tabito-hara

internal/service/ec2/vpc_nat_gateway_data_source_test.go

@@ -31,6 +31,8 @@ func TestAccVPCNATGatewayDataSource_basic(t *testing.T) {
 				Check: resource.ComposeAggregateTestCheckFunc(
 					resource.TestCheckResourceAttrPair(dataSourceNameByID, "allocation_id", resourceName, "allocation_id"),
 					resource.TestCheckResourceAttrPair(dataSourceNameByID, names.AttrAssociationID, resourceName, names.AttrAssociationID),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "availability_mode", resourceName, "availability_mode"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "availability_zone_address.#", resourceName, "availability_zone_address.#"),
 					resource.TestCheckResourceAttrPair(dataSourceNameByID, "connectivity_type", resourceName, "connectivity_type"),
 					resource.TestCheckResourceAttrPair(dataSourceNameByID, names.AttrNetworkInterfaceID, resourceName, names.AttrNetworkInterfaceID),
 					resource.TestCheckResourceAttrPair(dataSourceNameByID, "private_ip", resourceName, "private_ip"),
@@ -42,6 +44,8 @@ func TestAccVPCNATGatewayDataSource_basic(t *testing.T) {
 
 					resource.TestCheckResourceAttrPair(dataSourceNameBySubnetID, "allocation_id", resourceName, "allocation_id"),
 					resource.TestCheckResourceAttrPair(dataSourceNameBySubnetID, names.AttrAssociationID, resourceName, names.AttrAssociationID),
+					resource.TestCheckResourceAttrPair(dataSourceNameBySubnetID, "availability_mode", resourceName, "availability_mode"),
+					resource.TestCheckResourceAttrPair(dataSourceNameBySubnetID, "availability_zone_address.#", resourceName, "availability_zone_address.#"),
 					resource.TestCheckResourceAttrPair(dataSourceNameBySubnetID, "connectivity_type", resourceName, "connectivity_type"),
 					resource.TestCheckResourceAttrPair(dataSourceNameBySubnetID, names.AttrNetworkInterfaceID, resourceName, names.AttrNetworkInterfaceID),
 					resource.TestCheckResourceAttrPair(dataSourceNameBySubnetID, "private_ip", resourceName, "private_ip"),
@@ -53,6 +57,8 @@ func TestAccVPCNATGatewayDataSource_basic(t *testing.T) {
 
 					resource.TestCheckResourceAttrPair(dataSourceNameByTags, "allocation_id", resourceName, "allocation_id"),
 					resource.TestCheckResourceAttrPair(dataSourceNameByTags, names.AttrAssociationID, resourceName, names.AttrAssociationID),
+					resource.TestCheckResourceAttrPair(dataSourceNameByTags, "availability_mode", resourceName, "availability_mode"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByTags, "availability_zone_address.#", resourceName, "availability_zone_address.#"),
 					resource.TestCheckResourceAttrPair(dataSourceNameByTags, "connectivity_type", resourceName, "connectivity_type"),
 					resource.TestCheckResourceAttrPair(dataSourceNameByTags, names.AttrNetworkInterfaceID, resourceName, names.AttrNetworkInterfaceID),
 					resource.TestCheckResourceAttrPair(dataSourceNameByTags, "private_ip", resourceName, "private_ip"),
@@ -67,6 +73,45 @@ func TestAccVPCNATGatewayDataSource_basic(t *testing.T) {
 	})
 }
 
+func TestAccVPCNATGatewayDataSource_availabilityModeRegional(t *testing.T) {
+	ctx := acctest.Context(t)
+	rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
+	dataSourceNameByID := "data.aws_nat_gateway.test_by_id"
+	resourceName := "aws_nat_gateway.test"
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:                 func() { acctest.PreCheck(ctx, t) },
+		ErrorCheck:               acctest.ErrorCheck(t, names.EC2ServiceID),
+		ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccVPCNATGatewayDataSourceConfig_availabilityModeRegional(rName),
+				Check: resource.ComposeAggregateTestCheckFunc(
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "auto_provision_zones", resourceName, "auto_provision_zones"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "auto_scaling_ips", resourceName, "auto_scaling_ips"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "availability_mode", resourceName, "availability_mode"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "availability_zone_address.#", resourceName, "availability_zone_address.#"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "availability_zone_address.0.allocation_ids.#", resourceName, "availability_zone_address.0.allocation_ids.#"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "availability_zone_address.0.allocation_ids.0", resourceName, "availability_zone_address.0.allocation_ids.0"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "availability_zone_address.0.availability_zone", resourceName, "availability_zone_address.0.availability_zone"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "availability_zone_address.0.availability_zone_id", resourceName, "availability_zone_address.0.availability_zone_id"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "connectivity_type", resourceName, "connectivity_type"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "regional_nat_gateway_address.#", resourceName, "regional_nat_gateway_address.#"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "regional_nat_gateway_address.0.allocation_id", resourceName, "regional_nat_gateway_address.0.allocation_id"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "regional_nat_gateway_address.0.association_id", resourceName, "regional_nat_gateway_address.0.association_id"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "regional_nat_gateway_address.0.available_zone", resourceName, "regional_nat_gateway_address.0.available_zone"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "regional_nat_gateway_address.0.available_zone_id", resourceName, "regional_nat_gateway_address.0.available_zone_id"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "regional_nat_gateway_address.0.public_ip", resourceName, "regional_nat_gateway_address.0.public_ip"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "regional_nat_gateway_address.0.status", resourceName, "regional_nat_gateway_address.0.status"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "route_table_id", resourceName, "route_table_id"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, "tags.#", resourceName, "tags.#"),
+					resource.TestCheckResourceAttrPair(dataSourceNameByID, names.AttrVPCID, resourceName, names.AttrVPCID),
+				),
+			},
+		},
+	})
+}
+
 func testAccVPCNATGatewayDataSourceConfig_basic(rName string) string {
 	return acctest.ConfigCompose(testAccNATGatewayConfig_base(rName), fmt.Sprintf(`
 resource "aws_nat_gateway" "test" {
@@ -95,3 +140,24 @@ data "aws_nat_gateway" "test_by_tags" {
 }
 `, rName))
 }
+
+func testAccVPCNATGatewayDataSourceConfig_availabilityModeRegional(rName string) string {
+	return acctest.ConfigCompose(
+		testAccVPCNATGatewayConfig_availabilityModeRegionalBase(rName, 1), `
+resource "aws_nat_gateway" "test" {
+  vpc_id            = aws_vpc.test.id
+  availability_mode = "regional"
+
+  availability_zone_address {
+    allocation_ids    = [aws_eip.test[0].id]
+    availability_zone = data.aws_availability_zones.available.names[0]
+  }
+
+  depends_on = [aws_internet_gateway.test]
+}
+
+data "aws_nat_gateway" "test_by_id" {
+  id = aws_nat_gateway.test.id
+}
+`)
+}