package command import ( "bytes" "io/ioutil" "os" "testing" "github.com/hashicorp/terraform/helper/copy" "github.com/mitchellh/cli" ) func TestStatePull(t *testing.T) { // Create a temporary working directory that is empty td := tempDir(t) copy.CopyDir(testFixturePath("state-pull-backend"), td) defer os.RemoveAll(td) defer testChdir(t, td)() expected, err := ioutil.ReadFile("local-state.tfstate") if err != nil { t.Fatalf("error reading state: %v", err) } p := testProvider() ui := new(cli.MockUi) c := &StatePullCommand{ Meta: Meta{ testingOverrides: metaOverridesForProvider(p), Ui: ui, }, } args := []string{} if code := c.Run(args); code != 0 { t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String()) } actual := ui.OutputWriter.Bytes() if bytes.Equal(actual, expected) { t.Fatalf("expected:\n%s\n\nto include: %q", actual, expected) } } func TestStatePull_noState(t *testing.T) { tmp, cwd := testCwd(t) defer testFixCwd(t, tmp, cwd) p := testProvider() ui := cli.NewMockUi() c := &StatePullCommand{ Meta: Meta{ testingOverrides: metaOverridesForProvider(p), Ui: ui, }, } args := []string{} if code := c.Run(args); code != 0 { t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String()) } actual := ui.OutputWriter.String() if actual != "" { t.Fatalf("bad: %s", actual) } }