package maint import ( "strings" "testing" "github.com/hashicorp/consul/agent" "github.com/hashicorp/consul/agent/structs" "github.com/mitchellh/cli" ) func TestMaintCommand_noTabs(t *testing.T) { t.Parallel() if strings.ContainsRune(New(cli.NewMockUi()).Help(), '\t') { t.Fatal("help has tabs") } } func TestMaintCommand_ConflictingArgs(t *testing.T) { t.Parallel() ui := cli.NewMockUi() c := New(ui) c.flags.SetOutput(ui.ErrorWriter) if code := c.Run([]string{"-enable", "-disable"}); code != 1 { t.Fatalf("expected return code 1, got %d", code) } if code := c.Run([]string{"-disable", "-reason=broken"}); code != 1 { t.Fatalf("expected return code 1, got %d", code) } if code := c.Run([]string{"-reason=broken"}); code != 1 { t.Fatalf("expected return code 1, got %d", code) } if code := c.Run([]string{"-service=redis"}); code != 1 { t.Fatalf("expected return code 1, got %d", code) } } func TestMaintCommand_NoArgs(t *testing.T) { t.Parallel() a := agent.NewTestAgent(t, t.Name(), ``) defer a.Shutdown() // Register the service and put it into maintenance mode service := &structs.NodeService{ ID: "test", Service: "test", } if err := a.AddService(service, nil, false, "", agent.ConfigSourceLocal); err != nil { t.Fatalf("err: %v", err) } if err := a.EnableServiceMaintenance(structs.NewServiceID("test", nil), "broken 1", ""); err != nil { t.Fatalf("err: %s", err) } // Enable node maintenance a.EnableNodeMaintenance("broken 2", "") // Run consul maint with no args (list mode) ui := cli.NewMockUi() c := New(ui) c.flags.SetOutput(ui.ErrorWriter) args := []string{"-http-addr=" + a.HTTPAddr()} code := c.Run(args) if code != 0 { t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) } // Ensure the service shows up in the list out := ui.OutputWriter.String() if !strings.Contains(out, "test") { t.Fatalf("bad:\n%s", out) } if !strings.Contains(out, "broken 1") { t.Fatalf("bad:\n%s", out) } // Ensure the node shows up in the list if !strings.Contains(out, a.Config.NodeName) { t.Fatalf("bad:\n%s", out) } if !strings.Contains(out, "broken 2") { t.Fatalf("bad:\n%s", out) } } func TestMaintCommand_EnableNodeMaintenance(t *testing.T) { t.Parallel() a := agent.NewTestAgent(t, t.Name(), ``) defer a.Shutdown() ui := cli.NewMockUi() c := New(ui) c.flags.SetOutput(ui.ErrorWriter) args := []string{ "-http-addr=" + a.HTTPAddr(), "-enable", "-reason=broken", } code := c.Run(args) if code != 0 { t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) } if !strings.Contains(ui.OutputWriter.String(), "now enabled") { t.Fatalf("bad: %#v", ui.OutputWriter.String()) } } func TestMaintCommand_DisableNodeMaintenance(t *testing.T) { t.Parallel() a := agent.NewTestAgent(t, t.Name(), ``) defer a.Shutdown() ui := cli.NewMockUi() c := New(ui) c.flags.SetOutput(ui.ErrorWriter) args := []string{ "-http-addr=" + a.HTTPAddr(), "-disable", } code := c.Run(args) if code != 0 { t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) } if !strings.Contains(ui.OutputWriter.String(), "now disabled") { t.Fatalf("bad: %#v", ui.OutputWriter.String()) } } func TestMaintCommand_EnableServiceMaintenance(t *testing.T) { t.Parallel() a := agent.NewTestAgent(t, t.Name(), ``) defer a.Shutdown() // Register the service service := &structs.NodeService{ ID: "test", Service: "test", } if err := a.AddService(service, nil, false, "", agent.ConfigSourceLocal); err != nil { t.Fatalf("err: %v", err) } ui := cli.NewMockUi() c := New(ui) c.flags.SetOutput(ui.ErrorWriter) args := []string{ "-http-addr=" + a.HTTPAddr(), "-enable", "-service=test", "-reason=broken", } code := c.Run(args) if code != 0 { t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) } if !strings.Contains(ui.OutputWriter.String(), "now enabled") { t.Fatalf("bad: %#v", ui.OutputWriter.String()) } } func TestMaintCommand_DisableServiceMaintenance(t *testing.T) { t.Parallel() a := agent.NewTestAgent(t, t.Name(), ``) defer a.Shutdown() // Register the service service := &structs.NodeService{ ID: "test", Service: "test", } if err := a.AddService(service, nil, false, "", agent.ConfigSourceLocal); err != nil { t.Fatalf("err: %v", err) } ui := cli.NewMockUi() c := New(ui) c.flags.SetOutput(ui.ErrorWriter) args := []string{ "-http-addr=" + a.HTTPAddr(), "-disable", "-service=test", } code := c.Run(args) if code != 0 { t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String()) } if !strings.Contains(ui.OutputWriter.String(), "now disabled") { t.Fatalf("bad: %#v", ui.OutputWriter.String()) } } func TestMaintCommand_ServiceMaintenance_NoService(t *testing.T) { t.Parallel() a := agent.NewTestAgent(t, t.Name(), ``) defer a.Shutdown() ui := cli.NewMockUi() c := New(ui) c.flags.SetOutput(ui.ErrorWriter) args := []string{ "-http-addr=" + a.HTTPAddr(), "-enable", "-service=redis", "-reason=broken", } code := c.Run(args) if code != 1 { t.Fatalf("expected response code 1, got %d", code) } if !strings.Contains(ui.ErrorWriter.String(), "No service registered") { t.Fatalf("bad: %#v", ui.ErrorWriter.String()) } }