package command import ( "os" "strings" "testing" "github.com/hashicorp/terraform/helper/copy" "github.com/mitchellh/cli" ) func TestGet(t *testing.T) { tmp, cwd := testCwd(t) defer testFixCwd(t, tmp, cwd) ui := new(cli.MockUi) c := &GetCommand{ Meta: Meta{ testingOverrides: metaOverridesForProvider(testProvider()), Ui: ui, dataDir: tempDir(t), }, } args := []string{ testFixturePath("get"), } if code := c.Run(args); code != 0 { t.Fatalf("bad: \n%s", ui.ErrorWriter.String()) } output := ui.OutputWriter.String() if !strings.Contains(output, "module.foo") { t.Fatalf("doesn't look like get: %s", output) } if strings.Contains(output, "(update)") { t.Fatalf("doesn't look like get: %s", output) } } func TestGet_multipleArgs(t *testing.T) { ui := new(cli.MockUi) c := &GetCommand{ Meta: Meta{ testingOverrides: metaOverridesForProvider(testProvider()), Ui: ui, dataDir: tempDir(t), }, } args := []string{ "bad", "bad", } if code := c.Run(args); code != 1 { t.Fatalf("bad: \n%s", ui.OutputWriter.String()) } } func TestGet_noArgs(t *testing.T) { td := tempDir(t) copy.CopyDir(testFixturePath("get"), td) defer os.RemoveAll(td) defer testChdir(t, td)() ui := new(cli.MockUi) c := &GetCommand{ Meta: Meta{ testingOverrides: metaOverridesForProvider(testProvider()), Ui: ui, dataDir: tempDir(t), }, } args := []string{} if code := c.Run(args); code != 0 { t.Fatalf("bad: \n%s", ui.ErrorWriter.String()) } output := ui.OutputWriter.String() if !strings.Contains(output, "module.foo") { t.Fatalf("doesn't look like get: %s", output) } if strings.Contains(output, "(update)") { t.Fatalf("doesn't look like get: %s", output) } } func TestGet_update(t *testing.T) { tmp, cwd := testCwd(t) defer testFixCwd(t, tmp, cwd) ui := new(cli.MockUi) c := &GetCommand{ Meta: Meta{ testingOverrides: metaOverridesForProvider(testProvider()), Ui: ui, dataDir: tempDir(t), }, } args := []string{ "-update", testFixturePath("get"), } if code := c.Run(args); code != 0 { t.Fatalf("bad: \n%s", ui.ErrorWriter.String()) } output := ui.OutputWriter.String() if !strings.Contains(output, `Updating source "./foo"`) { t.Fatalf("doesn't look like get: %s", output) } }