mirror of https://github.com/hashicorp/consul
72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
|
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())
|
||
|
|
}
|
||
|
|
}
|