provider/google: add location to storage tests.

Add location to storage tests that need it, which fixes the failing
TestAccStorageStorageClass test.
This commit is contained in:
Paddy 2017-03-03 15:51:36 -08:00
parent 09a2886546
commit a318cd9d61
1 changed files with 13 additions and 6 deletions

View File

@ -68,7 +68,7 @@ func TestAccStorageStorageClass(t *testing.T) {
CheckDestroy: testAccGoogleStorageDestroy, CheckDestroy: testAccGoogleStorageDestroy,
Steps: []resource.TestStep{ Steps: []resource.TestStep{
{ {
Config: testGoogleStorageBucketsReaderStorageClass(bucketName, "MULTI_REGIONAL"), Config: testGoogleStorageBucketsReaderStorageClass(bucketName, "MULTI_REGIONAL", ""),
Check: resource.ComposeTestCheckFunc( Check: resource.ComposeTestCheckFunc(
testAccCheckCloudStorageBucketExists( testAccCheckCloudStorageBucketExists(
"google_storage_bucket.bucket", bucketName), "google_storage_bucket.bucket", bucketName),
@ -77,7 +77,7 @@ func TestAccStorageStorageClass(t *testing.T) {
), ),
}, },
{ {
Config: testGoogleStorageBucketsReaderStorageClass(bucketName, "NEARLINE"), Config: testGoogleStorageBucketsReaderStorageClass(bucketName, "NEARLINE", ""),
Check: resource.ComposeTestCheckFunc( Check: resource.ComposeTestCheckFunc(
testAccCheckCloudStorageBucketExists( testAccCheckCloudStorageBucketExists(
"google_storage_bucket.bucket", bucketName), "google_storage_bucket.bucket", bucketName),
@ -86,12 +86,14 @@ func TestAccStorageStorageClass(t *testing.T) {
), ),
}, },
{ {
Config: testGoogleStorageBucketsReaderStorageClass(bucketName, "REGIONAL"), Config: testGoogleStorageBucketsReaderStorageClass(bucketName, "REGIONAL", "us-central1"),
Check: resource.ComposeTestCheckFunc( Check: resource.ComposeTestCheckFunc(
testAccCheckCloudStorageBucketExists( testAccCheckCloudStorageBucketExists(
"google_storage_bucket.bucket", bucketName), "google_storage_bucket.bucket", bucketName),
resource.TestCheckResourceAttr( resource.TestCheckResourceAttr(
"google_storage_bucket.bucket", "storage_class", "REGIONAL"), "google_storage_bucket.bucket", "storage_class", "REGIONAL"),
resource.TestCheckResourceAttr(
"google_storage_bucket.bucket", "location", "us-central1"),
), ),
}, },
}, },
@ -266,11 +268,16 @@ resource "google_storage_bucket" "bucket" {
`, bucketName) `, bucketName)
} }
func testGoogleStorageBucketsReaderStorageClass(bucketName string, storageClass string) string { func testGoogleStorageBucketsReaderStorageClass(bucketName, storageClass, location string) string {
var locationBlock string
if location != "" {
locationBlock = fmt.Sprintf(`
location = "%s"`, location)
}
return fmt.Sprintf(` return fmt.Sprintf(`
resource "google_storage_bucket" "bucket" { resource "google_storage_bucket" "bucket" {
name = "%s" name = "%s"
storage_class = "%s" storage_class = "%s"%s
} }
`, bucketName, storageClass) `, bucketName, storageClass, locationBlock)
} }