271 lines
8.3 KiB
Go
271 lines
8.3 KiB
Go
package azure
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/Azure/azure-sdk-for-go/storage"
|
|
"github.com/hashicorp/terraform/backend"
|
|
"github.com/hashicorp/terraform/helper/acctest"
|
|
"github.com/hashicorp/terraform/state/remote"
|
|
)
|
|
|
|
func TestRemoteClient_impl(t *testing.T) {
|
|
var _ remote.Client = new(RemoteClient)
|
|
var _ remote.ClientLocker = new(RemoteClient)
|
|
}
|
|
|
|
func TestRemoteClientAccessKeyBasic(t *testing.T) {
|
|
testAccAzureBackend(t)
|
|
rs := acctest.RandString(4)
|
|
res := testResourceNames(rs, "testState")
|
|
armClient := buildTestClient(t, res)
|
|
|
|
ctx := context.TODO()
|
|
err := armClient.buildTestResources(ctx, &res)
|
|
if err != nil {
|
|
armClient.destroyTestResources(ctx, res)
|
|
t.Fatalf("Error creating Test Resources: %q", err)
|
|
}
|
|
defer armClient.destroyTestResources(ctx, res)
|
|
|
|
b := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
|
"storage_account_name": res.storageAccountName,
|
|
"container_name": res.storageContainerName,
|
|
"key": res.storageKeyName,
|
|
"access_key": res.storageAccountAccessKey,
|
|
})).(*Backend)
|
|
|
|
state, err := b.StateMgr(backend.DefaultStateName)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
remote.TestClient(t, state.(*remote.State).Client)
|
|
}
|
|
|
|
func TestRemoteClientManagedServiceIdentityBasic(t *testing.T) {
|
|
testAccAzureBackendRunningInAzure(t)
|
|
rs := acctest.RandString(4)
|
|
res := testResourceNames(rs, "testState")
|
|
armClient := buildTestClient(t, res)
|
|
|
|
ctx := context.TODO()
|
|
err := armClient.buildTestResources(ctx, &res)
|
|
if err != nil {
|
|
armClient.destroyTestResources(ctx, res)
|
|
t.Fatalf("Error creating Test Resources: %q", err)
|
|
}
|
|
defer armClient.destroyTestResources(ctx, res)
|
|
|
|
b := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
|
"storage_account_name": res.storageAccountName,
|
|
"container_name": res.storageContainerName,
|
|
"key": res.storageKeyName,
|
|
"resource_group_name": res.resourceGroup,
|
|
"use_msi": true,
|
|
"arm_subscription_id": os.Getenv("ARM_SUBSCRIPTION_ID"),
|
|
"arm_tenant_id": os.Getenv("ARM_TENANT_ID"),
|
|
"environment": os.Getenv("ARM_ENVIRONMENT"),
|
|
})).(*Backend)
|
|
|
|
state, err := b.StateMgr(backend.DefaultStateName)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
remote.TestClient(t, state.(*remote.State).Client)
|
|
}
|
|
|
|
func TestRemoteClientServicePrincipalBasic(t *testing.T) {
|
|
testAccAzureBackend(t)
|
|
rs := acctest.RandString(4)
|
|
res := testResourceNames(rs, "testState")
|
|
armClient := buildTestClient(t, res)
|
|
|
|
ctx := context.TODO()
|
|
err := armClient.buildTestResources(ctx, &res)
|
|
if err != nil {
|
|
armClient.destroyTestResources(ctx, res)
|
|
t.Fatalf("Error creating Test Resources: %q", err)
|
|
}
|
|
defer armClient.destroyTestResources(ctx, res)
|
|
|
|
b := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
|
"storage_account_name": res.storageAccountName,
|
|
"container_name": res.storageContainerName,
|
|
"key": res.storageKeyName,
|
|
"resource_group_name": res.resourceGroup,
|
|
"arm_subscription_id": os.Getenv("ARM_SUBSCRIPTION_ID"),
|
|
"arm_tenant_id": os.Getenv("ARM_TENANT_ID"),
|
|
"arm_client_id": os.Getenv("ARM_CLIENT_ID"),
|
|
"arm_client_secret": os.Getenv("ARM_CLIENT_SECRET"),
|
|
"environment": os.Getenv("ARM_ENVIRONMENT"),
|
|
})).(*Backend)
|
|
|
|
state, err := b.StateMgr(backend.DefaultStateName)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
remote.TestClient(t, state.(*remote.State).Client)
|
|
}
|
|
|
|
func TestRemoteClientAccessKeyLocks(t *testing.T) {
|
|
testAccAzureBackend(t)
|
|
rs := acctest.RandString(4)
|
|
res := testResourceNames(rs, "testState")
|
|
armClient := buildTestClient(t, res)
|
|
|
|
ctx := context.TODO()
|
|
err := armClient.buildTestResources(ctx, &res)
|
|
if err != nil {
|
|
armClient.destroyTestResources(ctx, res)
|
|
t.Fatalf("Error creating Test Resources: %q", err)
|
|
}
|
|
defer armClient.destroyTestResources(ctx, res)
|
|
|
|
b1 := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
|
"storage_account_name": res.storageAccountName,
|
|
"container_name": res.storageContainerName,
|
|
"key": res.storageKeyName,
|
|
"access_key": res.storageAccountAccessKey,
|
|
})).(*Backend)
|
|
|
|
b2 := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
|
"storage_account_name": res.storageAccountName,
|
|
"container_name": res.storageContainerName,
|
|
"key": res.storageKeyName,
|
|
"access_key": res.storageAccountAccessKey,
|
|
})).(*Backend)
|
|
|
|
s1, err := b1.StateMgr(backend.DefaultStateName)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
s2, err := b2.StateMgr(backend.DefaultStateName)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
remote.TestRemoteLocks(t, s1.(*remote.State).Client, s2.(*remote.State).Client)
|
|
}
|
|
|
|
func TestRemoteClientServicePrincipalLocks(t *testing.T) {
|
|
testAccAzureBackend(t)
|
|
rs := acctest.RandString(4)
|
|
res := testResourceNames(rs, "testState")
|
|
armClient := buildTestClient(t, res)
|
|
|
|
ctx := context.TODO()
|
|
err := armClient.buildTestResources(ctx, &res)
|
|
if err != nil {
|
|
armClient.destroyTestResources(ctx, res)
|
|
t.Fatalf("Error creating Test Resources: %q", err)
|
|
}
|
|
defer armClient.destroyTestResources(ctx, res)
|
|
|
|
b1 := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
|
"storage_account_name": res.storageAccountName,
|
|
"container_name": res.storageContainerName,
|
|
"key": res.storageKeyName,
|
|
"resource_group_name": res.resourceGroup,
|
|
"arm_subscription_id": os.Getenv("ARM_SUBSCRIPTION_ID"),
|
|
"arm_tenant_id": os.Getenv("ARM_TENANT_ID"),
|
|
"arm_client_id": os.Getenv("ARM_CLIENT_ID"),
|
|
"arm_client_secret": os.Getenv("ARM_CLIENT_SECRET"),
|
|
"environment": os.Getenv("ARM_ENVIRONMENT"),
|
|
})).(*Backend)
|
|
|
|
b2 := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
|
"storage_account_name": res.storageAccountName,
|
|
"container_name": res.storageContainerName,
|
|
"key": res.storageKeyName,
|
|
"resource_group_name": res.resourceGroup,
|
|
"arm_subscription_id": os.Getenv("ARM_SUBSCRIPTION_ID"),
|
|
"arm_tenant_id": os.Getenv("ARM_TENANT_ID"),
|
|
"arm_client_id": os.Getenv("ARM_CLIENT_ID"),
|
|
"arm_client_secret": os.Getenv("ARM_CLIENT_SECRET"),
|
|
"environment": os.Getenv("ARM_ENVIRONMENT"),
|
|
})).(*Backend)
|
|
|
|
s1, err := b1.StateMgr(backend.DefaultStateName)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
s2, err := b2.StateMgr(backend.DefaultStateName)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
remote.TestRemoteLocks(t, s1.(*remote.State).Client, s2.(*remote.State).Client)
|
|
}
|
|
|
|
func TestPutMaintainsMetaData(t *testing.T) {
|
|
testAccAzureBackend(t)
|
|
rs := acctest.RandString(4)
|
|
res := testResourceNames(rs, "testState")
|
|
armClient := buildTestClient(t, res)
|
|
|
|
ctx := context.TODO()
|
|
err := armClient.buildTestResources(ctx, &res)
|
|
if err != nil {
|
|
armClient.destroyTestResources(ctx, res)
|
|
t.Fatalf("Error creating Test Resources: %q", err)
|
|
}
|
|
defer armClient.destroyTestResources(ctx, res)
|
|
|
|
headerName := "acceptancetest"
|
|
expectedValue := "f3b56bad-33ad-4b93-a600-7a66e9cbd1eb"
|
|
|
|
client, err := armClient.getBlobClient(ctx)
|
|
if err != nil {
|
|
t.Fatalf("Error building Blob Client: %+v", err)
|
|
}
|
|
containerReference := client.GetContainerReference(res.storageContainerName)
|
|
blobReference := containerReference.GetBlobReference(res.storageKeyName)
|
|
|
|
err = blobReference.CreateBlockBlob(&storage.PutBlobOptions{})
|
|
if err != nil {
|
|
t.Fatalf("Error Creating Block Blob: %+v", err)
|
|
}
|
|
|
|
err = blobReference.GetMetadata(&storage.GetBlobMetadataOptions{})
|
|
if err != nil {
|
|
t.Fatalf("Error loading MetaData: %+v", err)
|
|
}
|
|
|
|
blobReference.Metadata[headerName] = expectedValue
|
|
err = blobReference.SetMetadata(&storage.SetBlobMetadataOptions{})
|
|
if err != nil {
|
|
t.Fatalf("Error setting MetaData: %+v", err)
|
|
}
|
|
|
|
// update the metadata using the Backend
|
|
remoteClient := RemoteClient{
|
|
keyName: res.storageKeyName,
|
|
containerName: res.storageContainerName,
|
|
|
|
blobClient: *client,
|
|
}
|
|
|
|
bytes := []byte(acctest.RandString(20))
|
|
err = remoteClient.Put(bytes)
|
|
if err != nil {
|
|
t.Fatalf("Error putting data: %+v", err)
|
|
}
|
|
|
|
// Verify it still exists
|
|
err = blobReference.GetMetadata(&storage.GetBlobMetadataOptions{})
|
|
if err != nil {
|
|
t.Fatalf("Error loading MetaData: %+v", err)
|
|
}
|
|
|
|
if blobReference.Metadata[headerName] != expectedValue {
|
|
t.Fatalf("%q was not set to %q in the MetaData: %+v", headerName, expectedValue, blobReference.Metadata)
|
|
}
|
|
}
|