316 lines
9.3 KiB
Go
316 lines
9.3 KiB
Go
package google
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform/helper/acctest"
|
|
"github.com/hashicorp/terraform/helper/resource"
|
|
"github.com/hashicorp/terraform/terraform"
|
|
"google.golang.org/api/compute/v1"
|
|
)
|
|
|
|
// Add two key value pairs
|
|
func TestAccComputeProjectMetadata_basic(t *testing.T) {
|
|
skipIfEnvNotSet(t,
|
|
[]string{
|
|
"GOOGLE_ORG",
|
|
"GOOGLE_BILLING_ACCOUNT",
|
|
}...,
|
|
)
|
|
|
|
billingId := os.Getenv("GOOGLE_BILLING_ACCOUNT")
|
|
var project compute.Project
|
|
projectID := "terrafom-test-" + acctest.RandString(10)
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckComputeProjectMetadataDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccComputeProject_basic0_metadata(projectID, pname, org, billingId),
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckComputeProjectExists(
|
|
"google_compute_project_metadata.fizzbuzz", projectID, &project),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "banana", "orange"),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "sofa", "darwinism"),
|
|
testAccCheckComputeProjectMetadataSize(projectID, 2),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
// Add three key value pairs, then replace one and modify a second
|
|
func TestAccComputeProjectMetadata_modify_1(t *testing.T) {
|
|
skipIfEnvNotSet(t,
|
|
[]string{
|
|
"GOOGLE_ORG",
|
|
"GOOGLE_BILLING_ACCOUNT",
|
|
}...,
|
|
)
|
|
|
|
billingId := os.Getenv("GOOGLE_BILLING_ACCOUNT")
|
|
var project compute.Project
|
|
projectID := "terrafom-test-" + acctest.RandString(10)
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckComputeProjectMetadataDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccComputeProject_modify0_metadata(projectID, pname, org, billingId),
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckComputeProjectExists(
|
|
"google_compute_project_metadata.fizzbuzz", projectID, &project),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "paper", "pen"),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "genghis_khan", "french bread"),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "happy", "smiling"),
|
|
testAccCheckComputeProjectMetadataSize(projectID, 3),
|
|
),
|
|
},
|
|
|
|
resource.TestStep{
|
|
Config: testAccComputeProject_modify1_metadata(projectID, pname, org, billingId),
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckComputeProjectExists(
|
|
"google_compute_project_metadata.fizzbuzz", projectID, &project),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "paper", "pen"),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "paris", "french bread"),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "happy", "laughing"),
|
|
testAccCheckComputeProjectMetadataSize(projectID, 3),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
// Add two key value pairs, and replace both
|
|
func TestAccComputeProjectMetadata_modify_2(t *testing.T) {
|
|
skipIfEnvNotSet(t,
|
|
[]string{
|
|
"GOOGLE_ORG",
|
|
"GOOGLE_BILLING_ACCOUNT",
|
|
}...,
|
|
)
|
|
|
|
billingId := os.Getenv("GOOGLE_BILLING_ACCOUNT")
|
|
var project compute.Project
|
|
projectID := "terraform-test-" + acctest.RandString(10)
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckComputeProjectMetadataDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccComputeProject_basic0_metadata(projectID, pname, org, billingId),
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckComputeProjectExists(
|
|
"google_compute_project_metadata.fizzbuzz", projectID, &project),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "banana", "orange"),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "sofa", "darwinism"),
|
|
testAccCheckComputeProjectMetadataSize(projectID, 2),
|
|
),
|
|
},
|
|
|
|
resource.TestStep{
|
|
Config: testAccComputeProject_basic1_metadata(projectID, pname, org, billingId),
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckComputeProjectExists(
|
|
"google_compute_project_metadata.fizzbuzz", projectID, &project),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "kiwi", "papaya"),
|
|
testAccCheckComputeProjectMetadataContains(projectID, "finches", "darwinism"),
|
|
testAccCheckComputeProjectMetadataSize(projectID, 2),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func testAccCheckComputeProjectMetadataDestroy(s *terraform.State) error {
|
|
config := testAccProvider.Meta().(*Config)
|
|
|
|
for _, rs := range s.RootModule().Resources {
|
|
if rs.Type != "google_compute_project_metadata" {
|
|
continue
|
|
}
|
|
|
|
project, err := config.clientCompute.Projects.Get(rs.Primary.ID).Do()
|
|
if err == nil && len(project.CommonInstanceMetadata.Items) > 0 {
|
|
return fmt.Errorf("Error, metadata items still exist in %s", rs.Primary.ID)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func testAccCheckComputeProjectExists(n, projectID string, project *compute.Project) 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.Projects.Get(projectID).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if "common_metadata" != rs.Primary.ID {
|
|
return fmt.Errorf("Common metadata not found, found %s", rs.Primary.ID)
|
|
}
|
|
|
|
*project = *found
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func testAccCheckComputeProjectMetadataContains(projectID, key, value string) resource.TestCheckFunc {
|
|
return func(s *terraform.State) error {
|
|
config := testAccProvider.Meta().(*Config)
|
|
project, err := config.clientCompute.Projects.Get(projectID).Do()
|
|
if err != nil {
|
|
return fmt.Errorf("Error, failed to load project service for %s: %s", config.Project, err)
|
|
}
|
|
|
|
for _, kv := range project.CommonInstanceMetadata.Items {
|
|
if kv.Key == key {
|
|
if kv.Value != nil && *kv.Value == value {
|
|
return nil
|
|
} else {
|
|
return fmt.Errorf("Error, key value mismatch, wanted (%s, %s), got (%s, %s)",
|
|
key, value, kv.Key, *kv.Value)
|
|
}
|
|
}
|
|
}
|
|
|
|
return fmt.Errorf("Error, key %s not present in %s", key, project.SelfLink)
|
|
}
|
|
}
|
|
|
|
func testAccCheckComputeProjectMetadataSize(projectID string, size int) resource.TestCheckFunc {
|
|
return func(s *terraform.State) error {
|
|
config := testAccProvider.Meta().(*Config)
|
|
project, err := config.clientCompute.Projects.Get(projectID).Do()
|
|
if err != nil {
|
|
return fmt.Errorf("Error, failed to load project service for %s: %s", config.Project, err)
|
|
}
|
|
|
|
if size > len(project.CommonInstanceMetadata.Items) {
|
|
return fmt.Errorf("Error, expected at least %d metadata items, got %d", size,
|
|
len(project.CommonInstanceMetadata.Items))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func testAccComputeProject_basic0_metadata(projectID, name, org, billing string) string {
|
|
return fmt.Sprintf(`
|
|
resource "google_project" "project" {
|
|
project_id = "%s"
|
|
name = "%s"
|
|
org_id = "%s"
|
|
billing_account = "%s"
|
|
}
|
|
|
|
resource "google_project_services" "services" {
|
|
project = "${google_project.project.project_id}"
|
|
services = ["compute-component.googleapis.com"]
|
|
}
|
|
|
|
resource "google_compute_project_metadata" "fizzbuzz" {
|
|
project = "${google_project.project.project_id}"
|
|
metadata {
|
|
banana = "orange"
|
|
sofa = "darwinism"
|
|
}
|
|
depends_on = ["google_project_services.services"]
|
|
}`, projectID, name, org, billing)
|
|
}
|
|
|
|
func testAccComputeProject_basic1_metadata(projectID, name, org, billing string) string {
|
|
return fmt.Sprintf(`
|
|
resource "google_project" "project" {
|
|
project_id = "%s"
|
|
name = "%s"
|
|
org_id = "%s"
|
|
billing_account = "%s"
|
|
}
|
|
|
|
resource "google_project_services" "services" {
|
|
project = "${google_project.project.project_id}"
|
|
services = ["compute-component.googleapis.com"]
|
|
}
|
|
|
|
resource "google_compute_project_metadata" "fizzbuzz" {
|
|
project = "${google_project.project.project_id}"
|
|
metadata {
|
|
kiwi = "papaya"
|
|
finches = "darwinism"
|
|
}
|
|
depends_on = ["google_project_services.services"]
|
|
}`, projectID, name, org, billing)
|
|
}
|
|
|
|
func testAccComputeProject_modify0_metadata(projectID, name, org, billing string) string {
|
|
return fmt.Sprintf(`
|
|
resource "google_project" "project" {
|
|
project_id = "%s"
|
|
name = "%s"
|
|
org_id = "%s"
|
|
billing_account = "%s"
|
|
}
|
|
|
|
resource "google_project_services" "services" {
|
|
project = "${google_project.project.project_id}"
|
|
services = ["compute-component.googleapis.com"]
|
|
}
|
|
|
|
resource "google_compute_project_metadata" "fizzbuzz" {
|
|
project = "${google_project.project.project_id}"
|
|
metadata {
|
|
paper = "pen"
|
|
genghis_khan = "french bread"
|
|
happy = "smiling"
|
|
}
|
|
depends_on = ["google_project_services.services"]
|
|
}`, projectID, name, org, billing)
|
|
}
|
|
|
|
func testAccComputeProject_modify1_metadata(projectID, name, org, billing string) string {
|
|
return fmt.Sprintf(`
|
|
resource "google_project" "project" {
|
|
project_id = "%s"
|
|
name = "%s"
|
|
org_id = "%s"
|
|
billing_account = "%s"
|
|
}
|
|
|
|
resource "google_project_services" "services" {
|
|
project = "${google_project.project.project_id}"
|
|
services = ["compute-component.googleapis.com"]
|
|
}
|
|
|
|
resource "google_compute_project_metadata" "fizzbuzz" {
|
|
project = "${google_project.project.project_id}"
|
|
metadata {
|
|
paper = "pen"
|
|
paris = "french bread"
|
|
happy = "laughing"
|
|
}
|
|
depends_on = ["google_project_services.services"]
|
|
}`, projectID, name, org, billing)
|
|
}
|