agent
74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestHub_BroadcastReachesSubscriber(t *testing.T) {
|
|
h := NewHub()
|
|
ch := h.Subscribe()
|
|
defer h.Unsubscribe(ch)
|
|
|
|
want := ServerEvent{Board: "issues", CardID: "0119", Action: "updated"}
|
|
h.Broadcast(want)
|
|
|
|
select {
|
|
case got := <-ch:
|
|
if got != want {
|
|
t.Fatalf("got %+v, want %+v", got, want)
|
|
}
|
|
case <-time.After(time.Second):
|
|
t.Fatal("timeout waiting for broadcast")
|
|
}
|
|
}
|
|
|
|
func TestHub_UnsubscribeStopsDelivery(t *testing.T) {
|
|
h := NewHub()
|
|
ch := h.Subscribe()
|
|
if got := h.Count(); got != 1 {
|
|
t.Fatalf("Count() = %d, want 1", got)
|
|
}
|
|
h.Unsubscribe(ch)
|
|
if got := h.Count(); got != 0 {
|
|
t.Fatalf("Count() after Unsubscribe = %d, want 0", got)
|
|
}
|
|
// channel should be closed
|
|
if _, ok := <-ch; ok {
|
|
t.Fatalf("expected closed channel after Unsubscribe")
|
|
}
|
|
// double-unsubscribe is a no-op
|
|
h.Unsubscribe(ch)
|
|
|
|
// broadcast should not panic and should reach nobody
|
|
h.Broadcast(ServerEvent{Board: "issues", CardID: "x", Action: "updated"})
|
|
}
|
|
|
|
func TestHub_MultipleSubscribersAllReceive(t *testing.T) {
|
|
h := NewHub()
|
|
const n = 5
|
|
chans := make([]chan ServerEvent, n)
|
|
for i := range chans {
|
|
chans[i] = h.Subscribe()
|
|
}
|
|
defer func() {
|
|
for _, ch := range chans {
|
|
h.Unsubscribe(ch)
|
|
}
|
|
}()
|
|
|
|
want := ServerEvent{Board: "flows", CardID: "abc", Action: "created", EventType: "card_added"}
|
|
h.Broadcast(want)
|
|
|
|
for i, ch := range chans {
|
|
select {
|
|
case got := <-ch:
|
|
if got != want {
|
|
t.Fatalf("sub %d: got %+v, want %+v", i, got, want)
|
|
}
|
|
case <-time.After(time.Second):
|
|
t.Fatalf("sub %d: timeout", i)
|
|
}
|
|
}
|
|
}
|