forked from flier/curator.go
/
watcher_test.go
68 lines (45 loc) · 1.25 KB
/
watcher_test.go
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
package curator
import (
"runtime"
"testing"
"time"
"github.com/samuel/go-zookeeper/zk"
"github.com/stretchr/testify/assert"
)
func TestWatchers(t *testing.T) {
var events [3][]*zk.Event
w := NewWatchers(NewWatcher(func(event *zk.Event) {
events[0] = append(events[0], event)
}))
w1 := w.Add(NewWatcher(func(event *zk.Event) {
events[1] = append(events[1], event)
}))
w2 := w.Add(NewWatcher(func(event *zk.Event) {
events[2] = append(events[2], event)
}))
assert.Equal(t, w1, w.watchers[1])
assert.Equal(t, w2, w.watchers[2])
c := make(chan zk.Event)
go w.Watch(c)
evt := zk.Event{}
c <- evt
time.Sleep(100 * time.Millisecond)
close(c)
assert.Equal(t, []*zk.Event{&evt}, events[0])
assert.Equal(t, []*zk.Event{&evt}, events[1])
assert.Equal(t, []*zk.Event{&evt}, events[2])
// remove watcher and fire event again
assert.Equal(t, w.Remove(w1), w1)
assert.Equal(t, w.Remove(w2), w2)
assert.Equal(t, 1, len(w.watchers))
c = make(chan zk.Event)
go w.Watch(c)
evt = zk.Event{}
c <- evt
runtime.Gosched()
close(c)
assert.Equal(t, 2, len(events[0]))
assert.Equal(t, 1, len(events[1]))
assert.Equal(t, 1, len(events[2]))
assert.Equal(t, &evt, events[0][1])
}