terraform/builtin/providers/profitbricks/resource_profitbricks_loadb...

200 lines
5.1 KiB
Go
Raw Normal View History

package profitbricks
import (
"fmt"
"testing"
"github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/terraform"
"github.com/profitbricks/profitbricks-sdk-go"
)
func TestAccProfitBricksLoadbalancer_Basic(t *testing.T) {
var loadbalancer profitbricks.Loadbalancer
lbName := "loadbalancer"
resource.Test(t, resource.TestCase{
PreCheck: func() {
testAccPreCheck(t)
},
Providers: testAccProviders,
CheckDestroy: testAccCheckDProfitBricksLoadbalancerDestroyCheck,
Steps: []resource.TestStep{
resource.TestStep{
Config: fmt.Sprintf(testAccCheckProfitbricksLoadbalancerConfig_basic, lbName),
Check: resource.ComposeTestCheckFunc(
testAccCheckProfitBricksLoadbalancerExists("profitbricks_loadbalancer.example", &loadbalancer),
testAccCheckProfitBricksLoadbalancerAttributes("profitbricks_loadbalancer.example", lbName),
resource.TestCheckResourceAttr("profitbricks_loadbalancer.example", "name", lbName),
),
},
resource.TestStep{
Config: testAccCheckProfitbricksLoadbalancerConfig_update,
Check: resource.ComposeTestCheckFunc(
testAccCheckProfitBricksLoadbalancerAttributes("profitbricks_loadbalancer.example", "updated"),
resource.TestCheckResourceAttr("profitbricks_loadbalancer.example", "name", "updated"),
),
},
},
})
}
func testAccCheckDProfitBricksLoadbalancerDestroyCheck(s *terraform.State) error {
for _, rs := range s.RootModule().Resources {
if rs.Type != "profitbricks_loadbalancer" {
continue
}
resp := profitbricks.GetLoadbalancer(rs.Primary.Attributes["datacenter_id"], rs.Primary.ID)
if resp.StatusCode < 299 {
resp := profitbricks.DeleteDatacenter(rs.Primary.Attributes["datacenter_id"])
if resp.StatusCode > 299 {
return fmt.Errorf("Firewall still exists %s %s", rs.Primary.ID, string(resp.Body))
}
}
}
return nil
}
func testAccCheckProfitBricksLoadbalancerAttributes(n string, name string) resource.TestCheckFunc {
return func(s *terraform.State) error {
rs, ok := s.RootModule().Resources[n]
if !ok {
return fmt.Errorf("testAccCheckProfitBricksLoadbalancerAttributes: Not found: %s", n)
}
if rs.Primary.Attributes["name"] != name {
return fmt.Errorf("Bad name: %s", rs.Primary.Attributes["name"])
}
return nil
}
}
func testAccCheckProfitBricksLoadbalancerExists(n string, loadbalancer *profitbricks.Loadbalancer) resource.TestCheckFunc {
return func(s *terraform.State) error {
rs, ok := s.RootModule().Resources[n]
if !ok {
return fmt.Errorf("testAccCheckProfitBricksLoadbalancerExists: Not found: %s", n)
}
if rs.Primary.ID == "" {
return fmt.Errorf("No Record ID is set")
}
foundLB := profitbricks.GetLoadbalancer(rs.Primary.Attributes["datacenter_id"], rs.Primary.ID)
if foundLB.StatusCode != 200 {
return fmt.Errorf("Error occured while fetching Loadbalancer: %s", rs.Primary.ID)
}
if foundLB.Id != rs.Primary.ID {
return fmt.Errorf("Record not found")
}
loadbalancer = &foundLB
return nil
}
}
const testAccCheckProfitbricksLoadbalancerConfig_basic = `
resource "profitbricks_datacenter" "foobar" {
name = "loadbalancer-test"
location = "us/las"
}
resource "profitbricks_server" "webserver" {
name = "webserver"
datacenter_id = "${profitbricks_datacenter.foobar.id}"
cores = 1
ram = 1024
availability_zone = "ZONE_1"
cpu_family = "AMD_OPTERON"
volume {
name = "system"
size = 5
disk_type = "SSD"
image_name ="ubuntu-16.04"
image_password = "test1234"
}
nic {
lan = "1"
dhcp = true
firewall_active = true
firewall {
protocol = "TCP"
name = "SSH"
port_range_start = 22
port_range_end = 22
}
}
}
resource "profitbricks_nic" "database_nic" {
datacenter_id = "${profitbricks_datacenter.foobar.id}"
server_id = "${profitbricks_server.webserver.id}"
lan = "2"
dhcp = true
firewall_active = true
name = "updated"
}
resource "profitbricks_loadbalancer" "example" {
datacenter_id = "${profitbricks_datacenter.foobar.id}"
nic_id = "${profitbricks_nic.database_nic.id}"
name = "%s"
dhcp = true
}`
const testAccCheckProfitbricksLoadbalancerConfig_update = `
resource "profitbricks_datacenter" "foobar" {
name = "loadbalancer-test"
location = "us/las"
}
resource "profitbricks_server" "webserver" {
name = "webserver"
datacenter_id = "${profitbricks_datacenter.foobar.id}"
cores = 1
ram = 1024
availability_zone = "ZONE_1"
cpu_family = "AMD_OPTERON"
volume {
name = "system"
size = 5
disk_type = "SSD"
image_name ="ubuntu-16.04"
image_password = "test1234"
}
nic {
lan = "1"
dhcp = true
firewall_active = true
firewall {
protocol = "TCP"
name = "SSH"
port_range_start = 22
port_range_end = 22
}
}
}
resource "profitbricks_nic" "database_nic" {
datacenter_id = "${profitbricks_datacenter.foobar.id}"
server_id = "${profitbricks_server.webserver.id}"
lan = "2"
dhcp = true
firewall_active = true
name = "updated"
}
resource "profitbricks_loadbalancer" "example" {
datacenter_id = "${profitbricks_datacenter.foobar.id}"
nic_id = "${profitbricks_nic.database_nic.id}"
name = "updated"
dhcp = true
}`