package leave import ( "strings" "testing" "github.com/hashicorp/consul/agent" "github.com/mitchellh/cli" ) func TestLeaveCommand_noTabs(t *testing.T) { if strings.ContainsRune(New(nil).Help(), '\t') { t.Fatal("usage has tabs") } } func TestLeaveCommand(t *testing.T) { a := agent.NewTestAgent(t.Name(), ``) defer a.Shutdown() ui := cli.NewMockUi() c := New(ui) args := []string{"-http-addr=" + a.HTTPAddr()} code := c.Run(args) if code != 0 { t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) } if !strings.Contains(ui.OutputWriter.String(), "leave complete") { t.Fatalf("bad: %#v", ui.OutputWriter.String()) } } func TestLeaveCommand_FailOnNonFlagArgs(t *testing.T) { a := agent.NewTestAgent(t.Name(), ``) defer a.Shutdown() ui := cli.NewMockUi() c := New(ui) args := []string{"-http-addr=" + a.HTTPAddr(), "appserver1"} code := c.Run(args) if code == 0 { t.Fatalf("bad: failed to check for unexpected args") } }