package digitalocean import ( "fmt" "strings" "testing" "github.com/hashicorp/terraform/helper/resource" "github.com/hashicorp/terraform/terraform" "github.com/pearkes/digitalocean" ) func TestAccDigitalOceanDroplet_Basic(t *testing.T) { var droplet digitalocean.Droplet resource.Test(t, resource.TestCase{ PreCheck: func() { testAccPreCheck(t) }, Providers: testAccProviders, CheckDestroy: testAccCheckDigitalOceanDropletDestroy, Steps: []resource.TestStep{ resource.TestStep{ Config: testAccCheckDigitalOceanDropletConfig_basic, Check: resource.ComposeTestCheckFunc( testAccCheckDigitalOceanDropletExists("digitalocean_droplet.foobar", &droplet), testAccCheckDigitalOceanDropletAttributes(&droplet), resource.TestCheckResourceAttr( "digitalocean_droplet.foobar", "name", "foo"), resource.TestCheckResourceAttr( "digitalocean_droplet.foobar", "size", "512mb"), resource.TestCheckResourceAttr( "digitalocean_droplet.foobar", "image", "centos-5-8-x32"), resource.TestCheckResourceAttr( "digitalocean_droplet.foobar", "region", "nyc3"), resource.TestCheckResourceAttr( "digitalocean_droplet.foobar", "user_data", "foobar"), ), }, }, }) } func TestAccDigitalOceanDroplet_Update(t *testing.T) { var droplet digitalocean.Droplet resource.Test(t, resource.TestCase{ PreCheck: func() { testAccPreCheck(t) }, Providers: testAccProviders, CheckDestroy: testAccCheckDigitalOceanDropletDestroy, Steps: []resource.TestStep{ resource.TestStep{ Config: testAccCheckDigitalOceanDropletConfig_basic, Check: resource.ComposeTestCheckFunc( testAccCheckDigitalOceanDropletExists("digitalocean_droplet.foobar", &droplet), testAccCheckDigitalOceanDropletAttributes(&droplet), ), }, resource.TestStep{ Config: testAccCheckDigitalOceanDropletConfig_RenameAndResize, Check: resource.ComposeTestCheckFunc( testAccCheckDigitalOceanDropletExists("digitalocean_droplet.foobar", &droplet), testAccCheckDigitalOceanDropletRenamedAndResized(&droplet), resource.TestCheckResourceAttr( "digitalocean_droplet.foobar", "name", "baz"), resource.TestCheckResourceAttr( "digitalocean_droplet.foobar", "size", "1gb"), ), }, }, }) } func TestAccDigitalOceanDroplet_PrivateNetworkingIpv6(t *testing.T) { var droplet digitalocean.Droplet resource.Test(t, resource.TestCase{ PreCheck: func() { testAccPreCheck(t) }, Providers: testAccProviders, CheckDestroy: testAccCheckDigitalOceanDropletDestroy, Steps: []resource.TestStep{ resource.TestStep{ Config: testAccCheckDigitalOceanDropletConfig_PrivateNetworkingIpv6, Check: resource.ComposeTestCheckFunc( testAccCheckDigitalOceanDropletExists("digitalocean_droplet.foobar", &droplet), testAccCheckDigitalOceanDropletAttributes_PrivateNetworkingIpv6(&droplet), resource.TestCheckResourceAttr( "digitalocean_droplet.foobar", "private_networking", "true"), resource.TestCheckResourceAttr( "digitalocean_droplet.foobar", "ipv6", "true"), ), }, }, }) } func testAccCheckDigitalOceanDropletDestroy(s *terraform.State) error { client := testAccProvider.client for _, rs := range s.RootModule().Resources { if rs.Type != "digitalocean_droplet" { continue } // Try to find the Droplet _, err := client.RetrieveDroplet(rs.Primary.ID) // Wait if err != nil && !strings.Contains(err.Error(), "404") { return fmt.Errorf( "Error waiting for droplet (%s) to be destroyed: %s", rs.Primary.ID, err) } } return nil } func testAccCheckDigitalOceanDropletAttributes(droplet *digitalocean.Droplet) resource.TestCheckFunc { return func(s *terraform.State) error { if droplet.ImageSlug() != "centos-5-8-x32" { return fmt.Errorf("Bad image_slug: %s", droplet.ImageSlug()) } if droplet.SizeSlug != "512mb" { return fmt.Errorf("Bad size_slug: %s", droplet.SizeSlug) } if droplet.RegionSlug() != "nyc3" { return fmt.Errorf("Bad region_slug: %s", droplet.RegionSlug()) } if droplet.Name != "foo" { return fmt.Errorf("Bad name: %s", droplet.Name) } return nil } } func testAccCheckDigitalOceanDropletRenamedAndResized(droplet *digitalocean.Droplet) resource.TestCheckFunc { return func(s *terraform.State) error { if droplet.SizeSlug != "1gb" { return fmt.Errorf("Bad size_slug: %s", droplet.SizeSlug) } if droplet.Name != "baz" { return fmt.Errorf("Bad name: %s", droplet.Name) } return nil } } func testAccCheckDigitalOceanDropletAttributes_PrivateNetworkingIpv6(droplet *digitalocean.Droplet) resource.TestCheckFunc { return func(s *terraform.State) error { if droplet.ImageSlug() != "centos-5-8-x32" { return fmt.Errorf("Bad image_slug: %s", droplet.ImageSlug()) } if droplet.SizeSlug != "1gb" { return fmt.Errorf("Bad size_slug: %s", droplet.SizeSlug) } if droplet.RegionSlug() != "sgp1" { return fmt.Errorf("Bad region_slug: %s", droplet.RegionSlug()) } if droplet.Name != "baz" { return fmt.Errorf("Bad name: %s", droplet.Name) } if droplet.IPV4Address("private") == "" { return fmt.Errorf("No ipv4 private: %s", droplet.IPV4Address("private")) } // if droplet.IPV6Address("private") == "" { // return fmt.Errorf("No ipv6 private: %s", droplet.IPV6Address("private")) // } if droplet.NetworkingType() != "private" { return fmt.Errorf("Bad networking type: %s", droplet.NetworkingType()) } if droplet.IPV4Address("public") == "" { return fmt.Errorf("No ipv4 public: %s", droplet.IPV4Address("public")) } if droplet.IPV6Address("public") == "" { return fmt.Errorf("No ipv6 public: %s", droplet.IPV6Address("public")) } return nil } } func testAccCheckDigitalOceanDropletExists(n string, droplet *digitalocean.Droplet) 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 Droplet ID is set") } client := testAccProvider.client retrieveDroplet, err := client.RetrieveDroplet(rs.Primary.ID) if err != nil { return err } if retrieveDroplet.StringId() != rs.Primary.ID { return fmt.Errorf("Droplet not found") } *droplet = retrieveDroplet return nil } } func Test_new_droplet_state_refresh_func(t *testing.T) { droplet := digitalocean.Droplet{ Name: "foobar", } resourceMap, _ := resource_digitalocean_droplet_update_state( &terraform.InstanceState{Attributes: map[string]string{}}, &droplet) // See if we can access our attribute if _, ok := resourceMap.Attributes["name"]; !ok { t.Fatalf("bad name: %s", resourceMap.Attributes) } } const testAccCheckDigitalOceanDropletConfig_basic = ` resource "digitalocean_droplet" "foobar" { name = "foo" size = "512mb" image = "centos-5-8-x32" region = "nyc3" user_data = "foobar" } ` const testAccCheckDigitalOceanDropletConfig_RenameAndResize = ` resource "digitalocean_droplet" "foobar" { name = "baz" size = "1gb" image = "centos-5-8-x32" region = "nyc3" } ` // IPV6 only in singapore const testAccCheckDigitalOceanDropletConfig_PrivateNetworkingIpv6 = ` resource "digitalocean_droplet" "foobar" { name = "baz" size = "1gb" image = "centos-5-8-x32" region = "sgp1" ipv6 = true private_networking = true } `