test: broadcast (#94)

waku/v2/node/broadcast_test.go

@@ -0,0 +1,46 @@
+package node
+
+import (
+	"sync"
+	"testing"
+
+	"github.com/status-im/go-waku/waku/v2/protocol"
+	"github.com/stretchr/testify/require"
+)
+
+// Adapted from https://github.com/dustin/go-broadcast/commit/f664265f5a662fb4d1df7f3533b1e8d0e0277120
+// by Dustin Sallings (c) 2013, which was released under MIT license
+
+func TestBroadcast(t *testing.T) {
+	wg := sync.WaitGroup{}
+
+	b := NewBroadcaster(100)
+	defer b.Close()
+
+	for i := 0; i < 5; i++ {
+		wg.Add(1)
+
+		cch := make(chan *protocol.Envelope)
+
+		b.Register(cch)
+
+		go func() {
+			defer wg.Done()
+			defer b.Unregister(cch)
+			<-cch
+		}()
+
+	}
+
+	env := new(protocol.Envelope)
+	b.Submit(env)
+
+	wg.Wait()
+}
+
+func TestBroadcastCleanup(t *testing.T) {
+	b := NewBroadcaster(100)
+	b.Register(make(chan *protocol.Envelope))
+	err := b.Close()
+	require.NoError(t, err)
+}