| // Copyright (c) HashiCorp, Inc. |
| // SPDX-License-Identifier: MPL-2.0 |
| package compute_test |
| |
| import ( |
| "fmt" |
| "strconv" |
| "testing" |
| |
| "github.com/hashicorp/terraform-provider-google-beta/google-beta/tpgresource" |
| |
| "github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource" |
| "github.com/hashicorp/terraform-plugin-sdk/v2/terraform" |
| "github.com/hashicorp/terraform-provider-google-beta/google-beta/acctest" |
| "github.com/hashicorp/terraform-provider-google-beta/google-beta/envvar" |
| |
| compute "google.golang.org/api/compute/v0.beta" |
| ) |
| |
| func TestAccComputeRegionDisk_basic(t *testing.T) { |
| t.Parallel() |
| |
| diskName := fmt.Sprintf("tf-test-%s", acctest.RandString(t, 10)) |
| |
| var disk compute.Disk |
| |
| acctest.VcrTest(t, resource.TestCase{ |
| PreCheck: func() { acctest.AccTestPreCheck(t) }, |
| ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories(t), |
| CheckDestroy: testAccCheckComputeRegionDiskDestroyProducer(t), |
| Steps: []resource.TestStep{ |
| { |
| Config: testAccComputeRegionDisk_basic(diskName, "self_link"), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk", &disk), |
| ), |
| }, |
| { |
| ResourceName: "google_compute_region_disk.regiondisk", |
| ImportState: true, |
| ImportStateVerify: true, |
| ImportStateVerifyIgnore: []string{"labels", "terraform_labels"}, |
| }, |
| { |
| Config: testAccComputeRegionDisk_basic(diskName, "name"), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk", &disk), |
| ), |
| }, |
| { |
| ResourceName: "google_compute_region_disk.regiondisk", |
| ImportState: true, |
| ImportStateVerify: true, |
| ImportStateVerifyIgnore: []string{"labels", "terraform_labels"}, |
| }, |
| }, |
| }) |
| } |
| |
| func TestAccComputeRegionDisk_basicUpdate(t *testing.T) { |
| t.Parallel() |
| |
| diskName := fmt.Sprintf("tf-test-%s", acctest.RandString(t, 10)) |
| |
| var disk compute.Disk |
| |
| acctest.VcrTest(t, resource.TestCase{ |
| PreCheck: func() { acctest.AccTestPreCheck(t) }, |
| ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories(t), |
| CheckDestroy: testAccCheckComputeRegionDiskDestroyProducer(t), |
| Steps: []resource.TestStep{ |
| { |
| Config: testAccComputeRegionDisk_basic(diskName, "self_link"), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk", &disk), |
| ), |
| }, |
| { |
| ResourceName: "google_compute_region_disk.regiondisk", |
| ImportState: true, |
| ImportStateVerify: true, |
| ImportStateVerifyIgnore: []string{"labels", "terraform_labels"}, |
| }, |
| { |
| Config: testAccComputeRegionDisk_basicUpdated(diskName, "self_link"), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk", &disk), |
| resource.TestCheckResourceAttr("google_compute_region_disk.regiondisk", "size", "100"), |
| testAccCheckComputeRegionDiskHasLabel(&disk, "my-label", "my-updated-label-value"), |
| testAccCheckComputeRegionDiskHasLabel(&disk, "a-new-label", "a-new-label-value"), |
| testAccCheckComputeRegionDiskHasLabelFingerprint(&disk, "google_compute_region_disk.regiondisk"), |
| ), |
| }, |
| { |
| ResourceName: "google_compute_region_disk.regiondisk", |
| ImportState: true, |
| ImportStateVerify: true, |
| ImportStateVerifyIgnore: []string{"labels", "terraform_labels"}, |
| }, |
| }, |
| }) |
| } |
| |
| func TestAccComputeRegionDisk_encryption(t *testing.T) { |
| t.Parallel() |
| |
| diskName := fmt.Sprintf("tf-test-%s", acctest.RandString(t, 10)) |
| var disk compute.Disk |
| |
| acctest.VcrTest(t, resource.TestCase{ |
| PreCheck: func() { acctest.AccTestPreCheck(t) }, |
| ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories(t), |
| CheckDestroy: testAccCheckComputeRegionDiskDestroyProducer(t), |
| Steps: []resource.TestStep{ |
| { |
| Config: testAccComputeRegionDisk_encryption(diskName), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk", &disk), |
| testAccCheckRegionDiskEncryptionKey( |
| "google_compute_region_disk.regiondisk", &disk), |
| ), |
| }, |
| }, |
| }) |
| } |
| |
| func TestAccComputeRegionDisk_deleteDetach(t *testing.T) { |
| t.Parallel() |
| |
| diskName := fmt.Sprintf("tf-test-%s", acctest.RandString(t, 10)) |
| regionDiskName := fmt.Sprintf("tf-test-%s", acctest.RandString(t, 10)) |
| regionDiskName2 := fmt.Sprintf("tf-test-%s", acctest.RandString(t, 10)) |
| instanceName := fmt.Sprintf("tf-test-%s", acctest.RandString(t, 10)) |
| var disk compute.Disk |
| |
| acctest.VcrTest(t, resource.TestCase{ |
| PreCheck: func() { acctest.AccTestPreCheck(t) }, |
| ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories(t), |
| CheckDestroy: testAccCheckComputeRegionDiskDestroyProducer(t), |
| Steps: []resource.TestStep{ |
| { |
| Config: testAccComputeRegionDisk_deleteDetach(instanceName, diskName, regionDiskName), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk", &disk), |
| ), |
| }, |
| // this needs to be an additional step so we refresh and see the instance |
| // listed as attached to the disk; the instance is created after the |
| // disk. and the disk's properties aren't refreshed unless there's |
| // another step |
| { |
| Config: testAccComputeRegionDisk_deleteDetach(instanceName, diskName, regionDiskName), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk", &disk), |
| testAccCheckComputeRegionDiskInstances( |
| "google_compute_region_disk.regiondisk", &disk), |
| ), |
| }, |
| // Change the disk name to destroy it, which detaches it from the instance |
| { |
| Config: testAccComputeRegionDisk_deleteDetach(instanceName, diskName, regionDiskName2), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk", &disk), |
| ), |
| }, |
| // Add the extra step like before |
| { |
| Config: testAccComputeRegionDisk_deleteDetach(instanceName, diskName, regionDiskName2), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk", &disk), |
| testAccCheckComputeRegionDiskInstances( |
| "google_compute_region_disk.regiondisk", &disk), |
| ), |
| }, |
| }, |
| }) |
| } |
| |
| func TestAccComputeRegionDisk_cloneDisk(t *testing.T) { |
| t.Parallel() |
| |
| diskName := fmt.Sprintf("tf-test-%s", acctest.RandString(t, 10)) |
| |
| var disk compute.Disk |
| |
| acctest.VcrTest(t, resource.TestCase{ |
| PreCheck: func() { acctest.AccTestPreCheck(t) }, |
| ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories(t), |
| CheckDestroy: testAccCheckComputeRegionDiskDestroyProducer(t), |
| Steps: []resource.TestStep{ |
| { |
| Config: testAccComputeRegionDisk_diskClone(diskName, "self_link"), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk-clone", &disk), |
| ), |
| }, |
| { |
| ResourceName: "google_compute_region_disk.regiondisk-clone", |
| ImportState: true, |
| ImportStateVerify: true, |
| }, |
| }, |
| }) |
| } |
| |
| func TestAccComputeRegionDisk_featuresUpdated(t *testing.T) { |
| t.Parallel() |
| |
| diskName := fmt.Sprintf("tf-test-%s", acctest.RandString(t, 10)) |
| |
| var disk compute.Disk |
| |
| acctest.VcrTest(t, resource.TestCase{ |
| PreCheck: func() { acctest.AccTestPreCheck(t) }, |
| ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories(t), |
| CheckDestroy: testAccCheckComputeRegionDiskDestroyProducer(t), |
| Steps: []resource.TestStep{ |
| { |
| Config: testAccComputeRegionDisk_features(diskName), |
| Check: resource.ComposeTestCheckFunc( |
| testAccCheckComputeRegionDiskExists( |
| t, "google_compute_region_disk.regiondisk", &disk), |
| ), |
| }, |
| { |
| ResourceName: "google_compute_region_disk.regiondisk", |
| ImportState: true, |
| ImportStateVerify: true, |
| }, |
| { |
| Config: testAccComputeRegionDisk_featuresUpdated(diskName), |
| }, |
| { |
| ResourceName: "google_compute_region_disk.regiondisk", |
| ImportState: true, |
| ImportStateVerify: true, |
| }, |
| }, |
| }) |
| } |
| |
| func testAccCheckComputeRegionDiskExists(t *testing.T, n string, disk *compute.Disk) resource.TestCheckFunc { |
| return func(s *terraform.State) error { |
| p := envvar.GetTestProjectFromEnv() |
| rs, ok := s.RootModule().Resources[n] |
| if !ok { |
| return fmt.Errorf("Not found: %s", n) |
| } |
| |
| if rs.Primary.Attributes["name"] == "" { |
| return fmt.Errorf("No ID is set") |
| } |
| |
| config := acctest.GoogleProviderConfig(t) |
| |
| found, err := config.NewComputeClient(config.UserAgent).RegionDisks.Get( |
| p, rs.Primary.Attributes["region"], rs.Primary.Attributes["name"]).Do() |
| if err != nil { |
| return err |
| } |
| |
| if found.Name != rs.Primary.Attributes["name"] { |
| return fmt.Errorf("RegionDisk not found") |
| } |
| |
| *disk = *found |
| |
| return nil |
| } |
| } |
| |
| func testAccCheckComputeRegionDiskHasLabel(disk *compute.Disk, key, value string) resource.TestCheckFunc { |
| return func(s *terraform.State) error { |
| val, ok := disk.Labels[key] |
| if !ok { |
| return fmt.Errorf("Label with key %s not found", key) |
| } |
| |
| if val != value { |
| return fmt.Errorf("Label value did not match for key %s: expected %s but found %s", key, value, val) |
| } |
| return nil |
| } |
| } |
| |
| func testAccCheckComputeRegionDiskHasLabelFingerprint(disk *compute.Disk, resourceName string) resource.TestCheckFunc { |
| return func(s *terraform.State) error { |
| state := s.RootModule().Resources[resourceName] |
| if state == nil { |
| return fmt.Errorf("Unable to find resource named %s", resourceName) |
| } |
| |
| labelFingerprint := state.Primary.Attributes["label_fingerprint"] |
| if labelFingerprint != disk.LabelFingerprint { |
| return fmt.Errorf("Label fingerprints do not match: api returned %s but state has %s", |
| disk.LabelFingerprint, labelFingerprint) |
| } |
| |
| return nil |
| } |
| } |
| |
| func testAccCheckRegionDiskEncryptionKey(n string, disk *compute.Disk) resource.TestCheckFunc { |
| return func(s *terraform.State) error { |
| rs, ok := s.RootModule().Resources[n] |
| if !ok { |
| return fmt.Errorf("Not found: %s", n) |
| } |
| |
| attr := rs.Primary.Attributes["disk_encryption_key.0.sha256"] |
| if disk.DiskEncryptionKey == nil { |
| return fmt.Errorf("RegionDisk %s has mismatched encryption key.\nTF State: %+v\nGCP State: <empty>", n, attr) |
| } else if attr != disk.DiskEncryptionKey.Sha256 { |
| return fmt.Errorf("RegionDisk %s has mismatched encryption key.\nTF State: %+v.\nGCP State: %+v", |
| n, attr, disk.DiskEncryptionKey.Sha256) |
| } |
| return nil |
| } |
| } |
| |
| func testAccCheckComputeRegionDiskInstances(n string, disk *compute.Disk) resource.TestCheckFunc { |
| return func(s *terraform.State) error { |
| rs, ok := s.RootModule().Resources[n] |
| if !ok { |
| return fmt.Errorf("Not found: %s", n) |
| } |
| |
| attr := rs.Primary.Attributes["users.#"] |
| if strconv.Itoa(len(disk.Users)) != attr { |
| return fmt.Errorf("RegionDisk %s has mismatched users.\nTF State: %+v\nGCP State: %+v", n, rs.Primary.Attributes["users"], disk.Users) |
| } |
| |
| for pos, user := range disk.Users { |
| if tpgresource.ConvertSelfLinkToV1(rs.Primary.Attributes["users."+strconv.Itoa(pos)]) != tpgresource.ConvertSelfLinkToV1(user) { |
| return fmt.Errorf("RegionDisk %s has mismatched users.\nTF State: %+v.\nGCP State: %+v", |
| n, rs.Primary.Attributes["users"], disk.Users) |
| } |
| } |
| return nil |
| } |
| } |
| |
| func testAccComputeRegionDisk_basic(diskName, refSelector string) string { |
| return fmt.Sprintf(` |
| resource "google_compute_disk" "disk" { |
| name = "%s" |
| image = "debian-cloud/debian-11" |
| size = 50 |
| type = "pd-ssd" |
| zone = "us-central1-a" |
| } |
| |
| resource "google_compute_snapshot" "snapdisk" { |
| name = "%s" |
| source_disk = google_compute_disk.disk.name |
| zone = "us-central1-a" |
| } |
| |
| resource "google_compute_region_disk" "regiondisk" { |
| name = "%s" |
| snapshot = google_compute_snapshot.snapdisk.%s |
| type = "pd-ssd" |
| replica_zones = ["us-central1-a", "us-central1-f"] |
| } |
| `, diskName, diskName, diskName, refSelector) |
| } |
| |
| func testAccComputeRegionDisk_basicUpdated(diskName, refSelector string) string { |
| return fmt.Sprintf(` |
| resource "google_compute_disk" "disk" { |
| name = "%s" |
| image = "debian-cloud/debian-11" |
| size = 50 |
| type = "pd-ssd" |
| zone = "us-central1-a" |
| } |
| |
| resource "google_compute_snapshot" "snapdisk" { |
| name = "%s" |
| source_disk = google_compute_disk.disk.name |
| zone = "us-central1-a" |
| } |
| |
| resource "google_compute_region_disk" "regiondisk" { |
| name = "%s" |
| snapshot = google_compute_snapshot.snapdisk.%s |
| type = "pd-ssd" |
| region = "us-central1" |
| |
| replica_zones = ["us-central1-a", "us-central1-f"] |
| |
| size = 100 |
| labels = { |
| my-label = "my-updated-label-value" |
| a-new-label = "a-new-label-value" |
| } |
| } |
| `, diskName, diskName, diskName, refSelector) |
| } |
| |
| func testAccComputeRegionDisk_encryption(diskName string) string { |
| return fmt.Sprintf(` |
| resource "google_compute_disk" "disk" { |
| name = "%s" |
| image = "debian-cloud/debian-11" |
| size = 50 |
| type = "pd-ssd" |
| zone = "us-central1-a" |
| } |
| |
| resource "google_compute_snapshot" "snapdisk" { |
| name = "%s" |
| zone = "us-central1-a" |
| |
| source_disk = google_compute_disk.disk.name |
| } |
| |
| resource "google_compute_region_disk" "regiondisk" { |
| name = "%s" |
| snapshot = google_compute_snapshot.snapdisk.self_link |
| type = "pd-ssd" |
| |
| replica_zones = ["us-central1-a", "us-central1-f"] |
| |
| disk_encryption_key { |
| raw_key = "SGVsbG8gZnJvbSBHb29nbGUgQ2xvdWQgUGxhdGZvcm0=" |
| } |
| } |
| `, diskName, diskName, diskName) |
| } |
| |
| func testAccComputeRegionDisk_deleteDetach(instanceName, diskName, regionDiskName string) string { |
| return fmt.Sprintf(` |
| resource "google_compute_disk" "disk" { |
| name = "%s" |
| image = "debian-cloud/debian-11" |
| size = 50 |
| type = "pd-ssd" |
| zone = "us-central1-a" |
| } |
| |
| resource "google_compute_snapshot" "snapdisk" { |
| name = "%s" |
| source_disk = google_compute_disk.disk.name |
| zone = "us-central1-a" |
| } |
| |
| resource "google_compute_region_disk" "regiondisk" { |
| name = "%s" |
| snapshot = google_compute_snapshot.snapdisk.self_link |
| type = "pd-ssd" |
| |
| replica_zones = ["us-central1-a", "us-central1-f"] |
| } |
| |
| resource "google_compute_instance" "inst" { |
| name = "%s" |
| machine_type = "e2-medium" |
| zone = "us-central1-a" |
| |
| boot_disk { |
| initialize_params { |
| image = "debian-cloud/debian-11" |
| } |
| } |
| |
| attached_disk { |
| source = google_compute_region_disk.regiondisk.self_link |
| } |
| |
| network_interface { |
| network = "default" |
| } |
| } |
| `, diskName, diskName, regionDiskName, instanceName) |
| } |
| |
| func testAccComputeRegionDisk_diskClone(diskName, refSelector string) string { |
| return fmt.Sprintf(` |
| resource "google_compute_region_disk" "regiondisk" { |
| name = "%s" |
| snapshot = google_compute_snapshot.snapdisk.id |
| type = "pd-ssd" |
| region = "us-central1" |
| physical_block_size_bytes = 4096 |
| |
| replica_zones = ["us-central1-a", "us-central1-f"] |
| } |
| |
| resource "google_compute_disk" "disk" { |
| name = "%s" |
| image = "debian-11-bullseye-v20220719" |
| size = 50 |
| type = "pd-ssd" |
| zone = "us-central1-a" |
| } |
| |
| resource "google_compute_snapshot" "snapdisk" { |
| name = "%s" |
| source_disk = google_compute_disk.disk.name |
| zone = "us-central1-a" |
| } |
| |
| resource "google_compute_region_disk" "regiondisk-clone" { |
| name = "%s" |
| source_disk = google_compute_region_disk.regiondisk.%s |
| type = "pd-ssd" |
| region = "us-central1" |
| physical_block_size_bytes = 4096 |
| |
| replica_zones = ["us-central1-a", "us-central1-f"] |
| } |
| `, diskName, diskName, diskName, diskName+"-clone", refSelector) |
| } |
| |
| func testAccComputeRegionDisk_features(diskName string) string { |
| return fmt.Sprintf(` |
| resource "google_compute_region_disk" "regiondisk" { |
| name = "%s" |
| type = "pd-ssd" |
| size = 50 |
| region = "us-central1" |
| |
| guest_os_features { |
| type = "SECURE_BOOT" |
| } |
| |
| replica_zones = ["us-central1-a", "us-central1-f"] |
| } |
| `, diskName) |
| } |
| |
| func testAccComputeRegionDisk_featuresUpdated(diskName string) string { |
| return fmt.Sprintf(` |
| resource "google_compute_region_disk" "regiondisk" { |
| name = "%s" |
| type = "pd-ssd" |
| size = 50 |
| region = "us-central1" |
| |
| guest_os_features { |
| type = "SECURE_BOOT" |
| } |
| |
| guest_os_features { |
| type = "MULTI_IP_SUBNET" |
| } |
| |
| replica_zones = ["us-central1-a", "us-central1-f"] |
| } |
| `, diskName) |
| } |
