waku-org/go-waku

package history

import (
    "sync"
    "testing"

    "github.com/stretchr/testify/require"
)

func TestEmitter(t *testing.T) {
    emitter := NewEmitter[int]()

    subscr1 := emitter.Subscribe()
    subscr2 := emitter.Subscribe()

    wg := sync.WaitGroup{}
    wg.Add(3)

    go func() {
        defer wg.Done()
        emitter.Emit(1)
        emitter.Emit(2)
    }()

    go func() {
        defer wg.Done()
        require.Equal(t, 1, <-subscr1)
        require.Equal(t, 2, <-subscr1)
    }()

    go func() {
        defer wg.Done()
        require.Equal(t, 1, <-subscr2)
        require.Equal(t, 2, <-subscr2)
    }()

    wg.Wait()
}

func TestOneShotEmitter(t *testing.T) {
    emitter := NewOneshotEmitter[struct{}]()

    subscr1 := emitter.Subscribe()
    subscr2 := emitter.Subscribe()

    wg := sync.WaitGroup{}
    wg.Add(3)

    go func() {
        defer wg.Done()
        emitter.Emit(struct{}{})
    }()

    go func() {
        defer wg.Done()
        for range subscr1 {
        }
    }()

    go func() {
        defer wg.Done()
        for range subscr2 {
        }
    }()

    wg.Wait()
}