161 lines
3 KiB
Go
161 lines
3 KiB
Go
|
package consul
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/hashicorp/consul/types"
|
||
|
|
"github.com/hashicorp/serf/serf"
|
||
|
|
)
|
||
|
|
|
||
|
|
func makeNode(dc, name, id string, server bool) *serf.Member {
|
||
|
|
var role string
|
||
|
|
if server {
|
||
|
|
role = "consul"
|
||
|
|
} else {
|
||
|
|
role = "node"
|
||
|
|
}
|
||
|
|
|
||
|
|
return &serf.Member{
|
||
|
|
Name: name,
|
||
|
|
Tags: map[string]string{
|
||
|
|
"role": role,
|
||
|
|
"dc": dc,
|
||
|
|
"id": id,
|
||
|
|
"port": "8300",
|
||
|
|
"build": "0.7.5",
|
||
|
|
"vsn": "2",
|
||
|
|
"vsn_max": "3",
|
||
|
|
"vsn_min": "2",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMerge_LAN(t *testing.T) {
|
||
|
|
cases := []struct {
|
||
|
|
members []*serf.Member
|
||
|
|
expect string
|
||
|
|
}{
|
||
|
|
// Client in the wrong datacenter.
|
||
|
|
{
|
||
|
|
members: []*serf.Member{
|
||
|
|
makeNode("dc2",
|
||
|
|
"node1",
|
||
|
|
"96430788-246f-4379-94ce-257f7429e340",
|
||
|
|
false),
|
||
|
|
},
|
||
|
|
expect: "wrong datacenter",
|
||
|
|
},
|
||
|
|
// Server in the wrong datacenter.
|
||
|
|
{
|
||
|
|
members: []*serf.Member{
|
||
|
|
makeNode("dc2",
|
||
|
|
"node1",
|
||
|
|
"96430788-246f-4379-94ce-257f7429e340",
|
||
|
|
true),
|
||
|
|
},
|
||
|
|
expect: "wrong datacenter",
|
||
|
|
},
|
||
|
|
// Node ID conflict with delegate's ID.
|
||
|
|
{
|
||
|
|
members: []*serf.Member{
|
||
|
|
makeNode("dc1",
|
||
|
|
"node1",
|
||
|
|
"ee954a2f-80de-4b34-8780-97b942a50a99",
|
||
|
|
true),
|
||
|
|
},
|
||
|
|
expect: "with this agent's ID",
|
||
|
|
},
|
||
|
|
// Cluster with existing conflicting node IDs.
|
||
|
|
{
|
||
|
|
members: []*serf.Member{
|
||
|
|
makeNode("dc1",
|
||
|
|
"node1",
|
||
|
|
"6185913b-98d7-4441-bd8f-f7f7d854a4af",
|
||
|
|
true),
|
||
|
|
makeNode("dc1",
|
||
|
|
"node2",
|
||
|
|
"6185913b-98d7-4441-bd8f-f7f7d854a4af",
|
||
|
|
true),
|
||
|
|
},
|
||
|
|
expect: "with member",
|
||
|
|
},
|
||
|
|
// Good cluster.
|
||
|
|
{
|
||
|
|
members: []*serf.Member{
|
||
|
|
makeNode("dc1",
|
||
|
|
"node1",
|
||
|
|
"6185913b-98d7-4441-bd8f-f7f7d854a4af",
|
||
|
|
true),
|
||
|
|
makeNode("dc1",
|
||
|
|
"node2",
|
||
|
|
"cda916bc-a357-4a19-b886-59419fcee50c",
|
||
|
|
true),
|
||
|
|
},
|
||
|
|
expect: "",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
delegate := &lanMergeDelegate{
|
||
|
|
dc: "dc1",
|
||
|
|
nodeID: types.NodeID("ee954a2f-80de-4b34-8780-97b942a50a99"),
|
||
|
|
nodeName: "node0",
|
||
|
|
}
|
||
|
|
for i, c := range cases {
|
||
|
|
if err := delegate.NotifyMerge(c.members); c.expect == "" {
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("case %d: err: %v", i+1, err)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if err == nil || !strings.Contains(err.Error(), c.expect) {
|
||
|
|
t.Fatalf("case %d: err: %v", i+1, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMerge_WAN(t *testing.T) {
|
||
|
|
cases := []struct {
|
||
|
|
members []*serf.Member
|
||
|
|
expect string
|
||
|
|
}{
|
||
|
|
// Not a server
|
||
|
|
{
|
||
|
|
members: []*serf.Member{
|
||
|
|
makeNode("dc2",
|
||
|
|
"node1",
|
||
|
|
"96430788-246f-4379-94ce-257f7429e340",
|
||
|
|
false),
|
||
|
|
},
|
||
|
|
expect: "not a server",
|
||
|
|
},
|
||
|
|
// Good cluster.
|
||
|
|
{
|
||
|
|
members: []*serf.Member{
|
||
|
|
makeNode("dc2",
|
||
|
|
"node1",
|
||
|
|
"6185913b-98d7-4441-bd8f-f7f7d854a4af",
|
||
|
|
true),
|
||
|
|
makeNode("dc3",
|
||
|
|
"node2",
|
||
|
|
"cda916bc-a357-4a19-b886-59419fcee50c",
|
||
|
|
true),
|
||
|
|
},
|
||
|
|
expect: "",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
delegate := &wanMergeDelegate{}
|
||
|
|
for i, c := range cases {
|
||
|
|
if err := delegate.NotifyMerge(c.members); c.expect == "" {
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("case %d: err: %v", i+1, err)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if err == nil || !strings.Contains(err.Error(), c.expect) {
|
||
|
|
t.Fatalf("case %d: err: %v", i+1, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|