diff --git a/builtin/providers/aws/resource_aws_customer_gateway_test.go b/builtin/providers/aws/resource_aws_customer_gateway_test.go new file mode 100644 index 000000000..1d64bbd6b --- /dev/null +++ b/builtin/providers/aws/resource_aws_customer_gateway_test.go @@ -0,0 +1,110 @@ +package aws + +import ( + "fmt" + "testing" + + "github.com/awslabs/aws-sdk-go/aws" + "github.com/awslabs/aws-sdk-go/service/ec2" + "github.com/hashicorp/terraform/helper/resource" + "github.com/hashicorp/terraform/terraform" +) + +func TestAccCustomerGateway(t *testing.T) { + resource.Test(t, resource.TestCase{ + PreCheck: func() { testAccPreCheck(t) }, + Providers: testAccProviders, + CheckDestroy: testAccCheckCustomerGatewayDestroy, + Steps: []resource.TestStep{ + resource.TestStep{ + Config: testAccCustomerGatewayConfig, + Check: resource.ComposeTestCheckFunc( + testAccCheckCustomerGateway( + "aws_customer_gateway.foo", + ), + ), + }, + resource.TestStep{ + Config: testAccCustomerGatewayUpdate, + Check: resource.ComposeTestCheckFunc( + testAccCheckCustomerGateway( + "aws_customer_gateway.bar", + ), + ), + }, + }, + }) +} + +func testAccCheckCustomerGatewayDestroy(s *terraform.State) error { + if len(s.RootModule().Resources) > 0 { + return fmt.Errorf("Expected all resources to be gone, but found: %#v", s.RootModule().Resources) + } + + return nil +} + +func testAccCheckCustomerGateway(gatewayResource string) resource.TestCheckFunc { + return func(s *terraform.State) error { + rs, ok := s.RootModule().Resources[gatewayResource] + if !ok { + return fmt.Errorf("Not found: %s", gatewayResource) + } + + if rs.Primary.ID == "" { + return fmt.Errorf("No ID is set") + } + + gateway, ok := s.RootModule().Resources[gatewayResource] + if !ok { + return fmt.Errorf("Not found: %s", gatewayResource) + } + + ec2conn := testAccProvider.Meta().(*AWSClient).ec2conn + gatewayFilter := &ec2.Filter{ + Name: aws.String("customer-gateway-id"), + Values: []*string{aws.String(gateway.Primary.ID)}, + } + + _, err := ec2conn.DescribeCustomerGateways(&ec2.DescribeCustomerGatewaysInput{ + Filters: []*ec2.Filter{gatewayFilter}, + }) + + if err != nil { + return err + } + + return nil + } +} + +const testAccCustomerGatewayConfig = ` +resource "aws_customer_gateway" "foo" { + bgp_asn = 60000 + ip_address = "172.0.0.1" + type = ipsec.1 + tags { + Name = "foo-gateway" + } +} +` + +const testAccCustomerGatewayUpdate = ` +resource "aws_customer_gateway" "foo" { + bgp_asn = 60000 + ip_address = "172.0.0.1" + type = ipsec.1 + tags { + Name = "foo-gateway" + } +} + +resource "aws_customer_gateway" "bar" { + bgp_asn = 60000 + ip_address = "172.0.0.1" + type = ipsec.1 + tags { + Name = "foo-gateway" + } +} +`