Updated and fixed test cases

hashicorp · Jun 26, 2018 · 43ea438 · 43ea438
1 parent 1eeea37
commit 43ea438
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/aws/data_source_aws_route_tables_test.go b/aws/data_source_aws_route_tables_test.go
@@ -21,8 +21,9 @@ func TestAccDataSourceAwsRouteTables(t *testing.T) {
 			{
 				Config: testAccDataSourceAwsRouteTablesConfigWithDataSource(rInt),
 				Check: resource.ComposeTestCheckFunc(
-					resource.TestCheckResourceAttr("data.aws_route_tables.selected", "ids.#", "3"),
+					resource.TestCheckResourceAttr("data.aws_route_tables.test", "ids.#", "4"),
 					resource.TestCheckResourceAttr("data.aws_route_tables.private", "ids.#", "2"),
+					resource.TestCheckResourceAttr("data.aws_route_tables.test2", "ids.#", "1"),
 				),
 			},
 		},
@@ -39,6 +40,14 @@ resource "aws_vpc" "test" {
   }
 }
 
+resource "aws_vpc" "test2" {
+  cidr_block = "172.%d.0.0/16"
+
+  tags {
+    Name = "terraform-test2acc-route-tables-data-source"
+  }
+}
+
 resource "aws_route_table" "test_public_a" {
   vpc_id            = "${aws_vpc.test.id}"
 
@@ -66,17 +75,21 @@ resource "aws_route_table" "test_private_b" {
   }
 }
 
-data "aws_route_tables" "selected" {
+data "aws_route_tables" "test" {
   vpc_id = "${aws_vpc.test.id}"
 }
 
+data "aws_route_tables" "test2" {
+  vpc_id = "${aws_vpc.test2.id}"
+}
+
 data "aws_route_tables" "private" {
   vpc_id = "${aws_vpc.test.id}"
   tags {
     Tier = "Private"
   }
 }
-`, rInt)
+`, rInt, rInt)
 }
 
 func testAccDataSourceAwsRouteTablesConfig(rInt int) string {