81 lines
1.6 KiB
Go
81 lines
1.6 KiB
Go
![]() |
package command
|
||
|
|
||
|
import (
|
||
|
"io/ioutil"
|
||
|
"os"
|
||
|
"reflect"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/hashicorp/terraform/terraform"
|
||
|
"github.com/mitchellh/cli"
|
||
|
)
|
||
|
|
||
|
func TestDiff_noState(t *testing.T) {
|
||
|
p := testProvider()
|
||
|
ui := new(cli.MockUi)
|
||
|
c := &DiffCommand{
|
||
|
TFConfig: testTFConfig(p),
|
||
|
Ui: ui,
|
||
|
}
|
||
|
|
||
|
args := []string{
|
||
|
testFixturePath("diff"),
|
||
|
}
|
||
|
if code := c.Run(args); code != 0 {
|
||
|
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
|
||
|
}
|
||
|
|
||
|
// Verify that the provider was called with the existing state
|
||
|
expectedState := &terraform.ResourceState{
|
||
|
Type: "test_instance",
|
||
|
}
|
||
|
if !reflect.DeepEqual(p.DiffState, expectedState) {
|
||
|
t.Fatalf("bad: %#v", p.DiffState)
|
||
|
}
|
||
|
}
|
||
|
func TestDiff_state(t *testing.T) {
|
||
|
// Write out some prior state
|
||
|
tf, err := ioutil.TempFile("", "tf")
|
||
|
if err != nil {
|
||
|
t.Fatalf("err: %s", err)
|
||
|
}
|
||
|
statePath := tf.Name()
|
||
|
defer os.Remove(tf.Name())
|
||
|
|
||
|
originalState := &terraform.State{
|
||
|
Resources: map[string]*terraform.ResourceState{
|
||
|
"test_instance.foo": &terraform.ResourceState{
|
||
|
ID: "bar",
|
||
|
Type: "test_instance",
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
err = terraform.WriteState(originalState, tf)
|
||
|
tf.Close()
|
||
|
if err != nil {
|
||
|
t.Fatalf("err: %s", err)
|
||
|
}
|
||
|
|
||
|
p := testProvider()
|
||
|
ui := new(cli.MockUi)
|
||
|
c := &DiffCommand{
|
||
|
TFConfig: testTFConfig(p),
|
||
|
Ui: ui,
|
||
|
}
|
||
|
|
||
|
args := []string{
|
||
|
"-state", statePath,
|
||
|
testFixturePath("diff"),
|
||
|
}
|
||
|
if code := c.Run(args); code != 0 {
|
||
|
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
|
||
|
}
|
||
|
|
||
|
// Verify that the provider was called with the existing state
|
||
|
expectedState := originalState.Resources["test_instance.foo"]
|
||
|
if !reflect.DeepEqual(p.DiffState, expectedState) {
|
||
|
t.Fatalf("bad: %#v", p.DiffState)
|
||
|
}
|
||
|
}
|