115 lines
2.9 KiB
Go
115 lines
2.9 KiB
Go
package aws
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform/helper/acctest"
|
|
"github.com/hashicorp/terraform/helper/resource"
|
|
"github.com/hashicorp/terraform/terraform"
|
|
)
|
|
|
|
func TestAccDataSourceAwsRoute53Zone(t *testing.T) {
|
|
rInt := acctest.RandInt()
|
|
publicResourceName := "aws_route53_zone.test"
|
|
publicDomain := fmt.Sprintf("terraformtestacchz-%d.com.", rInt)
|
|
privateResourceName := "aws_route53_zone.test_private"
|
|
privateDomain := fmt.Sprintf("test.acc-%d.", rInt)
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
Steps: []resource.TestStep{
|
|
{
|
|
Config: testAccDataSourceAwsRoute53ZoneConfig(rInt),
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccDataSourceAwsRoute53ZoneCheck(
|
|
publicResourceName, "data.aws_route53_zone.by_zone_id", publicDomain),
|
|
testAccDataSourceAwsRoute53ZoneCheck(
|
|
publicResourceName, "data.aws_route53_zone.by_name", publicDomain),
|
|
testAccDataSourceAwsRoute53ZoneCheck(
|
|
privateResourceName, "data.aws_route53_zone.by_vpc", privateDomain),
|
|
testAccDataSourceAwsRoute53ZoneCheck(
|
|
privateResourceName, "data.aws_route53_zone.by_tag", privateDomain),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
// rsName for the name of the created resource
|
|
// dsName for the name of the created data source
|
|
// zName for the name of the domain
|
|
func testAccDataSourceAwsRoute53ZoneCheck(rsName, dsName, zName string) resource.TestCheckFunc {
|
|
return func(s *terraform.State) error {
|
|
rs, ok := s.RootModule().Resources[rsName]
|
|
if !ok {
|
|
return fmt.Errorf("root module has no resource called %s", rsName)
|
|
}
|
|
|
|
hostedZone, ok := s.RootModule().Resources[dsName]
|
|
if !ok {
|
|
return fmt.Errorf("can't find zone %q in state", dsName)
|
|
}
|
|
|
|
attr := rs.Primary.Attributes
|
|
if attr["id"] != hostedZone.Primary.Attributes["id"] {
|
|
return fmt.Errorf(
|
|
"id is %s; want %s",
|
|
attr["id"],
|
|
hostedZone.Primary.Attributes["id"],
|
|
)
|
|
}
|
|
|
|
if attr["name"] != zName {
|
|
return fmt.Errorf("Route53 Zone name is %q; want %q", attr["name"], zName)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func testAccDataSourceAwsRoute53ZoneConfig(rInt int) string {
|
|
return fmt.Sprintf(`
|
|
provider "aws" {
|
|
region = "us-east-1"
|
|
}
|
|
|
|
resource "aws_vpc" "test" {
|
|
cidr_block = "172.16.0.0/16"
|
|
}
|
|
|
|
resource "aws_route53_zone" "test_private" {
|
|
name = "test.acc-%d."
|
|
vpc_id = "${aws_vpc.test.id}"
|
|
tags {
|
|
Environment = "dev-%d"
|
|
}
|
|
}
|
|
|
|
data "aws_route53_zone" "by_vpc" {
|
|
name = "${aws_route53_zone.test_private.name}"
|
|
vpc_id = "${aws_vpc.test.id}"
|
|
}
|
|
|
|
data "aws_route53_zone" "by_tag" {
|
|
name = "${aws_route53_zone.test_private.name}"
|
|
private_zone = true
|
|
tags {
|
|
Environment = "dev-%d"
|
|
}
|
|
}
|
|
|
|
resource "aws_route53_zone" "test" {
|
|
name = "terraformtestacchz-%d.com."
|
|
}
|
|
|
|
data "aws_route53_zone" "by_zone_id" {
|
|
zone_id = "${aws_route53_zone.test.zone_id}"
|
|
}
|
|
|
|
data "aws_route53_zone" "by_name" {
|
|
name = "${data.aws_route53_zone.by_zone_id.name}"
|
|
}`, rInt, rInt, rInt, rInt)
|
|
}
|