144 lines
3.0 KiB
Go
144 lines
3.0 KiB
Go
package local
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform/backend"
|
|
"github.com/hashicorp/terraform/terraform"
|
|
)
|
|
|
|
func TestLocal_impl(t *testing.T) {
|
|
var _ backend.Enhanced = new(Local)
|
|
var _ backend.Local = new(Local)
|
|
var _ backend.CLI = new(Local)
|
|
}
|
|
|
|
func checkState(t *testing.T, path, expected string) {
|
|
// Read the state
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
state, err := terraform.ReadState(f)
|
|
f.Close()
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
actual := strings.TrimSpace(state.String())
|
|
expected = strings.TrimSpace(expected)
|
|
if actual != expected {
|
|
t.Fatalf("state does not match! actual:\n%s\n\nexpected:\n%s", actual, expected)
|
|
}
|
|
}
|
|
|
|
func TestLocal_addAndRemoveStates(t *testing.T) {
|
|
defer testTmpDir(t)()
|
|
dflt := backend.DefaultStateName
|
|
expectedStates := []string{dflt}
|
|
|
|
b := &Local{}
|
|
states, current, err := b.States()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if current != dflt {
|
|
t.Fatalf("expected %q, got %q", dflt, current)
|
|
}
|
|
|
|
if !reflect.DeepEqual(states, expectedStates) {
|
|
t.Fatal("expected []string{%q}, got %q", dflt, states)
|
|
}
|
|
|
|
expectedA := "test_A"
|
|
if err := b.ChangeState(expectedA); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
states, current, err = b.States()
|
|
if current != expectedA {
|
|
t.Fatalf("expected %q, got %q", expectedA, current)
|
|
}
|
|
|
|
expectedStates = append(expectedStates, expectedA)
|
|
if !reflect.DeepEqual(states, expectedStates) {
|
|
t.Fatalf("expected %q, got %q", expectedStates, states)
|
|
}
|
|
|
|
expectedB := "test_B"
|
|
if err := b.ChangeState(expectedB); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
states, current, err = b.States()
|
|
if current != expectedB {
|
|
t.Fatalf("expected %q, got %q", expectedB, current)
|
|
}
|
|
|
|
expectedStates = append(expectedStates, expectedB)
|
|
if !reflect.DeepEqual(states, expectedStates) {
|
|
t.Fatalf("expected %q, got %q", expectedStates, states)
|
|
}
|
|
|
|
if err := b.DeleteState(expectedA); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
states, current, err = b.States()
|
|
if current != expectedB {
|
|
t.Fatalf("expected %q, got %q", dflt, current)
|
|
}
|
|
|
|
expectedStates = []string{dflt, expectedB}
|
|
if !reflect.DeepEqual(states, expectedStates) {
|
|
t.Fatalf("expected %q, got %q", expectedStates, states)
|
|
}
|
|
|
|
if err := b.DeleteState(expectedB); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
states, current, err = b.States()
|
|
if current != dflt {
|
|
t.Fatalf("expected %q, got %q", dflt, current)
|
|
}
|
|
|
|
expectedStates = []string{dflt}
|
|
if !reflect.DeepEqual(states, expectedStates) {
|
|
t.Fatalf("expected %q, got %q", expectedStates, states)
|
|
}
|
|
|
|
if err := b.DeleteState(dflt); err == nil {
|
|
t.Fatal("expected error deleting default state")
|
|
}
|
|
}
|
|
|
|
// change into a tmp dir and return a deferable func to change back and cleanup
|
|
func testTmpDir(t *testing.T) func() {
|
|
tmp, err := ioutil.TempDir("", "tf")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
old, err := os.Getwd()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := os.Chdir(tmp); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
return func() {
|
|
// ignore errors and try to clean up
|
|
os.Chdir(old)
|
|
os.RemoveAll(tmp)
|
|
}
|
|
}
|