From 74813821ec38d7c18b0efed9aad37c1b7cf1aadd Mon Sep 17 00:00:00 2001
From: James Bardin <j.bardin@gmail.com>
Date: Thu, 7 Jul 2016 16:18:00 -0400
Subject: [PATCH] Add remote state init test

Verify that a remote state file is correctly initialized in the same
manner as used by the `terraform remote config`
---
 state/remote/remote_test.go | 72 +++++++++++++++++++------------------
 1 file changed, 37 insertions(+), 35 deletions(-)

diff --git a/state/remote/remote_test.go b/state/remote/remote_test.go
index 04f8757477ca..db3c795c80dd 100644
--- a/state/remote/remote_test.go
+++ b/state/remote/remote_test.go
@@ -2,7 +2,6 @@ package remote
 
 import (
 	"bytes"
-	"fmt"
 	"io/ioutil"
 	"os"
 	"testing"
@@ -74,18 +73,7 @@ func TestRemoteClient_stateInit(t *testing.T) {
 	// remote state.
 	localStateFile.Close()
 	os.Remove(localStateFile.Name())
-	//defer os.Remove(localStateFile.Name())
-	fmt.Println("LOCAL:", localStateFile.Name())
-
-	local := &state.LocalState{
-		Path: localStateFile.Name(),
-	}
-	if err := local.RefreshState(); err != nil {
-		t.Fatal(err)
-	}
-	localState := local.State()
-
-	fmt.Println("localState.Empty():", localState.Empty())
+	defer os.Remove(localStateFile.Name())
 
 	remoteStateFile, err := ioutil.TempFile("", "tf")
 	if err != nil {
@@ -93,41 +81,55 @@ func TestRemoteClient_stateInit(t *testing.T) {
 	}
 	remoteStateFile.Close()
 	os.Remove(remoteStateFile.Name())
-	//defer os.Remove(remoteStateFile.Name()
-	fmt.Println("LOCAL:", localStateFile.Name())
-	fmt.Println("REMOTE:", remoteStateFile.Name())
+	defer os.Remove(remoteStateFile.Name())
 
-	remoteClient := &FileClient{
-		Path: remoteStateFile.Name(),
+	// Now we need an empty state to initialize the state files.
+	newState := terraform.NewState()
+	newState.Remote = &terraform.RemoteState{
+		Type:   "_local",
+		Config: map[string]string{"path": remoteStateFile.Name()},
 	}
 
-	durable := &State{
-		Client: remoteClient,
+	remoteClient := &FileClient{
+		Path: remoteStateFile.Name(),
 	}
 
 	cache := &state.CacheState{
-		Cache:   local,
-		Durable: durable,
+		Cache: &state.LocalState{
+			Path: localStateFile.Name(),
+		},
+		Durable: &State{
+			Client: remoteClient,
+		},
 	}
 
-	if err := cache.RefreshState(); err != nil {
+	// This will write the local state file, and set the state field in the CacheState
+	err = cache.WriteState(newState)
+	if err != nil {
 		t.Fatal(err)
 	}
 
-	switch cache.RefreshResult() {
+	// This will persist the local state we just wrote to the remote state file
+	err = cache.PersistState()
+	if err != nil {
+		t.Fatal(err)
+	}
 
-	// we should be "refreshing" the remote state to initialize it
-	case state.CacheRefreshLocalNewer:
-		// Write our local state out to the durable storage to start.
-		if err := cache.WriteState(localState); err != nil {
-			t.Fatal("Error preparing remote state:", err)
-		}
-		if err := cache.PersistState(); err != nil {
-			t.Fatal("Error preparing remote state:", err)
-		}
-	default:
+	// now compare the two state files just to be sure
+	localData, err := ioutil.ReadFile(localStateFile.Name())
+	if err != nil {
+		t.Fatal(err)
+	}
 
-		t.Fatal("unexpected refresh result:", cache.RefreshResult())
+	remoteData, err := ioutil.ReadFile(remoteStateFile.Name())
+	if err != nil {
+		t.Fatal(err)
 	}
 
+	if !bytes.Equal(localData, remoteData) {
+		t.Log("state files don't match")
+		t.Log("Local:\n", string(localData))
+		t.Log("Remote:\n", string(remoteData))
+		t.Fatal("failed to initialize remote state")
+	}
 }