55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
package command
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/consul/command/agent"
|
|
"github.com/hashicorp/consul/command/base"
|
|
"github.com/mitchellh/cli"
|
|
)
|
|
|
|
func testLeaveCommand(t *testing.T) (*cli.MockUi, *LeaveCommand) {
|
|
ui := new(cli.MockUi)
|
|
return ui, &LeaveCommand{
|
|
Command: base.Command{
|
|
UI: ui,
|
|
Flags: base.FlagSetClientHTTP,
|
|
},
|
|
}
|
|
}
|
|
|
|
func TestLeaveCommand_implements(t *testing.T) {
|
|
var _ cli.Command = &LeaveCommand{}
|
|
}
|
|
|
|
func TestLeaveCommandRun(t *testing.T) {
|
|
a := agent.NewTestAgent(t.Name(), nil)
|
|
defer a.Shutdown()
|
|
|
|
ui, c := testLeaveCommand(t)
|
|
args := []string{"-http-addr=" + a.HTTPAddr()}
|
|
|
|
code := c.Run(args)
|
|
if code != 0 {
|
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String())
|
|
}
|
|
|
|
if !strings.Contains(ui.OutputWriter.String(), "leave complete") {
|
|
t.Fatalf("bad: %#v", ui.OutputWriter.String())
|
|
}
|
|
}
|
|
|
|
func TestLeaveCommandFailOnNonFlagArgs(t *testing.T) {
|
|
a := agent.NewTestAgent(t.Name(), nil)
|
|
defer a.Shutdown()
|
|
|
|
_, c := testLeaveCommand(t)
|
|
args := []string{"-http-addr=" + a.HTTPAddr(), "appserver1"}
|
|
|
|
code := c.Run(args)
|
|
if code == 0 {
|
|
t.Fatalf("bad: failed to check for unexpected args")
|
|
}
|
|
}
|