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
101
102
| package serf
|
| import (
| "reflect"
| "testing"
| "time"
| )
|
| func TestUserEventCoalesce_Basic(t *testing.T) {
| outCh := make(chan Event, 64)
| shutdownCh := make(chan struct{})
| defer close(shutdownCh)
|
| c := &userEventCoalescer{
| events: make(map[string]*latestUserEvents),
| }
|
| inCh := coalescedEventCh(outCh, shutdownCh,
| 5*time.Millisecond, 5*time.Millisecond, c)
|
| send := []Event{
| UserEvent{
| LTime: 1,
| Name: "foo",
| Coalesce: true,
| },
| UserEvent{
| LTime: 2,
| Name: "foo",
| Coalesce: true,
| },
| UserEvent{
| LTime: 2,
| Name: "bar",
| Payload: []byte("test1"),
| Coalesce: true,
| },
| UserEvent{
| LTime: 2,
| Name: "bar",
| Payload: []byte("test2"),
| Coalesce: true,
| },
| }
|
| for _, e := range send {
| inCh <- e
| }
|
| var gotFoo, gotBar1, gotBar2 bool
| timeout := time.After(10 * time.Millisecond)
| USEREVENTFORLOOP:
| for {
| select {
| case e := <-outCh:
| ue := e.(UserEvent)
| switch ue.Name {
| case "foo":
| if ue.LTime != 2 {
| t.Fatalf("bad ltime for foo %#v", ue)
| }
| gotFoo = true
| case "bar":
| if ue.LTime != 2 {
| t.Fatalf("bad ltime for bar %#v", ue)
| }
| if reflect.DeepEqual(ue.Payload, []byte("test1")) {
| gotBar1 = true
| }
| if reflect.DeepEqual(ue.Payload, []byte("test2")) {
| gotBar2 = true
| }
| }
| case <-timeout:
| break USEREVENTFORLOOP
| }
| }
|
| if !gotFoo || !gotBar1 || !gotBar2 {
| t.Fatalf("missing messages %v %v %v", gotFoo, gotBar1, gotBar2)
| }
| }
|
| func TestUserEventCoalesce_passThrough(t *testing.T) {
| cases := []struct {
| e Event
| handle bool
| }{
| {UserEvent{Coalesce: false}, false},
| {UserEvent{Coalesce: true}, true},
| {MemberEvent{Type: EventMemberJoin}, false},
| {MemberEvent{Type: EventMemberLeave}, false},
| {MemberEvent{Type: EventMemberFailed}, false},
| }
|
| for _, tc := range cases {
| c := &userEventCoalescer{}
| if tc.handle != c.Handle(tc.e) {
| t.Fatalf("bad: %#v", tc.e)
| }
| }
| }
|
|
