109 lines
2.3 KiB
Go
109 lines
2.3 KiB
Go
package agent_test
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/consul/consul/agent"
|
|
"github.com/hashicorp/serf/serf"
|
|
)
|
|
|
|
func TestServer_Key_params(t *testing.T) {
|
|
ipv4a := net.ParseIP("127.0.0.1")
|
|
ipv4b := net.ParseIP("1.2.3.4")
|
|
|
|
tests := []struct {
|
|
name string
|
|
sd1 *agent.Server
|
|
sd2 *agent.Server
|
|
equal bool
|
|
}{
|
|
{
|
|
name: "Addr inequality",
|
|
sd1: &agent.Server{
|
|
Name: "s1",
|
|
Datacenter: "dc1",
|
|
Port: 8300,
|
|
Addr: &net.IPAddr{IP: ipv4a},
|
|
},
|
|
sd2: &agent.Server{
|
|
Name: "s1",
|
|
Datacenter: "dc1",
|
|
Port: 8300,
|
|
Addr: &net.IPAddr{IP: ipv4b},
|
|
},
|
|
equal: true,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
if test.sd1.Key().Equal(test.sd2.Key()) != test.equal {
|
|
t.Errorf("Expected a %v result from test %s", test.equal, test.name)
|
|
}
|
|
|
|
// Test Key to make sure it actually works as a key
|
|
m := make(map[agent.Key]bool)
|
|
m[*test.sd1.Key()] = true
|
|
if _, found := m[*test.sd2.Key()]; found != test.equal {
|
|
t.Errorf("Expected a %v result from map test %s", test.equal, test.name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsConsulServer(t *testing.T) {
|
|
m := serf.Member{
|
|
Name: "foo",
|
|
Addr: net.IP([]byte{127, 0, 0, 1}),
|
|
Tags: map[string]string{
|
|
"role": "consul",
|
|
"dc": "east-aws",
|
|
"port": "10000",
|
|
"vsn": "1",
|
|
},
|
|
}
|
|
ok, parts := agent.IsConsulServer(m)
|
|
if !ok || parts.Datacenter != "east-aws" || parts.Port != 10000 {
|
|
t.Fatalf("bad: %v %v", ok, parts)
|
|
}
|
|
if parts.Name != "foo" {
|
|
t.Fatalf("bad: %v", parts)
|
|
}
|
|
if parts.Bootstrap {
|
|
t.Fatalf("unexpected bootstrap")
|
|
}
|
|
if parts.Expect != 0 {
|
|
t.Fatalf("bad: %v", parts.Expect)
|
|
}
|
|
m.Tags["bootstrap"] = "1"
|
|
m.Tags["disabled"] = "1"
|
|
ok, parts = agent.IsConsulServer(m)
|
|
if !ok {
|
|
t.Fatalf("expected a valid consul server")
|
|
}
|
|
if !parts.Bootstrap {
|
|
t.Fatalf("expected bootstrap")
|
|
}
|
|
if parts.Addr.String() != "127.0.0.1:10000" {
|
|
t.Fatalf("bad addr: %v", parts.Addr)
|
|
}
|
|
if parts.Version != 1 {
|
|
t.Fatalf("bad: %v", parts)
|
|
}
|
|
m.Tags["expect"] = "3"
|
|
delete(m.Tags, "bootstrap")
|
|
delete(m.Tags, "disabled")
|
|
ok, parts = agent.IsConsulServer(m)
|
|
if !ok || parts.Expect != 3 {
|
|
t.Fatalf("bad: %v", parts.Expect)
|
|
}
|
|
if parts.Bootstrap {
|
|
t.Fatalf("unexpected bootstrap")
|
|
}
|
|
|
|
delete(m.Tags, "role")
|
|
ok, parts = agent.IsConsulServer(m)
|
|
if ok {
|
|
t.Fatalf("unexpected ok server")
|
|
}
|
|
}
|