enhanced the bulletin code

bulletin_test.go

@@ -2,26 +2,70 @@ package hodu_test
 
 import "fmt"
 import "hodu"
+import "sync"
 import "testing"
+import "time"
 
 func TestBulletin(t *testing.T) {
-	var b *hodu.Bulletin
-	var s1 *hodu.BulletinSubscription
-	var s2 *hodu.BulletinSubscription
+	var b *hodu.Bulletin[string]
+	var s1 *hodu.BulletinSubscription[string]
+	var s2 *hodu.BulletinSubscription[string]
+	var wg sync.WaitGroup
+	var nmsgs1 int
+	var nmsgs2 int
 
-	b = hodu.NewBulletin()
+	b = hodu.NewBulletin[string]()
 
-	s1 = b.Subscribe("t1")
-	s2 = b.Subscribe("t2")
+	s1, _ = b.Subscribe("t1")
+	s2, _ = b.Subscribe("t2")
 
+	wg.Add(1)
 	go func() {
-		fmt.Printf ("s1: %+v\n", s1.Receive())
-	}()
+		var m string
+		var ok bool
+		var c1 chan string
+		var c2 chan string
+
+		c1 = s1.C
+		c2 = s2.C
+
+		defer wg.Done()
+		for c1 != nil || c2 != nil {
+			select {
+				case m, ok = <-c1:
+					if ok { fmt.Printf ("s1: %+v\n", m); nmsgs1++ } else { c1 = nil; fmt.Printf ("s1 closed\n")}
+
+				case m, ok = <-c2:
+					if ok { fmt.Printf ("s2: %+v\n", m); nmsgs2++ } else { c2 = nil; fmt.Printf ("s2 closed\n") }
+			}
+
+		}
 
-	go func() {
-		fmt.Printf ("s2: %+v\n", s2.Receive())
 	}()
 
 	b.Publish("t1", "donkey")
+	b.Publish("t2", "monkey")
+	b.Publish("t1", "donkey kong")
+	b.Publish("t2", "monkey hong")
+	b.Publish("t3", "home")
+	b.Publish("t2", "fire")
+	b.Publish("t1", "sunflower")
+	b.Publish("t2", "itsy bitsy spider")
+	b.Publish("t3", "marigold")
+	b.Publish("t3", "parrot")
+	time.Sleep(100 * time.Millisecond)
+	b.Publish("t2", "tiger")
+	time.Sleep(100 * time.Millisecond)
+	b.Unsubscribe(s2)
+	b.Publish("t2", "lion king")
+	b.Publish("t2", "fly to the skyp")
+	time.Sleep(100 * time.Millisecond)
+
+	b.Close()
+	wg.Wait()
+	fmt.Printf ("---------------------\n")
+
+	if nmsgs1 != 3 { t.Errorf("number of messages for s1 received must be 3, but got %d\n", nmsgs1) }
+	if nmsgs2 != 5 { t.Errorf("number of messages for s2 received must be 5, but got %d\n", nmsgs2) }
 }