package aws import ( "fmt" "testing" "github.com/hashicorp/terraform/helper/resource" "github.com/hashicorp/terraform/terraform" "github.com/hashicorp/aws-sdk-go/aws" "github.com/hashicorp/aws-sdk-go/gen/route53" ) func TestCleanPrefix(t *testing.T) { cases := []struct { Input, Prefix, Output string }{ {"/hostedzone/foo", "/hostedzone/", "foo"}, {"/change/foo", "/change/", "foo"}, {"/bar", "/test", "/bar"}, } for _, tc := range cases { actual := cleanPrefix(tc.Input, tc.Prefix) if actual != tc.Output { t.Fatalf("input: %s\noutput: %s", tc.Input, actual) } } } func TestCleanZoneID(t *testing.T) { cases := []struct { Input, Output string }{ {"/hostedzone/foo", "foo"}, {"/change/foo", "/change/foo"}, {"/bar", "/bar"}, } for _, tc := range cases { actual := cleanZoneID(tc.Input) if actual != tc.Output { t.Fatalf("input: %s\noutput: %s", tc.Input, actual) } } } func TestCleanChangeID(t *testing.T) { cases := []struct { Input, Output string }{ {"/hostedzone/foo", "/hostedzone/foo"}, {"/change/foo", "foo"}, {"/bar", "/bar"}, } for _, tc := range cases { actual := cleanChangeID(tc.Input) if actual != tc.Output { t.Fatalf("input: %s\noutput: %s", tc.Input, actual) } } } func TestAccRoute53Zone(t *testing.T) { resource.Test(t, resource.TestCase{ PreCheck: func() { testAccPreCheck(t) }, Providers: testAccProviders, CheckDestroy: testAccCheckRoute53ZoneDestroy, Steps: []resource.TestStep{ resource.TestStep{ Config: testAccRoute53ZoneConfig, Check: resource.ComposeTestCheckFunc( testAccCheckRoute53ZoneExists("aws_route53_zone.main"), ), }, }, }) } func testAccCheckRoute53ZoneDestroy(s *terraform.State) error { conn := testAccProvider.Meta().(*AWSClient).r53conn for _, rs := range s.RootModule().Resources { if rs.Type != "aws_route53_zone" { continue } _, err := conn.GetHostedZone(&route53.GetHostedZoneRequest{ID: aws.String(rs.Primary.ID)}) if err == nil { return fmt.Errorf("Hosted zone still exists") } } return nil } func testAccCheckRoute53ZoneExists(n string) resource.TestCheckFunc { return func(s *terraform.State) error { rs, ok := s.RootModule().Resources[n] if !ok { return fmt.Errorf("Not found: %s", n) } if rs.Primary.ID == "" { return fmt.Errorf("No hosted zone ID is set") } conn := testAccProvider.Meta().(*AWSClient).r53conn _, err := conn.GetHostedZone(&route53.GetHostedZoneRequest{ID: aws.String(rs.Primary.ID)}) if err != nil { return fmt.Errorf("Hosted zone err: %v", err) } return nil } } const testAccRoute53ZoneConfig = ` resource "aws_route53_zone" "main" { name = "hashicorp.com" } `