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
78
79
80
81
82
83
84
85
86
87
|
package client
import (
"encoding/json"
"sync"
"sync/atomic"
"testing"
"github.com/stretchr/testify/assert"
)
func TestSubscriptionsSubscribe(t *testing.T) {
bufSize := 100
subs := newSubscriptions(bufSize)
var receivedNotifications atomic.Int32
var wg sync.WaitGroup
runSubNotify := func(sub *Subscription) {
wg.Add(1)
go func() {
defer wg.Done()
for i := 0; i < bufSize; i++ {
b, err := json.Marshal(i)
assert.Nil(t, err)
err = sub.notify(b)
assert.Nil(t, err)
}
}()
}
runSubRecv := func(sub *Subscription) {
wg.Add(1)
go func() {
defer wg.Done()
i := 0
for nt := range sub.Recv() {
var v int
err := json.Unmarshal(nt, &v)
assert.Nil(t, err)
assert.Equal(t, v, i)
receivedNotifications.Add(1)
i += 1
if i == bufSize {
break
}
}
sub.stop()
}()
}
wg.Add(1)
go func() {
defer wg.Done()
for i := 0; i < 3; i++ {
sub := subs.subscribe(i)
runSubNotify(sub)
runSubRecv(sub)
}
}()
wg.Wait()
assert.Equal(t, receivedNotifications.Load(), int32(bufSize*3))
}
func TestSubscriptionsUnsubscribe(t *testing.T) {
bufSize := 100
subs := newSubscriptions(bufSize)
var wg sync.WaitGroup
sub := subs.subscribe(1)
subs.unsubscribe(1)
_, ok := <-sub.Recv()
assert.False(t, ok)
b, err := json.Marshal(1)
assert.Nil(t, err)
err = sub.notify(b)
if assert.Error(t, err) {
assert.ErrorIs(t, err, subscriptionIsClosedErr)
}
wg.Wait()
}
|
