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
88
89
90
91
92
93
94
95
96
97
98
99
100
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
package sse
import (
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
// Tests are accessing subscriber in a non-threadsafe way.
// Maybe fix this in the future so we can test with -race enabled
func TestStreamAddSubscriber(t *testing.T) {
s := newStream("test", 1024, true, false, nil, nil)
s.run()
defer s.close()
s.event <- &Event{Data: []byte("test")}
sub := s.addSubscriber(0, nil)
assert.Equal(t, 1, s.getSubscriberCount())
s.event <- &Event{Data: []byte("test")}
msg, err := wait(sub.connection, time.Second*1)
require.Nil(t, err)
assert.Equal(t, []byte(`test`), msg)
assert.Equal(t, 0, len(sub.connection))
}
func TestStreamRemoveSubscriber(t *testing.T) {
s := newStream("test", 1024, true, false, nil, nil)
s.run()
defer s.close()
sub := s.addSubscriber(0, nil)
time.Sleep(time.Millisecond * 100)
s.deregister <- sub
time.Sleep(time.Millisecond * 100)
assert.Equal(t, 0, s.getSubscriberCount())
}
func TestStreamSubscriberClose(t *testing.T) {
s := newStream("test", 1024, true, false, nil, nil)
s.run()
defer s.close()
sub := s.addSubscriber(0, nil)
sub.close()
time.Sleep(time.Millisecond * 100)
assert.Equal(t, 0, s.getSubscriberCount())
}
func TestStreamDisableAutoReplay(t *testing.T) {
s := newStream("test", 1024, true, false, nil, nil)
s.run()
defer s.close()
s.AutoReplay = false
s.event <- &Event{Data: []byte("test")}
time.Sleep(time.Millisecond * 100)
sub := s.addSubscriber(0, nil)
assert.Equal(t, 0, len(sub.connection))
}
func TestStreamMultipleSubscribers(t *testing.T) {
var subs []*Subscriber
s := newStream("test", 1024, true, false, nil, nil)
s.run()
for i := 0; i < 10; i++ {
subs = append(subs, s.addSubscriber(0, nil))
}
// Wait for all subscribers to be added
time.Sleep(time.Millisecond * 100)
s.event <- &Event{Data: []byte("test")}
for _, sub := range subs {
msg, err := wait(sub.connection, time.Second*1)
require.Nil(t, err)
assert.Equal(t, []byte(`test`), msg)
}
s.close()
// Wait for all subscribers to close
time.Sleep(time.Millisecond * 100)
assert.Equal(t, 0, s.getSubscriberCount())
}
|
