package command import ( "strings" "testing" "github.com/hashicorp/vault/version" "github.com/mitchellh/cli" ) func testVersionCommand(tb testing.TB) (*cli.MockUi, *VersionCommand) { tb.Helper() ui := cli.NewMockUi() return ui, &VersionCommand{ VersionInfo: &version.VersionInfo{}, BaseCommand: &BaseCommand{ UI: ui, }, } } func TestVersionCommand_Run(t *testing.T) { t.Parallel() t.Run("output", func(t *testing.T) { t.Parallel() ui, cmd := testVersionCommand(t) code := cmd.Run(nil) if exp := 0; code != exp { t.Errorf("expected %d to be %d", code, exp) } expected := "Vault" combined := ui.OutputWriter.String() + ui.ErrorWriter.String() if !strings.Contains(combined, expected) { t.Errorf("expected %q to equal %q", combined, expected) } }) t.Run("no_tabs", func(t *testing.T) { t.Parallel() _, cmd := testVersionCommand(t) assertNoTabs(t, cmd) }) }