open-consul/command/agent/command_test.go

197 lines
3.7 KiB
Go
Raw Normal View History

2013-12-19 20:18:06 +00:00
package agent
import (
2014-10-12 19:27:03 +00:00
"fmt"
"io/ioutil"
"log"
"os"
"path/filepath"
"strings"
2013-12-19 20:18:06 +00:00
"testing"
2014-10-12 19:27:03 +00:00
"github.com/hashicorp/consul/testutil"
"github.com/mitchellh/cli"
2013-12-19 20:18:06 +00:00
)
func TestCommand_implements(t *testing.T) {
var _ cli.Command = new(Command)
}
func TestValidDatacenter(t *testing.T) {
shouldMatch := []string{
"dc1",
"east-aws-001",
"PROD_aws01-small",
}
noMatch := []string{
"east.aws",
"east!aws",
"first,second",
}
for _, m := range shouldMatch {
if !validDatacenter.MatchString(m) {
t.Fatalf("expected match: %s", m)
}
}
for _, m := range noMatch {
if validDatacenter.MatchString(m) {
t.Fatalf("expected no match: %s", m)
}
}
}
2014-10-12 19:27:03 +00:00
func TestRetryJoin(t *testing.T) {
dir, agent := makeAgent(t, nextConfig())
defer os.RemoveAll(dir)
defer agent.Shutdown()
conf2 := nextConfig()
tmpDir, err := ioutil.TempDir("", "consul")
if err != nil {
t.Fatalf("err: %s", err)
}
defer os.RemoveAll(tmpDir)
doneCh := make(chan struct{})
shutdownCh := make(chan struct{})
defer func() {
close(shutdownCh)
<-doneCh
}()
cmd := &Command{
ShutdownCh: shutdownCh,
Ui: new(cli.MockUi),
}
serfAddr := fmt.Sprintf(
"%s:%d",
agent.config.BindAddr,
agent.config.Ports.SerfLan)
2014-11-19 21:53:17 +00:00
serfWanAddr := fmt.Sprintf(
"%s:%d",
agent.config.BindAddr,
agent.config.Ports.SerfWan)
2014-10-12 19:27:03 +00:00
args := []string{
2014-11-19 21:53:17 +00:00
"-server",
2014-10-12 19:27:03 +00:00
"-data-dir", tmpDir,
2014-10-15 00:49:17 +00:00
"-node", fmt.Sprintf(`"%s"`, conf2.NodeName),
2014-10-12 19:27:03 +00:00
"-retry-join", serfAddr,
"-retry-interval", "1s",
2014-11-19 21:53:17 +00:00
"-retry-join-wan", serfWanAddr,
"-retry-interval-wan", "1s",
2014-10-12 19:27:03 +00:00
}
go func() {
if code := cmd.Run(args); code != 0 {
log.Printf("bad: %d", code)
}
close(doneCh)
}()
testutil.WaitForResult(func() (bool, error) {
mem := agent.LANMembers()
if len(mem) != 2 {
return false, fmt.Errorf("bad: %#v", mem)
}
2014-11-19 21:53:17 +00:00
mem = agent.WANMembers()
if len(mem) != 2 {
return false, fmt.Errorf("bad (wan): %#v", mem)
}
2014-10-12 19:27:03 +00:00
return true, nil
}, func(err error) {
t.Fatalf(err.Error())
})
}
func TestRetryJoinFail(t *testing.T) {
conf := nextConfig()
tmpDir, err := ioutil.TempDir("", "consul")
if err != nil {
t.Fatalf("err: %s", err)
}
defer os.RemoveAll(tmpDir)
shutdownCh := make(chan struct{})
defer close(shutdownCh)
cmd := &Command{
ShutdownCh: shutdownCh,
Ui: new(cli.MockUi),
}
serfAddr := fmt.Sprintf("%s:%d", conf.BindAddr, conf.Ports.SerfLan)
args := []string{
"-data-dir", tmpDir,
"-retry-join", serfAddr,
"-retry-max", "1",
}
if code := cmd.Run(args); code == 0 {
t.Fatalf("bad: %d", code)
}
}
func TestRetryJoinWanFail(t *testing.T) {
conf := nextConfig()
tmpDir, err := ioutil.TempDir("", "consul")
if err != nil {
t.Fatalf("err: %s", err)
}
defer os.RemoveAll(tmpDir)
shutdownCh := make(chan struct{})
defer close(shutdownCh)
cmd := &Command{
ShutdownCh: shutdownCh,
Ui: new(cli.MockUi),
}
serfAddr := fmt.Sprintf("%s:%d", conf.BindAddr, conf.Ports.SerfWan)
args := []string{
2014-11-19 21:53:17 +00:00
"-server",
"-data-dir", tmpDir,
"-retry-join-wan", serfAddr,
2014-11-19 21:53:17 +00:00
"-retry-max-wan", "1",
}
if code := cmd.Run(args); code == 0 {
t.Fatalf("bad: %d", code)
}
}
func TestArgConflict(t *testing.T) {
ui := new(cli.MockUi)
c := &Command{Ui: ui}
dir, err := ioutil.TempDir("", "agent")
if err != nil {
t.Fatalf("err: %s", err)
}
key := "HS5lJ+XuTlYKWaeGYyG+/A=="
fileLAN := filepath.Join(dir, SerfLANKeyring)
if err := testutil.InitKeyring(fileLAN, key); err != nil {
t.Fatalf("err: %s", err)
}
args := []string{
"-encrypt=" + key,
"-data-dir=" + dir,
}
code := c.Run(args)
if code != 1 {
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String())
}
if !strings.Contains(ui.ErrorWriter.String(), "keyring files exist") {
t.Fatalf("bad: %#v", ui.ErrorWriter.String())
}
}