pubsub/pubsub_test.go at main · ilij4/pubsub · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
package pubsub

import (
	"testing"
	"time"

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

const (
	busId = "test-bus"
	value = "test"
)

func TestNew(t *testing.T) {
	bus := New()
	assert.NotNil(t, bus)
}

func TestSubscribe(t *testing.T) {
	bus := New()
	bus.Subscribe(value, func() {})
}

func TestSubscribeOnce(t *testing.T) {
	bus := New()
	bus.SubscribeOnce(value, func() {})
}

func TestPublish(t *testing.T) {
	bus := New()
	bus.Publish(value, "test")
}

func TestSubscribeOnceAsync(t *testing.T) {
	bus := New()
	bus.SubscribeOnceAsync(value, func() {})
}

func TestSubscribeAsync(t *testing.T) {
	bus := New()
	bus.SubscribeAsync(value, func() {})
}

func TestSimplePubAndSub(t *testing.T) {
	bus := New()
	handler := func(s string) {
		t.Log("handler called got value: ", s)
		assert.Equal(t, value, s)
	}
	bus.Subscribe(busId, handler)
	bus.Publish(busId, value)
}

func TestPublishAsync(t *testing.T) {
	bus := New()
	handler := func(s string) {
		t.Log("handler called got value: ", s)
		assert.Equal(t, value, s)
	}
	sub, _ := bus.SubscribeAsync(busId, handler)
	bus.Publish(busId, value)
	bus.WaitAsync()
	sub.Unsubscribe()
}

func TestPublishRetain(t *testing.T) {
	bus := New()
	bus.PublishRetain(busId, value)
	handler := func(s string) {
		t.Log("handler called got value: ", s)
		assert.Equal(t, value, s)
	}
	bus.Subscribe(busId, handler)
	bus.WaitAsync()
	time.Sleep(1 * time.Second)
}