package command import ( "fmt" "github.com/hashicorp/consul/command/agent" "github.com/hashicorp/consul/consul" "io" "io/ioutil" "math/rand" "net" "os" "sync/atomic" "testing" "time" ) var offset uint64 func init() { // Seed the random number generator rand.Seed(time.Now().UnixNano()) } type agentWrapper struct { dir string config *agent.Config agent *agent.Agent rpc *agent.AgentRPC http *agent.HTTPServer addr string httpAddr string } func (a *agentWrapper) Shutdown() { a.rpc.Shutdown() a.agent.Shutdown() a.http.Shutdown() os.RemoveAll(a.dir) } func testAgent(t *testing.T) *agentWrapper { return testAgentWithConfig(t, func(c *agent.Config) {}) } func testAgentWithConfig(t *testing.T, cb func(c *agent.Config)) *agentWrapper { l, err := net.Listen("tcp", "127.0.0.1:0") if err != nil { t.Fatalf("err: %s", err) } lw := agent.NewLogWriter(512) mult := io.MultiWriter(os.Stderr, lw) conf := nextConfig() cb(conf) dir, err := ioutil.TempDir("", "agent") if err != nil { t.Fatalf(fmt.Sprintf("err: %v", err)) } conf.DataDir = dir a, err := agent.Create(conf, lw) if err != nil { os.RemoveAll(dir) t.Fatalf(fmt.Sprintf("err: %v", err)) } rpc := agent.NewAgentRPC(a, l, mult, lw) conf.Addresses.HTTP = "127.0.0.1" httpAddr := fmt.Sprintf("127.0.0.1:%d", conf.Ports.HTTP) http, err := agent.NewHTTPServers(a, conf, nil, os.Stderr) if err != nil { os.RemoveAll(dir) t.Fatalf(fmt.Sprintf("err: %v", err)) } if http == nil || len(http) == 0 { os.RemoveAll(dir) t.Fatalf(fmt.Sprintf("Could not create HTTP server to listen on: %s", httpAddr)) } return &agentWrapper{ dir: dir, config: conf, agent: a, rpc: rpc, http: http[0], addr: l.Addr().String(), httpAddr: httpAddr, } } func nextConfig() *agent.Config { idx := int(atomic.AddUint64(&offset, 1)) conf := agent.DefaultConfig() conf.Bootstrap = true conf.Datacenter = "dc1" conf.NodeName = fmt.Sprintf("Node %d", idx) conf.BindAddr = "127.0.0.1" conf.Server = true conf.Ports.HTTP = 10000 + 10*idx conf.Ports.HTTPS = 10401 + 10*idx conf.Ports.RPC = 10100 + 10*idx conf.Ports.SerfLan = 10201 + 10*idx conf.Ports.SerfWan = 10202 + 10*idx conf.Ports.Server = 10300 + 10*idx cons := consul.DefaultConfig() conf.ConsulConfig = cons cons.SerfLANConfig.MemberlistConfig.ProbeTimeout = 100 * time.Millisecond cons.SerfLANConfig.MemberlistConfig.ProbeInterval = 100 * time.Millisecond cons.SerfLANConfig.MemberlistConfig.GossipInterval = 100 * time.Millisecond cons.SerfWANConfig.MemberlistConfig.ProbeTimeout = 100 * time.Millisecond cons.SerfWANConfig.MemberlistConfig.ProbeInterval = 100 * time.Millisecond cons.SerfWANConfig.MemberlistConfig.GossipInterval = 100 * time.Millisecond cons.RaftConfig.LeaderLeaseTimeout = 20 * time.Millisecond cons.RaftConfig.HeartbeatTimeout = 40 * time.Millisecond cons.RaftConfig.ElectionTimeout = 40 * time.Millisecond return conf }