func TestRPCClientJoin(t *testing.T) {
client, a1, ipc := testRPCClient(t)
a2 := testAgent(nil)
defer ipc.Shutdown()
defer client.Close()
defer a1.Shutdown()
defer a2.Shutdown()
if err := a1.Start(); err != nil {
t.Fatalf("err: %s", err)
}
if err := a2.Start(); err != nil {
t.Fatalf("err: %s", err)
}
testutil.Yield()
n, err := client.Join([]string{a2.conf.MemberlistConfig.BindAddr}, false)
if err != nil {
t.Fatalf("err: %s", err)
}
if n != 1 {
t.Fatalf("n != 1: %d", n)
}
}
func clusterUp(clusterAddr string) (*client.RPCClient, error) {
client, err := client.NewRPCClient(clusterAddr)
if err != nil {
return nil, err
}
_, err = client.Join([]string{clusterAddr}, false)
if err != nil {
return nil, err
}
return client, nil
}
func TestRPCClientMembers(t *testing.T) {
client, a1, ipc := testRPCClient(t)
a2 := testAgent(nil)
defer ipc.Shutdown()
defer client.Close()
defer a1.Shutdown()
defer a2.Shutdown()
if err := a1.Start(); err != nil {
t.Fatalf("err: %s", err)
}
if err := a2.Start(); err != nil {
t.Fatalf("err: %s", err)
}
testutil.Yield()
mem, err := client.Members()
if err != nil {
t.Fatalf("err: %s", err)
}
if len(mem) != 1 {
t.Fatalf("bad: %#v", mem)
}
_, err = client.Join([]string{a2.conf.MemberlistConfig.BindAddr}, false)
if err != nil {
t.Fatalf("err: %s", err)
}
testutil.Yield()
mem, err = client.Members()
if err != nil {
t.Fatalf("err: %s", err)
}
if len(mem) != 2 {
t.Fatalf("bad: %#v", mem)
}
}
func TestRPCClientMembersFiltered(t *testing.T) {
client, a1, ipc := testRPCClient(t)
a2 := testAgent(nil)
defer ipc.Shutdown()
defer client.Close()
defer a1.Shutdown()
defer a2.Shutdown()
if err := a1.Start(); err != nil {
t.Fatalf("err: %s", err)
}
if err := a2.Start(); err != nil {
t.Fatalf("err: %s", err)
}
testutil.Yield()
_, err := client.Join([]string{a2.conf.MemberlistConfig.BindAddr}, false)
if err != nil {
t.Fatalf("err: %s", err)
}
err = client.UpdateTags(map[string]string{
"tag1": "val1",
"tag2": "val2",
}, []string{})
if err != nil {
t.Fatalf("bad: %s", err)
}
testutil.Yield()
// Make sure that filters work on member names
mem, err := client.MembersFiltered(map[string]string{}, "", ".*")
if err != nil {
t.Fatalf("bad: %s", err)
}
if len(mem) == 0 {
t.Fatalf("should have matched more than 0 members")
}
mem, err = client.MembersFiltered(map[string]string{}, "", "bad")
if err != nil {
t.Fatalf("bad: %s", err)
}
if len(mem) != 0 {
t.Fatalf("should have matched 0 members: %#v", mem)
}
// Make sure that filters work on member tags
mem, err = client.MembersFiltered(map[string]string{"tag1": "val.*"}, "", "")
if err != nil {
t.Fatalf("bad: %s", err)
}
if len(mem) != 1 {
t.Fatalf("should have matched 1 member: %#v", mem)
}
// Make sure tag filters work on multiple tags
mem, err = client.MembersFiltered(map[string]string{
"tag1": "val.*",
"tag2": "val2",
}, "", "")
if err != nil {
t.Fatalf("bad: %s", err)
}
if len(mem) != 1 {
t.Fatalf("should have matched one member: %#v", mem)
}
// Make sure all tags match when multiple tags are passed
mem, err = client.MembersFiltered(map[string]string{
"tag1": "val1",
"tag2": "bad",
}, "", "")
if err != nil {
t.Fatalf("bad: %s", err)
}
if len(mem) != 0 {
t.Fatalf("should have matched 0 members: %#v", mem)
}
// Make sure that filters work on member status
if err := client.ForceLeave(a2.conf.NodeName); err != nil {
t.Fatalf("bad: %s", err)
}
mem, err = client.MembersFiltered(map[string]string{}, "alive", "")
if err != nil {
t.Fatalf("err: %s", err)
}
if len(mem) != 1 {
t.Fatalf("should have matched 1 member: %#v", mem)
}
mem, err = client.MembersFiltered(map[string]string{}, "leaving", "")
if err != nil {
t.Fatalf("err: %s", err)
}
if len(mem) != 1 {
t.Fatalf("should have matched 1 member: %#v", mem)
}
}