package serf
|
|
import (
|
"fmt"
|
"reflect"
|
"testing"
|
"time"
|
|
"basic.com/valib/serf.git/testutil"
|
)
|
|
func TestDefaultQuery(t *testing.T) {
|
ip1, returnFn1 := testutil.TakeIP()
|
defer returnFn1()
|
|
s1Config := testConfig(t, ip1)
|
s1, err := Create(s1Config)
|
if err != nil {
|
t.Fatalf("err: %v", err)
|
}
|
defer s1.Shutdown()
|
|
timeout := s1.DefaultQueryTimeout()
|
if timeout != s1Config.MemberlistConfig.GossipInterval*time.Duration(s1Config.QueryTimeoutMult) {
|
t.Fatalf("bad: %v", timeout)
|
}
|
|
params := s1.DefaultQueryParams()
|
if params.FilterNodes != nil {
|
t.Fatalf("bad: %v", *params)
|
}
|
if params.FilterTags != nil {
|
t.Fatalf("bad: %v", *params)
|
}
|
if params.RequestAck {
|
t.Fatalf("bad: %v", *params)
|
}
|
if params.Timeout != timeout {
|
t.Fatalf("bad: %v", *params)
|
}
|
}
|
|
func TestQueryParams_EncodeFilters(t *testing.T) {
|
q := &QueryParam{
|
FilterNodes: []string{"foo", "bar"},
|
FilterTags: map[string]string{
|
"role": "^web",
|
"datacenter": "aws$",
|
},
|
}
|
|
filters, err := q.encodeFilters()
|
if err != nil {
|
t.Fatalf("err: %v", err)
|
}
|
if len(filters) != 3 {
|
t.Fatalf("bad: %v", filters)
|
}
|
|
nodeFilt := filters[0]
|
if filterType(nodeFilt[0]) != filterNodeType {
|
t.Fatalf("bad: %v", nodeFilt)
|
}
|
|
tagFilt := filters[1]
|
if filterType(tagFilt[0]) != filterTagType {
|
t.Fatalf("bad: %v", tagFilt)
|
}
|
|
tagFilt = filters[2]
|
if filterType(tagFilt[0]) != filterTagType {
|
t.Fatalf("bad: %v", tagFilt)
|
}
|
}
|
|
func TestSerf_ShouldProcess(t *testing.T) {
|
ip1, returnFn1 := testutil.TakeIP()
|
defer returnFn1()
|
|
s1Config := testConfig(t, ip1)
|
s1Config.NodeName = "zip"
|
s1Config.Tags = map[string]string{
|
"role": "webserver",
|
"datacenter": "east-aws",
|
}
|
s1, err := Create(s1Config)
|
if err != nil {
|
t.Fatalf("err: %v", err)
|
}
|
defer s1.Shutdown()
|
|
// Try a matching query
|
q := &QueryParam{
|
FilterNodes: []string{"foo", "bar", "zip"},
|
FilterTags: map[string]string{
|
"role": "^web",
|
"datacenter": "aws$",
|
},
|
}
|
filters, err := q.encodeFilters()
|
if err != nil {
|
t.Fatalf("err: %v", err)
|
}
|
|
if !s1.shouldProcessQuery(filters) {
|
t.Fatalf("expected true")
|
}
|
|
// Omit node
|
q = &QueryParam{
|
FilterNodes: []string{"foo", "bar"},
|
}
|
filters, err = q.encodeFilters()
|
if err != nil {
|
t.Fatalf("err: %v", err)
|
}
|
if s1.shouldProcessQuery(filters) {
|
t.Fatalf("expected false")
|
}
|
|
// Filter on missing tag
|
q = &QueryParam{
|
FilterTags: map[string]string{
|
"other": "cool",
|
},
|
}
|
filters, err = q.encodeFilters()
|
if err != nil {
|
t.Fatalf("err: %v", err)
|
}
|
if s1.shouldProcessQuery(filters) {
|
t.Fatalf("expected false")
|
}
|
|
// Bad tag
|
q = &QueryParam{
|
FilterTags: map[string]string{
|
"role": "db",
|
},
|
}
|
filters, err = q.encodeFilters()
|
if err != nil {
|
t.Fatalf("err: %v", err)
|
}
|
if s1.shouldProcessQuery(filters) {
|
t.Fatalf("expected false")
|
}
|
}
|
|
func Test_kRandomMembers(t *testing.T) {
|
nodes := []Member{}
|
for i := 0; i < 90; i++ {
|
// Half the nodes are in a bad state
|
state := StatusAlive
|
switch i % 3 {
|
case 0:
|
state = StatusAlive
|
case 1:
|
state = StatusFailed
|
case 2:
|
state = StatusLeft
|
}
|
nodes = append(nodes, Member{
|
Name: fmt.Sprintf("test%d", i),
|
Status: state,
|
})
|
}
|
|
filterFunc := func(m Member) bool {
|
if m.Name == "test0" || m.Status != StatusAlive {
|
return true
|
}
|
return false
|
}
|
|
s1 := kRandomMembers(3, nodes, filterFunc)
|
s2 := kRandomMembers(3, nodes, filterFunc)
|
s3 := kRandomMembers(3, nodes, filterFunc)
|
|
if reflect.DeepEqual(s1, s2) {
|
t.Fatalf("unexpected equal")
|
}
|
if reflect.DeepEqual(s1, s3) {
|
t.Fatalf("unexpected equal")
|
}
|
if reflect.DeepEqual(s2, s3) {
|
t.Fatalf("unexpected equal")
|
}
|
|
for _, s := range [][]Member{s1, s2, s3} {
|
if len(s) != 3 {
|
t.Fatalf("bad len")
|
}
|
for _, m := range s {
|
if m.Name == "test0" {
|
t.Fatalf("Bad name")
|
}
|
if m.Status != StatusAlive {
|
t.Fatalf("Bad state")
|
}
|
}
|
}
|
}
|
