package agent import ( "fmt" "io" "os" "os/exec" "path/filepath" "strings" "testing" "github.com/hashicorp/consul/command/cli" mcli "github.com/mitchellh/cli" "github.com/hashicorp/consul/agent" "github.com/hashicorp/consul/sdk/testutil" "github.com/hashicorp/consul/sdk/testutil/retry" "github.com/hashicorp/consul/testrpc" ) // TestConfigFail should test command line flags that lead to an immediate error. func TestConfigFail(t *testing.T) { if testing.Short() { t.Skip("too slow for testing.Short") } t.Parallel() dataDir := testutil.TempDir(t, "consul") tests := []struct { args []string out string }{ { args: []string{"agent", "-server", "-bind=10.0.0.1", "-datacenter="}, out: "==> datacenter cannot be empty\n", }, { args: []string{"agent", "-server", "-bind=10.0.0.1", "-datacenter=foo", "some-other-arg"}, out: "==> Unexpected extra arguments: [some-other-arg]\n", }, { args: []string{"agent", "-server", "-bind=10.0.0.1"}, out: "==> data_dir cannot be empty\n", }, { args: []string{"agent", "-server", "-data-dir", dataDir, "-advertise", "0.0.0.0", "-bind", "10.0.0.1"}, out: "==> Advertise address cannot be 0.0.0.0, :: or [::]\n", }, { args: []string{"agent", "-server", "-data-dir", dataDir, "-advertise", "::", "-bind", "10.0.0.1"}, out: "==> Advertise address cannot be 0.0.0.0, :: or [::]\n", }, { args: []string{"agent", "-server", "-data-dir", dataDir, "-advertise", "[::]", "-bind", "10.0.0.1"}, out: "==> Advertise address cannot be 0.0.0.0, :: or [::]\n", }, { args: []string{"agent", "-server", "-data-dir", dataDir, "-advertise-wan", "0.0.0.0", "-bind", "10.0.0.1"}, out: "==> Advertise WAN address cannot be 0.0.0.0, :: or [::]\n", }, { args: []string{"agent", "-server", "-data-dir", dataDir, "-advertise-wan", "::", "-bind", "10.0.0.1"}, out: "==> Advertise WAN address cannot be 0.0.0.0, :: or [::]\n", }, { args: []string{"agent", "-server", "-data-dir", dataDir, "-advertise-wan", "[::]", "-bind", "10.0.0.1"}, out: "==> Advertise WAN address cannot be 0.0.0.0, :: or [::]\n", }, } for _, tt := range tests { t.Run(strings.Join(tt.args, " "), func(t *testing.T) { cmd := exec.Command("consul", tt.args...) b, err := cmd.CombinedOutput() if got, want := err, "exit status 1"; got == nil || got.Error() != want { t.Fatalf("got err %q want %q", got, want) } if got, want := string(b), tt.out; got != want { t.Fatalf("got %q want %q", got, want) } }) } } func TestRetryJoin(t *testing.T) { if testing.Short() { t.Skip("too slow for testing.Short") } a := agent.NewTestAgent(t, "") defer a.Shutdown() b := agent.NewTestAgent(t, ` retry_join = ["`+a.Config.SerfBindAddrLAN.String()+`"] retry_join_wan = ["`+a.Config.SerfBindAddrWAN.String()+`"] retry_interval = "100ms" `) defer b.Shutdown() testrpc.WaitForLeader(t, a.RPC, "dc1") retry.Run(t, func(r *retry.R) { if got, want := len(a.LANMembersInAgentPartition()), 2; got != want { r.Fatalf("got %d LAN members want %d", got, want) } }) retry.Run(t, func(r *retry.R) { if got, want := len(a.WANMembers()), 2; got != want { r.Fatalf("got %d WAN members want %d", got, want) } }) } func TestRetryJoinFail(t *testing.T) { if testing.Short() { t.Skip("too slow for testing.Short") } t.Parallel() tmpDir := testutil.TempDir(t, "consul") ui := newCaptureUI() cmd := New(ui) args := []string{ "-bind", "127.0.0.1", "-data-dir", tmpDir, "-retry-join", "127.0.0.1:99", "-retry-max", "1", "-retry-interval", "10ms", } if code := cmd.Run(args); code == 0 { t.Fatalf("bad: %d", code) } } func TestRetryJoinWanFail(t *testing.T) { if testing.Short() { t.Skip("too slow for testing.Short") } t.Parallel() tmpDir := testutil.TempDir(t, "consul") ui := newCaptureUI() cmd := New(ui) args := []string{ "-server", "-bind", "127.0.0.1", "-data-dir", tmpDir, "-retry-join-wan", "127.0.0.1:99", "-retry-max-wan", "1", "-retry-interval-wan", "10ms", } if code := cmd.Run(args); code == 0 { t.Fatalf("bad: %d", code) } } func TestProtectDataDir(t *testing.T) { t.Parallel() dir := testutil.TempDir(t, "consul") if err := os.MkdirAll(filepath.Join(dir, "mdb"), 0700); err != nil { t.Fatalf("err: %v", err) } cfgDir := testutil.TempDir(t, "consul-config") cfgFilePath := filepath.Join(cfgDir, "consul.json") cfgFile, err := os.Create(cfgFilePath) if err != nil { t.Fatalf("Unable to create file %v, got error:%v", cfgFilePath, err) } content := fmt.Sprintf(`{"server": true, "bind_addr" : "10.0.0.1", "data_dir": "%s"}`, dir) _, err = cfgFile.Write([]byte(content)) if err != nil { t.Fatalf("err: %v", err) } ui := newCaptureUI() cmd := New(ui) args := []string{"-config-file=" + cfgFile.Name()} if code := cmd.Run(args); code == 0 { t.Fatalf("should fail") } if out := ui.ErrorWriter.String(); !strings.Contains(out, dir) { t.Fatalf("expected mdb dir error, got: %s", out) } } func TestBadDataDirPermissions(t *testing.T) { t.Parallel() dir := testutil.TempDir(t, "consul") dataDir := filepath.Join(dir, "mdb") if err := os.MkdirAll(dataDir, 0400); err != nil { t.Fatalf("err: %v", err) } ui := newCaptureUI() cmd := New(ui) args := []string{"-data-dir=" + dataDir, "-server=true", "-bind=10.0.0.1"} if code := cmd.Run(args); code == 0 { t.Fatalf("Should fail with bad data directory permissions") } if out := ui.ErrorWriter.String(); !strings.Contains(out, "Permission denied") { t.Fatalf("expected permission denied error, got: %s", out) } } type captureUI struct { *mcli.MockUi } func (c *captureUI) Stdout() io.Writer { return c.MockUi.OutputWriter } func (c *captureUI) Stderr() io.Writer { return c.MockUi.ErrorWriter } func (c *captureUI) HeaderOutput(s string) { } func (c *captureUI) ErrorOutput(s string) { } func (c *captureUI) WarnOutput(s string) { } func (c *captureUI) SuccessOutput(s string) { } func (c *captureUI) UnchangedOutput(s string) { } func (c *captureUI) Table(tbl *cli.Table) { } func newCaptureUI() *captureUI { return &captureUI{MockUi: mcli.NewMockUi()} }