func TestRPCClientStream_User(t *testing.T) {
client, a1, ipc := testRPCClient(t)
defer ipc.Shutdown()
defer client.Close()
defer a1.Shutdown()
if err := a1.Start(); err != nil {
t.Fatalf("err: %s", err)
}
eventCh := make(chan map[string]interface{}, 64)
if handle, err := client.Stream("user", eventCh); err != nil {
t.Fatalf("err: %s", err)
} else {
defer client.Stop(handle)
}
testutil.Yield()
if err := client.UserEvent("deploy", []byte("foo"), false); err != nil {
t.Fatalf("err: %s", err)
}
testutil.Yield()
select {
case e := <-eventCh:
if e["Event"].(string) != "user" {
t.Fatalf("bad event: %#v", e)
}
if e["LTime"].(int64) != 1 {
t.Fatalf("bad event: %#v", e)
}
if e["Name"].(string) != "deploy" {
t.Fatalf("bad event: %#v", e)
}
if bytes.Compare(e["Payload"].([]byte), []byte("foo")) != 0 {
t.Fatalf("bad event: %#v", e)
}
if e["Coalesce"].(bool) != false {
t.Fatalf("bad event: %#v", e)
}
default:
t.Fatalf("should have event")
}
}
func TestRPCClientMonitor(t *testing.T) {
client, a1, ipc := testRPCClient(t)
defer ipc.Shutdown()
defer client.Close()
defer a1.Shutdown()
if err := a1.Start(); err != nil {
t.Fatalf("err: %s", err)
}
eventCh := make(chan string, 64)
if handle, err := client.Monitor("debug", eventCh); err != nil {
t.Fatalf("err: %s", err)
} else {
defer client.Stop(handle)
}
testutil.Yield()
select {
case e := <-eventCh:
if !strings.Contains(e, "Accepted client") {
t.Fatalf("bad: %s", e)
}
default:
t.Fatalf("should have backlog")
}
// Drain the rest of the messages as we know it
drainEventCh(eventCh)
// Join a bad thing to generate more events
a1.Join(nil, false)
testutil.Yield()
select {
case e := <-eventCh:
if !strings.Contains(e, "joining") {
t.Fatalf("bad: %s", e)
}
default:
t.Fatalf("should have message")
}
}
func TestRPCClientStream_Member(t *testing.T) {
client, a1, ipc := testRPCClient(t)
defer ipc.Shutdown()
defer client.Close()
defer a1.Shutdown()
a2 := testAgent(nil)
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()
eventCh := make(chan map[string]interface{}, 64)
if handle, err := client.Stream("*", eventCh); err != nil {
t.Fatalf("err: %s", err)
} else {
defer client.Stop(handle)
}
testutil.Yield()
s2Addr := a2.conf.MemberlistConfig.BindAddr
if _, err := a1.Join([]string{s2Addr}, false); err != nil {
t.Fatalf("err: %s", err)
}
testutil.Yield()
select {
case e := <-eventCh:
if e["Event"].(string) != "member-join" {
t.Fatalf("bad event: %#v", e)
}
members := e["Members"].([]interface{})
if len(members) != 1 {
t.Fatalf("should have 1 member")
}
member := members[0].(map[interface{}]interface{})
if _, ok := member["Name"].(string); !ok {
t.Fatalf("bad event: %#v", e)
}
if _, ok := member["Addr"].([]uint8); !ok {
t.Fatalf("bad event: %#v", e)
}
if _, ok := member["Port"].(uint64); !ok {
t.Fatalf("bad event: %#v", e)
}
if _, ok := member["Tags"].(map[interface{}]interface{}); !ok {
t.Fatalf("bad event: %#v", e)
}
if stat, _ := member["Status"].(string); stat != "alive" {
t.Fatalf("bad event: %#v", e)
}
if _, ok := member["ProtocolMin"].(int64); !ok {
t.Fatalf("bad event: %#v", e)
}
if _, ok := member["ProtocolMax"].(int64); !ok {
t.Fatalf("bad event: %#v", e)
}
if _, ok := member["ProtocolCur"].(int64); !ok {
t.Fatalf("bad event: %#v", e)
}
if _, ok := member["DelegateMin"].(int64); !ok {
t.Fatalf("bad event: %#v", e)
}
if _, ok := member["DelegateMax"].(int64); !ok {
t.Fatalf("bad event: %#v", e)
}
if _, ok := member["DelegateCur"].(int64); !ok {
t.Fatalf("bad event: %#v", e)
}
default:
t.Fatalf("should have event")
}
}