package command import ( "fmt" "github.com/hashicorp/serf/serf" "github.com/hashicorp/serf/testutil" "github.com/mitchellh/cli" "strings" "testing" "time" ) func TestForceLeaveCommand_implements(t *testing.T) { var _ cli.Command = &ForceLeaveCommand{} } func TestForceLeaveCommandRun(t *testing.T) { a1 := testAgent(t) a2 := testAgent(t) defer a1.Shutdown() defer a2.Shutdown() addr := fmt.Sprintf("127.0.0.1:%d", a2.config.SerfLanPort) _, err := a1.agent.JoinLAN([]string{addr}) if err != nil { t.Fatalf("err: %s", err) } testutil.Yield() // Forcibly shutdown a2 so that it appears "failed" in a1 a2.Shutdown() time.Sleep(time.Second) ui := new(cli.MockUi) c := &ForceLeaveCommand{Ui: ui} args := []string{ "-rpc-addr=" + a1.addr, a2.config.NodeName, } code := c.Run(args) if code != 0 { t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) } m := a1.agent.LANMembers() if len(m) != 2 { t.Fatalf("should have 2 members: %#v", m) } if m[1].Status != serf.StatusLeft { t.Fatalf("should be left: %#v", m[1]) } } func TestForceLeaveCommandRun_noAddrs(t *testing.T) { ui := new(cli.MockUi) c := &ForceLeaveCommand{Ui: ui} args := []string{"-rpc-addr=foo"} code := c.Run(args) if code != 1 { t.Fatalf("bad: %d", code) } if !strings.Contains(ui.ErrorWriter.String(), "node name") { t.Fatalf("bad: %#v", ui.ErrorWriter.String()) } }