package cloudstack import ( "fmt" "testing" "github.com/hashicorp/terraform/helper/resource" "github.com/hashicorp/terraform/terraform" "github.com/xanzy/go-cloudstack/cloudstack" ) func TestAccCloudStackStaticNAT_basic(t *testing.T) { var ipaddr cloudstack.PublicIpAddress resource.Test(t, resource.TestCase{ PreCheck: func() { testAccPreCheck(t) }, Providers: testAccProviders, CheckDestroy: testAccCheckCloudStackStaticNATDestroy, Steps: []resource.TestStep{ resource.TestStep{ Config: testAccCloudStackStaticNAT_basic, Check: resource.ComposeTestCheckFunc( testAccCheckCloudStackStaticNATExists( "cloudstack_static_nat.foo", &ipaddr), testAccCheckCloudStackStaticNATAttributes(&ipaddr), ), }, }, }) } func testAccCheckCloudStackStaticNATExists( n string, ipaddr *cloudstack.PublicIpAddress) 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 static NAT ID is set") } cs := testAccProvider.Meta().(*cloudstack.CloudStackClient) ip, _, err := cs.Address.GetPublicIpAddressByID(rs.Primary.ID) if err != nil { return err } if ip.Id != rs.Primary.ID { return fmt.Errorf("Static NAT not found") } if !ip.Isstaticnat { return fmt.Errorf("Static NAT not enabled") } *ipaddr = *ip return nil } } func testAccCheckCloudStackStaticNATAttributes( ipaddr *cloudstack.PublicIpAddress) resource.TestCheckFunc { return func(s *terraform.State) error { if ipaddr.Associatednetworkname != CLOUDSTACK_NETWORK_1 { return fmt.Errorf("Bad network: %s", ipaddr.Associatednetworkname) } if ipaddr.Virtualmachinename != "terraform-test" { return fmt.Errorf("Bad virtual_machine: %s", ipaddr.Virtualmachinename) } return nil } } func testAccCheckCloudStackStaticNATDestroy(s *terraform.State) error { cs := testAccProvider.Meta().(*cloudstack.CloudStackClient) for _, rs := range s.RootModule().Resources { if rs.Type != "cloudstack_static_nat" { continue } if rs.Primary.ID == "" { return fmt.Errorf("No static NAT ID is set") } ip, _, err := cs.Address.GetPublicIpAddressByID(rs.Primary.ID) if err == nil && ip.Isstaticnat { return fmt.Errorf("Static NAT %s still enabled", rs.Primary.ID) } } return nil } var testAccCloudStackStaticNAT_basic = fmt.Sprintf(` resource "cloudstack_instance" "foobar" { name = "terraform-test" display_name = "terraform-test" service_offering= "%s" network = "%s" template = "%s" zone = "%s" user_data = "foobar\nfoo\nbar" expunge = true } resource "cloudstack_ipaddress" "foo" { network = "%s" } resource "cloudstack_static_nat" "foo" { ipaddress = "${cloudstack_ipaddress.foo.id}" network = "${cloudstack_ipaddress.foo.network}" virtual_machine = "${cloudstack_instance.foobar.id}" }`, CLOUDSTACK_SERVICE_OFFERING_1, CLOUDSTACK_NETWORK_1, CLOUDSTACK_TEMPLATE, CLOUDSTACK_ZONE, CLOUDSTACK_NETWORK_1, )