package google import ( "fmt" "testing" "github.com/hashicorp/terraform/helper/acctest" "github.com/hashicorp/terraform/helper/resource" "github.com/hashicorp/terraform/terraform" "google.golang.org/api/compute/v1" ) func TestAccAutoscaler_basic(t *testing.T) { var ascaler compute.Autoscaler resource.Test(t, resource.TestCase{ PreCheck: func() { testAccPreCheck(t) }, Providers: testAccProviders, CheckDestroy: testAccCheckAutoscalerDestroy, Steps: []resource.TestStep{ resource.TestStep{ Config: testAccAutoscaler_basic, Check: resource.ComposeTestCheckFunc( testAccCheckAutoscalerExists( "google_compute_autoscaler.foobar", &ascaler), ), }, }, }) } func TestAccAutoscaler_update(t *testing.T) { var ascaler compute.Autoscaler resource.Test(t, resource.TestCase{ PreCheck: func() { testAccPreCheck(t) }, Providers: testAccProviders, CheckDestroy: testAccCheckAutoscalerDestroy, Steps: []resource.TestStep{ resource.TestStep{ Config: testAccAutoscaler_basic, Check: resource.ComposeTestCheckFunc( testAccCheckAutoscalerExists( "google_compute_autoscaler.foobar", &ascaler), ), }, resource.TestStep{ Config: testAccAutoscaler_update, Check: resource.ComposeTestCheckFunc( testAccCheckAutoscalerExists( "google_compute_autoscaler.foobar", &ascaler), testAccCheckAutoscalerUpdated( "google_compute_autoscaler.foobar", 10), ), }, }, }) } func testAccCheckAutoscalerDestroy(s *terraform.State) error { config := testAccProvider.Meta().(*Config) for _, rs := range s.RootModule().Resources { if rs.Type != "google_compute_autoscaler" { continue } _, err := config.clientCompute.Autoscalers.Get( config.Project, rs.Primary.Attributes["zone"], rs.Primary.ID).Do() if err == nil { return fmt.Errorf("Autoscaler still exists") } } return nil } func testAccCheckAutoscalerExists(n string, ascaler *compute.Autoscaler) 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 ID is set") } config := testAccProvider.Meta().(*Config) found, err := config.clientCompute.Autoscalers.Get( config.Project, rs.Primary.Attributes["zone"], rs.Primary.ID).Do() if err != nil { return err } if found.Name != rs.Primary.ID { return fmt.Errorf("Autoscaler not found") } *ascaler = *found return nil } } func testAccCheckAutoscalerUpdated(n string, max int64) 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 ID is set") } config := testAccProvider.Meta().(*Config) ascaler, err := config.clientCompute.Autoscalers.Get( config.Project, rs.Primary.Attributes["zone"], rs.Primary.ID).Do() if err != nil { return err } if ascaler.AutoscalingPolicy.MaxNumReplicas != max { return fmt.Errorf("maximum replicas incorrect") } return nil } } var testAccAutoscaler_basic = fmt.Sprintf(` resource "google_compute_instance_template" "foobar" { name = "ascaler-test-%s" machine_type = "n1-standard-1" can_ip_forward = false tags = ["foo", "bar"] disk { source_image = "debian-cloud/debian-7-wheezy-v20160301" auto_delete = true boot = true } network_interface { network = "default" } metadata { foo = "bar" } service_account { scopes = ["userinfo-email", "compute-ro", "storage-ro"] } } resource "google_compute_target_pool" "foobar" { description = "Resource created for Terraform acceptance testing" name = "ascaler-test-%s" session_affinity = "CLIENT_IP_PROTO" } resource "google_compute_instance_group_manager" "foobar" { description = "Terraform test instance group manager" name = "ascaler-test-%s" instance_template = "${google_compute_instance_template.foobar.self_link}" target_pools = ["${google_compute_target_pool.foobar.self_link}"] base_instance_name = "foobar" zone = "us-central1-a" } resource "google_compute_autoscaler" "foobar" { description = "Resource created for Terraform acceptance testing" name = "ascaler-test-%s" zone = "us-central1-a" target = "${google_compute_instance_group_manager.foobar.self_link}" autoscaling_policy = { max_replicas = 5 min_replicas = 0 cooldown_period = 60 cpu_utilization = { target = 0.5 } } }`, acctest.RandString(10), acctest.RandString(10), acctest.RandString(10), acctest.RandString(10)) var testAccAutoscaler_update = fmt.Sprintf(` resource "google_compute_instance_template" "foobar" { name = "ascaler-test-%s" machine_type = "n1-standard-1" can_ip_forward = false tags = ["foo", "bar"] disk { source_image = "debian-cloud/debian-7-wheezy-v20160301" auto_delete = true boot = true } network_interface { network = "default" } metadata { foo = "bar" } service_account { scopes = ["userinfo-email", "compute-ro", "storage-ro"] } } resource "google_compute_target_pool" "foobar" { description = "Resource created for Terraform acceptance testing" name = "ascaler-test-%s" session_affinity = "CLIENT_IP_PROTO" } resource "google_compute_instance_group_manager" "foobar" { description = "Terraform test instance group manager" name = "ascaler-test-%s" instance_template = "${google_compute_instance_template.foobar.self_link}" target_pools = ["${google_compute_target_pool.foobar.self_link}"] base_instance_name = "foobar" zone = "us-central1-a" } resource "google_compute_autoscaler" "foobar" { description = "Resource created for Terraform acceptance testing" name = "ascaler-test-%s" zone = "us-central1-a" target = "${google_compute_instance_group_manager.foobar.self_link}" autoscaling_policy = { max_replicas = 10 min_replicas = 0 cooldown_period = 60 cpu_utilization = { target = 0.5 } } }`, acctest.RandString(10), acctest.RandString(10), acctest.RandString(10), acctest.RandString(10))