mirror of
https://github.com/status-im/consul.git
synced 2025-02-18 08:36:46 +00:00
82650f73e3
Move the HTTP and DNS endpoints into the agent and control their lifespan via the agent. This removes the requirement to manage HTTP and DNS servers indpendent of the agent since the agent is mostly useless without an endpoint and the endpoints without the agent.
127 lines
3.1 KiB
Go
127 lines
3.1 KiB
Go
package command
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"os"
|
|
"strings"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/consul/api"
|
|
"github.com/hashicorp/consul/command/agent"
|
|
"github.com/hashicorp/consul/consul"
|
|
"github.com/hashicorp/consul/logger"
|
|
"github.com/hashicorp/consul/testutil"
|
|
"github.com/hashicorp/consul/types"
|
|
"github.com/hashicorp/consul/version"
|
|
"github.com/hashicorp/go-uuid"
|
|
"github.com/mitchellh/cli"
|
|
)
|
|
|
|
func init() {
|
|
// Seed the random number generator
|
|
rand.Seed(time.Now().UnixNano())
|
|
|
|
version.Version = "0.8.0"
|
|
}
|
|
|
|
type server struct {
|
|
agent *agent.Agent
|
|
config *agent.Config
|
|
httpAddr string
|
|
dir string
|
|
}
|
|
|
|
func (a *server) Shutdown() {
|
|
a.agent.Shutdown()
|
|
os.RemoveAll(a.dir)
|
|
}
|
|
|
|
func testAgent(t *testing.T) *server {
|
|
return testAgentWithConfig(t, nil)
|
|
}
|
|
|
|
func testAgentWithAPIClient(t *testing.T) (*server, *api.Client) {
|
|
agent := testAgentWithConfig(t, func(c *agent.Config) {})
|
|
client, err := api.NewClient(&api.Config{Address: agent.httpAddr})
|
|
if err != nil {
|
|
t.Fatalf("consul client: %#v", err)
|
|
}
|
|
return agent, client
|
|
}
|
|
|
|
func testAgentWithConfig(t *testing.T, cb func(c *agent.Config)) *server {
|
|
return testAgentWithConfigReload(t, cb, nil)
|
|
}
|
|
|
|
func testAgentWithConfigReload(t *testing.T, cb func(c *agent.Config), reloadCh chan chan error) *server {
|
|
conf := nextConfig()
|
|
if cb != nil {
|
|
cb(conf)
|
|
}
|
|
|
|
conf.DataDir = testutil.TempDir(t, "agent")
|
|
a, err := agent.Create(conf, logger.NewLogWriter(512), nil, reloadCh)
|
|
if err != nil {
|
|
os.RemoveAll(conf.DataDir)
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
conf.Addresses.HTTP = "127.0.0.1"
|
|
addr := fmt.Sprintf("%s:%d", conf.Addresses.HTTP, conf.Ports.HTTP)
|
|
return &server{agent: a, config: conf, httpAddr: addr, dir: conf.DataDir}
|
|
}
|
|
|
|
var nextPort uint64 = 10000
|
|
|
|
func nextConfig() *agent.Config {
|
|
nodeID, err := uuid.GenerateUUID()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
port := int(atomic.AddUint64(&nextPort, 10))
|
|
|
|
conf := agent.DefaultConfig()
|
|
conf.Bootstrap = true
|
|
conf.Datacenter = "dc1"
|
|
conf.NodeName = fmt.Sprintf("Node %d", port)
|
|
conf.NodeID = types.NodeID(nodeID)
|
|
conf.BindAddr = "127.0.0.1"
|
|
conf.Server = true
|
|
conf.Version = version.Version
|
|
conf.Ports = agent.PortConfig{
|
|
DNS: port + 1,
|
|
HTTP: port + 2,
|
|
HTTPS: port + 3,
|
|
SerfLan: port + 4,
|
|
SerfWan: port + 5,
|
|
Server: port + 6,
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
func assertNoTabs(t *testing.T, c cli.Command) {
|
|
if strings.ContainsRune(c.Help(), '\t') {
|
|
t.Errorf("%#v help output contains tabs", c)
|
|
}
|
|
}
|