terraform/builtin/providers/cloudstack/resource_cloudstack_static_...

129 lines
3.0 KiB
Go
Raw Normal View History

2016-04-04 15:48:14 +02:00
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,
)