2017-10-11 13:47:35 +00:00
|
|
|
package forceleave
|
2013-12-31 21:06:33 +00:00
|
|
|
|
|
|
|
import (
|
2017-03-23 22:27:16 +00:00
|
|
|
"strings"
|
|
|
|
"testing"
|
|
|
|
|
2013-12-31 21:06:33 +00:00
|
|
|
"github.com/hashicorp/serf/serf"
|
|
|
|
"github.com/mitchellh/cli"
|
2021-10-26 20:08:55 +00:00
|
|
|
|
|
|
|
"github.com/hashicorp/consul/agent"
|
|
|
|
"github.com/hashicorp/consul/sdk/testutil/retry"
|
2013-12-31 21:06:33 +00:00
|
|
|
)
|
|
|
|
|
2017-10-11 13:47:35 +00:00
|
|
|
func TestForceLeaveCommand_noTabs(t *testing.T) {
|
2017-10-17 14:07:48 +00:00
|
|
|
t.Parallel()
|
2017-10-11 13:47:35 +00:00
|
|
|
if strings.ContainsRune(New(nil).Help(), '\t') {
|
2017-10-17 13:44:20 +00:00
|
|
|
t.Fatal("help has tabs")
|
2017-10-11 13:47:35 +00:00
|
|
|
}
|
2013-12-31 21:06:33 +00:00
|
|
|
}
|
|
|
|
|
2021-10-28 21:44:38 +00:00
|
|
|
// TODO(partitions): split this test and verify it works in partitions
|
2017-10-17 12:04:51 +00:00
|
|
|
func TestForceLeaveCommand(t *testing.T) {
|
2020-12-07 18:42:55 +00:00
|
|
|
if testing.Short() {
|
|
|
|
t.Skip("too slow for testing.Short")
|
|
|
|
}
|
|
|
|
|
2017-10-17 14:07:48 +00:00
|
|
|
t.Parallel()
|
2020-03-31 19:59:56 +00:00
|
|
|
a1 := agent.NewTestAgent(t, ``)
|
|
|
|
a2 := agent.NewTestAgent(t, ``)
|
2013-12-31 21:06:33 +00:00
|
|
|
defer a1.Shutdown()
|
|
|
|
defer a2.Shutdown()
|
|
|
|
|
2021-10-26 20:08:55 +00:00
|
|
|
_, err := a2.JoinLAN([]string{a1.Config.SerfBindAddrLAN.String()}, nil)
|
2013-12-31 21:06:33 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Forcibly shutdown a2 so that it appears "failed" in a1
|
|
|
|
a2.Shutdown()
|
|
|
|
|
2017-10-11 13:47:35 +00:00
|
|
|
ui := cli.NewMockUi()
|
|
|
|
c := New(ui)
|
2013-12-31 21:06:33 +00:00
|
|
|
args := []string{
|
2017-05-22 11:59:36 +00:00
|
|
|
"-http-addr=" + a1.HTTPAddr(),
|
|
|
|
a2.Config.NodeName,
|
2013-12-31 21:06:33 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
code := c.Run(args)
|
|
|
|
if code != 0 {
|
|
|
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String())
|
|
|
|
}
|
|
|
|
|
2021-10-26 20:08:55 +00:00
|
|
|
m := a1.LANMembersInAgentPartition()
|
2013-12-31 21:06:33 +00:00
|
|
|
if len(m) != 2 {
|
|
|
|
t.Fatalf("should have 2 members: %#v", m)
|
|
|
|
}
|
2017-05-04 22:52:53 +00:00
|
|
|
retry.Run(t, func(r *retry.R) {
|
2021-10-26 20:08:55 +00:00
|
|
|
m = a1.LANMembersInAgentPartition()
|
2017-04-29 16:34:02 +00:00
|
|
|
if got, want := m[1].Status, serf.StatusLeft; got != want {
|
|
|
|
r.Fatalf("got status %q want %q", got, want)
|
|
|
|
}
|
|
|
|
})
|
2013-12-31 21:06:33 +00:00
|
|
|
}
|
|
|
|
|
2021-10-28 21:44:38 +00:00
|
|
|
// TODO(partitions): split this test and verify it works in partitions
|
2019-12-02 19:06:15 +00:00
|
|
|
func TestForceLeaveCommand_NoNodeWithName(t *testing.T) {
|
2020-12-07 18:42:55 +00:00
|
|
|
if testing.Short() {
|
|
|
|
t.Skip("too slow for testing.Short")
|
|
|
|
}
|
|
|
|
|
2019-12-02 19:06:15 +00:00
|
|
|
t.Parallel()
|
2020-03-31 19:59:56 +00:00
|
|
|
a1 := agent.NewTestAgent(t, ``)
|
2019-12-02 19:06:15 +00:00
|
|
|
defer a1.Shutdown()
|
|
|
|
|
|
|
|
ui := cli.NewMockUi()
|
|
|
|
c := New(ui)
|
|
|
|
args := []string{
|
|
|
|
"-http-addr=" + a1.HTTPAddr(),
|
|
|
|
"garbage-name",
|
|
|
|
}
|
|
|
|
|
|
|
|
code := c.Run(args)
|
|
|
|
if code != 1 {
|
|
|
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-10-04 21:10:02 +00:00
|
|
|
func TestForceLeaveCommand_prune(t *testing.T) {
|
2020-12-07 18:42:55 +00:00
|
|
|
if testing.Short() {
|
|
|
|
t.Skip("too slow for testing.Short")
|
|
|
|
}
|
|
|
|
|
2019-10-04 21:10:02 +00:00
|
|
|
t.Parallel()
|
2020-03-31 20:12:33 +00:00
|
|
|
a1 := agent.StartTestAgent(t, agent.TestAgent{Name: "Agent1"})
|
2019-10-04 21:10:02 +00:00
|
|
|
defer a1.Shutdown()
|
2020-03-31 20:12:33 +00:00
|
|
|
a2 := agent.StartTestAgent(t, agent.TestAgent{Name: "Agent2"})
|
2019-10-04 21:10:02 +00:00
|
|
|
defer a2.Shutdown()
|
|
|
|
|
2021-10-26 20:08:55 +00:00
|
|
|
_, err := a2.JoinLAN([]string{a1.Config.SerfBindAddrLAN.String()}, nil)
|
2019-10-04 21:10:02 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Forcibly shutdown a2 so that it appears "failed" in a1
|
|
|
|
a2.Shutdown()
|
|
|
|
|
|
|
|
ui := cli.NewMockUi()
|
|
|
|
c := New(ui)
|
|
|
|
args := []string{
|
|
|
|
"-http-addr=" + a1.HTTPAddr(),
|
|
|
|
"-prune",
|
|
|
|
a2.Config.NodeName,
|
|
|
|
}
|
|
|
|
|
|
|
|
code := c.Run(args)
|
|
|
|
if code != 0 {
|
|
|
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String())
|
|
|
|
}
|
|
|
|
retry.Run(t, func(r *retry.R) {
|
2021-10-26 20:08:55 +00:00
|
|
|
m := len(a1.LANMembersInAgentPartition())
|
2019-10-04 21:10:02 +00:00
|
|
|
if m != 1 {
|
|
|
|
r.Fatalf("should have 1 members, got %#v", m)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2017-10-17 12:04:51 +00:00
|
|
|
func TestForceLeaveCommand_noAddrs(t *testing.T) {
|
2017-10-17 14:07:48 +00:00
|
|
|
t.Parallel()
|
2017-05-22 23:26:30 +00:00
|
|
|
ui := cli.NewMockUi()
|
2017-10-11 13:47:35 +00:00
|
|
|
c := New(ui)
|
2017-02-07 01:50:51 +00:00
|
|
|
args := []string{"-http-addr=foo"}
|
2013-12-31 21:06:33 +00:00
|
|
|
|
|
|
|
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())
|
|
|
|
}
|
|
|
|
}
|