2013-12-23 21:52:10 +00:00
|
|
|
package agent
|
|
|
|
|
|
|
|
import (
|
2014-11-24 08:36:03 +00:00
|
|
|
"bytes"
|
|
|
|
"encoding/json"
|
2013-12-23 21:52:10 +00:00
|
|
|
"fmt"
|
2013-12-30 22:42:41 +00:00
|
|
|
"io"
|
2013-12-23 21:52:10 +00:00
|
|
|
"io/ioutil"
|
|
|
|
"os"
|
2014-09-18 05:31:32 +00:00
|
|
|
"path/filepath"
|
2013-12-23 21:52:10 +00:00
|
|
|
"sync/atomic"
|
|
|
|
"testing"
|
|
|
|
"time"
|
2014-11-19 19:51:25 +00:00
|
|
|
|
|
|
|
"github.com/hashicorp/consul/consul"
|
|
|
|
"github.com/hashicorp/consul/consul/structs"
|
|
|
|
"github.com/hashicorp/consul/testutil"
|
2013-12-23 21:52:10 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
var offset uint64
|
|
|
|
|
|
|
|
func nextConfig() *Config {
|
2014-04-11 22:46:55 +00:00
|
|
|
idx := int(atomic.AddUint64(&offset, 1))
|
2013-12-23 21:52:10 +00:00
|
|
|
conf := DefaultConfig()
|
|
|
|
|
2014-01-21 00:22:59 +00:00
|
|
|
conf.AdvertiseAddr = "127.0.0.1"
|
2013-12-25 00:53:30 +00:00
|
|
|
conf.Bootstrap = true
|
2013-12-24 00:20:51 +00:00
|
|
|
conf.Datacenter = "dc1"
|
2013-12-30 22:42:41 +00:00
|
|
|
conf.NodeName = fmt.Sprintf("Node %d", idx)
|
2014-04-11 22:46:55 +00:00
|
|
|
conf.BindAddr = "127.0.0.1"
|
2014-11-19 19:51:25 +00:00
|
|
|
conf.Ports.DNS = 19000 + idx
|
|
|
|
conf.Ports.HTTP = 18800 + idx
|
|
|
|
conf.Ports.RPC = 18600 + idx
|
2014-04-11 22:46:55 +00:00
|
|
|
conf.Ports.SerfLan = 18200 + idx
|
2014-11-19 19:51:25 +00:00
|
|
|
conf.Ports.SerfWan = 18400 + idx
|
|
|
|
conf.Ports.Server = 18000 + idx
|
2013-12-23 21:52:10 +00:00
|
|
|
conf.Server = true
|
2014-08-12 21:48:36 +00:00
|
|
|
conf.ACLDatacenter = "dc1"
|
2014-08-19 21:28:49 +00:00
|
|
|
conf.ACLMasterToken = "root"
|
2013-12-23 21:52:10 +00:00
|
|
|
|
|
|
|
cons := consul.DefaultConfig()
|
|
|
|
conf.ConsulConfig = cons
|
|
|
|
|
2014-01-06 21:21:48 +00:00
|
|
|
cons.SerfLANConfig.MemberlistConfig.SuspicionMult = 3
|
2013-12-30 22:42:41 +00:00
|
|
|
cons.SerfLANConfig.MemberlistConfig.ProbeTimeout = 100 * time.Millisecond
|
|
|
|
cons.SerfLANConfig.MemberlistConfig.ProbeInterval = 100 * time.Millisecond
|
2013-12-23 21:52:10 +00:00
|
|
|
cons.SerfLANConfig.MemberlistConfig.GossipInterval = 100 * time.Millisecond
|
|
|
|
|
2014-01-06 21:21:48 +00:00
|
|
|
cons.SerfWANConfig.MemberlistConfig.SuspicionMult = 3
|
2013-12-30 22:42:41 +00:00
|
|
|
cons.SerfWANConfig.MemberlistConfig.ProbeTimeout = 100 * time.Millisecond
|
|
|
|
cons.SerfWANConfig.MemberlistConfig.ProbeInterval = 100 * time.Millisecond
|
2013-12-23 21:52:10 +00:00
|
|
|
cons.SerfWANConfig.MemberlistConfig.GossipInterval = 100 * time.Millisecond
|
|
|
|
|
2014-05-30 18:57:39 +00:00
|
|
|
cons.RaftConfig.LeaderLeaseTimeout = 20 * time.Millisecond
|
2013-12-23 21:52:10 +00:00
|
|
|
cons.RaftConfig.HeartbeatTimeout = 40 * time.Millisecond
|
|
|
|
cons.RaftConfig.ElectionTimeout = 40 * time.Millisecond
|
|
|
|
|
|
|
|
return conf
|
|
|
|
}
|
|
|
|
|
2013-12-30 22:42:41 +00:00
|
|
|
func makeAgentLog(t *testing.T, conf *Config, l io.Writer) (string, *Agent) {
|
2013-12-23 21:52:10 +00:00
|
|
|
dir, err := ioutil.TempDir("", "agent")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf(fmt.Sprintf("err: %v", err))
|
|
|
|
}
|
|
|
|
|
|
|
|
conf.DataDir = dir
|
2013-12-30 22:42:41 +00:00
|
|
|
agent, err := Create(conf, l)
|
2013-12-23 21:52:10 +00:00
|
|
|
if err != nil {
|
|
|
|
os.RemoveAll(dir)
|
|
|
|
t.Fatalf(fmt.Sprintf("err: %v", err))
|
|
|
|
}
|
|
|
|
|
|
|
|
return dir, agent
|
|
|
|
}
|
|
|
|
|
2014-09-18 05:31:32 +00:00
|
|
|
func makeAgentKeyring(t *testing.T, conf *Config, key string) (string, *Agent) {
|
|
|
|
dir, err := ioutil.TempDir("", "agent")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
conf.DataDir = dir
|
|
|
|
|
2014-10-04 20:43:10 +00:00
|
|
|
fileLAN := filepath.Join(dir, serfLANKeyring)
|
2014-10-10 18:13:30 +00:00
|
|
|
if err := initKeyring(fileLAN, key); err != nil {
|
2014-09-18 05:31:32 +00:00
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
2014-10-04 20:43:10 +00:00
|
|
|
fileWAN := filepath.Join(dir, serfWANKeyring)
|
2014-10-10 18:13:30 +00:00
|
|
|
if err := initKeyring(fileWAN, key); err != nil {
|
2014-09-18 05:31:32 +00:00
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
agent, err := Create(conf, nil)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
return dir, agent
|
|
|
|
}
|
|
|
|
|
2013-12-30 22:42:41 +00:00
|
|
|
func makeAgent(t *testing.T, conf *Config) (string, *Agent) {
|
|
|
|
return makeAgentLog(t, conf, nil)
|
|
|
|
}
|
|
|
|
|
2013-12-23 21:52:10 +00:00
|
|
|
func TestAgentStartStop(t *testing.T) {
|
|
|
|
dir, agent := makeAgent(t, nextConfig())
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
|
|
|
|
if err := agent.Leave(); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
if err := agent.Shutdown(); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
select {
|
|
|
|
case <-agent.ShutdownCh():
|
|
|
|
default:
|
|
|
|
t.Fatalf("should be closed")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestAgent_RPCPing(t *testing.T) {
|
|
|
|
dir, agent := makeAgent(t, nextConfig())
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := agent.RPC("Status.Ping", struct{}{}, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
2014-01-30 21:39:02 +00:00
|
|
|
|
|
|
|
func TestAgent_AddService(t *testing.T) {
|
|
|
|
dir, agent := makeAgent(t, nextConfig())
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
|
|
|
|
srv := &structs.NodeService{
|
|
|
|
ID: "redis",
|
|
|
|
Service: "redis",
|
2014-04-03 19:12:23 +00:00
|
|
|
Tags: []string{"foo"},
|
2014-01-30 21:39:02 +00:00
|
|
|
Port: 8000,
|
|
|
|
}
|
2014-11-07 02:24:04 +00:00
|
|
|
chk := &CheckType{
|
|
|
|
TTL: time.Minute,
|
|
|
|
Notes: "redis health check",
|
|
|
|
}
|
2014-01-30 21:39:02 +00:00
|
|
|
err := agent.AddService(srv, chk)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure we have a state mapping
|
|
|
|
if _, ok := agent.state.Services()["redis"]; !ok {
|
|
|
|
t.Fatalf("missing redis service")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure we have a check mapping
|
|
|
|
if _, ok := agent.state.Checks()["service:redis"]; !ok {
|
|
|
|
t.Fatalf("missing redis check")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure a TTL is setup
|
|
|
|
if _, ok := agent.checkTTLs["service:redis"]; !ok {
|
|
|
|
t.Fatalf("missing redis check ttl")
|
|
|
|
}
|
2014-11-07 02:24:04 +00:00
|
|
|
|
|
|
|
// Ensure the notes are passed through
|
|
|
|
if agent.state.Checks()["service:redis"].Notes == "" {
|
|
|
|
t.Fatalf("missing redis check notes")
|
|
|
|
}
|
2014-01-30 21:39:02 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestAgent_RemoveService(t *testing.T) {
|
|
|
|
dir, agent := makeAgent(t, nextConfig())
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
|
|
|
|
// Remove a service that doesn't exist
|
|
|
|
if err := agent.RemoveService("redis"); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2014-10-14 22:05:41 +00:00
|
|
|
// Remove the consul service
|
|
|
|
if err := agent.RemoveService("consul"); err == nil {
|
|
|
|
t.Fatalf("should have errored")
|
|
|
|
}
|
|
|
|
|
2014-01-30 21:39:02 +00:00
|
|
|
srv := &structs.NodeService{
|
|
|
|
ID: "redis",
|
|
|
|
Service: "redis",
|
|
|
|
Port: 8000,
|
|
|
|
}
|
|
|
|
chk := &CheckType{TTL: time.Minute}
|
|
|
|
if err := agent.AddService(srv, chk); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Remove the service
|
|
|
|
if err := agent.RemoveService("redis"); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure we have a state mapping
|
|
|
|
if _, ok := agent.state.Services()["redis"]; ok {
|
|
|
|
t.Fatalf("have redis service")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure we have a check mapping
|
|
|
|
if _, ok := agent.state.Checks()["service:redis"]; ok {
|
|
|
|
t.Fatalf("have redis check")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure a TTL is setup
|
|
|
|
if _, ok := agent.checkTTLs["service:redis"]; ok {
|
|
|
|
t.Fatalf("have redis check ttl")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestAgent_AddCheck(t *testing.T) {
|
|
|
|
dir, agent := makeAgent(t, nextConfig())
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
|
|
|
|
health := &structs.HealthCheck{
|
|
|
|
Node: "foo",
|
|
|
|
CheckID: "mem",
|
|
|
|
Name: "memory util",
|
2014-10-15 17:14:46 +00:00
|
|
|
Status: structs.HealthCritical,
|
2014-01-30 21:39:02 +00:00
|
|
|
}
|
|
|
|
chk := &CheckType{
|
|
|
|
Script: "exit 0",
|
|
|
|
Interval: 15 * time.Second,
|
|
|
|
}
|
|
|
|
err := agent.AddCheck(health, chk)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure we have a check mapping
|
|
|
|
if _, ok := agent.state.Checks()["mem"]; !ok {
|
|
|
|
t.Fatalf("missing mem check")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure a TTL is setup
|
|
|
|
if _, ok := agent.checkMonitors["mem"]; !ok {
|
|
|
|
t.Fatalf("missing mem monitor")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-21 21:42:42 +00:00
|
|
|
func TestAgent_AddCheck_MinInterval(t *testing.T) {
|
|
|
|
dir, agent := makeAgent(t, nextConfig())
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
|
|
|
|
health := &structs.HealthCheck{
|
|
|
|
Node: "foo",
|
|
|
|
CheckID: "mem",
|
|
|
|
Name: "memory util",
|
2014-10-15 17:14:46 +00:00
|
|
|
Status: structs.HealthCritical,
|
2014-04-21 21:42:42 +00:00
|
|
|
}
|
|
|
|
chk := &CheckType{
|
|
|
|
Script: "exit 0",
|
|
|
|
Interval: time.Microsecond,
|
|
|
|
}
|
|
|
|
err := agent.AddCheck(health, chk)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure we have a check mapping
|
|
|
|
if _, ok := agent.state.Checks()["mem"]; !ok {
|
|
|
|
t.Fatalf("missing mem check")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure a TTL is setup
|
|
|
|
if mon, ok := agent.checkMonitors["mem"]; !ok {
|
|
|
|
t.Fatalf("missing mem monitor")
|
|
|
|
} else if mon.Interval != MinInterval {
|
|
|
|
t.Fatalf("bad mem monitor interval")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-30 21:39:02 +00:00
|
|
|
func TestAgent_RemoveCheck(t *testing.T) {
|
|
|
|
dir, agent := makeAgent(t, nextConfig())
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
|
|
|
|
// Remove check that doesn't exist
|
|
|
|
if err := agent.RemoveCheck("mem"); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
health := &structs.HealthCheck{
|
|
|
|
Node: "foo",
|
|
|
|
CheckID: "mem",
|
|
|
|
Name: "memory util",
|
2014-10-15 17:14:46 +00:00
|
|
|
Status: structs.HealthCritical,
|
2014-01-30 21:39:02 +00:00
|
|
|
}
|
|
|
|
chk := &CheckType{
|
|
|
|
Script: "exit 0",
|
|
|
|
Interval: 15 * time.Second,
|
|
|
|
}
|
|
|
|
err := agent.AddCheck(health, chk)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Remove check
|
|
|
|
if err := agent.RemoveCheck("mem"); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure we have a check mapping
|
|
|
|
if _, ok := agent.state.Checks()["mem"]; ok {
|
|
|
|
t.Fatalf("have mem check")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure a TTL is setup
|
|
|
|
if _, ok := agent.checkMonitors["mem"]; ok {
|
|
|
|
t.Fatalf("have mem monitor")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestAgent_UpdateCheck(t *testing.T) {
|
|
|
|
dir, agent := makeAgent(t, nextConfig())
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
|
|
|
|
health := &structs.HealthCheck{
|
|
|
|
Node: "foo",
|
|
|
|
CheckID: "mem",
|
|
|
|
Name: "memory util",
|
2014-10-15 17:14:46 +00:00
|
|
|
Status: structs.HealthCritical,
|
2014-01-30 21:39:02 +00:00
|
|
|
}
|
|
|
|
chk := &CheckType{
|
|
|
|
TTL: 15 * time.Second,
|
|
|
|
}
|
|
|
|
err := agent.AddCheck(health, chk)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Remove check
|
|
|
|
if err := agent.UpdateCheck("mem", structs.HealthPassing, "foo"); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure we have a check mapping
|
|
|
|
status := agent.state.Checks()["mem"]
|
|
|
|
if status.Status != structs.HealthPassing {
|
|
|
|
t.Fatalf("bad: %v", status)
|
|
|
|
}
|
2014-04-21 23:20:22 +00:00
|
|
|
if status.Output != "foo" {
|
2014-01-30 21:39:02 +00:00
|
|
|
t.Fatalf("bad: %v", status)
|
|
|
|
}
|
|
|
|
}
|
2014-10-14 22:05:41 +00:00
|
|
|
|
|
|
|
func TestAgent_ConsulService(t *testing.T) {
|
|
|
|
dir, agent := makeAgent(t, nextConfig())
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
|
2014-10-17 21:43:52 +00:00
|
|
|
testutil.WaitForLeader(t, agent.RPC, "dc1")
|
|
|
|
|
|
|
|
// Consul service is registered
|
2014-10-14 22:05:41 +00:00
|
|
|
services := agent.state.Services()
|
2014-10-14 22:42:49 +00:00
|
|
|
if _, ok := services[consul.ConsulServiceID]; !ok {
|
|
|
|
t.Fatalf("%s service should be registered", consul.ConsulServiceID)
|
2014-10-14 22:05:41 +00:00
|
|
|
}
|
2014-10-17 21:43:52 +00:00
|
|
|
|
|
|
|
// Perform anti-entropy on consul service
|
|
|
|
if err := agent.state.syncService(consul.ConsulServiceID); err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Consul service should be in sync
|
|
|
|
if !agent.state.serviceStatus[consul.ConsulServiceID].inSync {
|
|
|
|
t.Fatalf("%s service should be in sync", consul.ConsulServiceID)
|
|
|
|
}
|
2014-10-14 22:05:41 +00:00
|
|
|
}
|
2014-11-24 08:36:03 +00:00
|
|
|
|
|
|
|
func TestAgent_PersistService(t *testing.T) {
|
|
|
|
config := nextConfig()
|
|
|
|
dir, agent := makeAgent(t, config)
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
|
|
|
|
svc := &structs.NodeService{
|
|
|
|
ID: "redis",
|
|
|
|
Service: "redis",
|
|
|
|
Tags: []string{"foo"},
|
|
|
|
Port: 8000,
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := agent.AddService(svc, nil); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
file := filepath.Join(agent.config.DataDir, servicesDir, svc.ID)
|
|
|
|
if _, err := os.Stat(file); err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
expected, err := json.Marshal(svc)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
content, err := ioutil.ReadFile(file)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
if !bytes.Equal(expected, content) {
|
|
|
|
t.Fatalf("bad: %s", string(content))
|
|
|
|
}
|
|
|
|
agent.Shutdown()
|
|
|
|
|
|
|
|
// Should load it back during later start
|
|
|
|
agent2, err := Create(config, nil)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
defer agent2.Shutdown()
|
|
|
|
|
|
|
|
if _, ok := agent2.state.services[svc.ID]; !ok {
|
|
|
|
t.Fatalf("bad: %#v", agent2.state.services)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Should remove the service file
|
|
|
|
if err := agent2.RemoveService(svc.ID); err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
if _, err := os.Stat(file); !os.IsNotExist(err) {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestAgent_PersistCheck(t *testing.T) {
|
|
|
|
config := nextConfig()
|
|
|
|
dir, agent := makeAgent(t, config)
|
|
|
|
defer os.RemoveAll(dir)
|
|
|
|
|
|
|
|
check := &structs.HealthCheck{
|
|
|
|
Node: config.NodeName,
|
|
|
|
CheckID: "service:redis1",
|
|
|
|
Name: "redischeck",
|
|
|
|
Status: structs.HealthPassing,
|
|
|
|
ServiceID: "redis",
|
|
|
|
ServiceName: "redis",
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := agent.AddCheck(check, nil); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
file := filepath.Join(agent.config.DataDir, checksDir, check.CheckID)
|
|
|
|
if _, err := os.Stat(file); err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
expected, err := json.Marshal(check)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
content, err := ioutil.ReadFile(file)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
if !bytes.Equal(expected, content) {
|
|
|
|
t.Fatalf("bad: %s", string(content))
|
|
|
|
}
|
|
|
|
agent.Shutdown()
|
|
|
|
|
|
|
|
// Should load it back during later start
|
|
|
|
agent2, err := Create(config, nil)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
defer agent2.Shutdown()
|
|
|
|
|
2014-11-24 09:15:18 +00:00
|
|
|
result, ok := agent2.state.checks[check.CheckID]
|
|
|
|
if !ok {
|
2014-11-24 08:36:03 +00:00
|
|
|
t.Fatalf("bad: %#v", agent2.state.checks)
|
|
|
|
}
|
2014-11-24 09:15:18 +00:00
|
|
|
if result.Status != structs.HealthCritical {
|
|
|
|
t.Fatalf("bad: %#v", result)
|
|
|
|
}
|
2014-11-24 08:36:03 +00:00
|
|
|
|
|
|
|
// Should remove the service file
|
|
|
|
if err := agent2.RemoveCheck(check.CheckID); err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
if _, err := os.Stat(file); !os.IsNotExist(err) {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|