/
relay_internal_test.go
95 lines (87 loc) · 2.29 KB
/
relay_internal_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
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
package tchannel
import (
"testing"
"time"
"github.com/uber/tchannel-go/typed"
"github.com/stretchr/testify/assert"
)
func TestFinishesCallResponses(t *testing.T) {
tests := []struct {
msgType messageType
flags byte
finishesCall bool
}{
{messageTypeCallRes, 0x00, true},
{messageTypeCallRes, 0x01, false},
{messageTypeCallRes, 0x02, true},
{messageTypeCallRes, 0x03, false},
{messageTypeCallRes, 0x04, true},
{messageTypeCallResContinue, 0x00, true},
{messageTypeCallResContinue, 0x01, false},
{messageTypeCallResContinue, 0x02, true},
{messageTypeCallResContinue, 0x03, false},
{messageTypeCallResContinue, 0x04, true},
// By definition, callreq should never terminate an RPC.
{messageTypeCallReq, 0x00, false},
{messageTypeCallReq, 0x01, false},
{messageTypeCallReq, 0x02, false},
{messageTypeCallReq, 0x03, false},
{messageTypeCallReq, 0x04, false},
}
for _, tt := range tests {
f := NewFrame(100)
fh := FrameHeader{
size: uint16(0xFF34),
messageType: tt.msgType,
ID: 0xDEADBEEF,
}
f.Header = fh
fh.write(typed.NewWriteBuffer(f.headerBuffer))
payload := typed.NewWriteBuffer(f.Payload)
payload.WriteSingleByte(tt.flags)
assert.Equal(t, tt.finishesCall, finishesCall(f), "Wrong isLast for flags %v and message type %v", tt.flags, tt.msgType)
}
}
func TestRelayTimerPoolMisuse(t *testing.T) {
tests := []struct {
msg string
f func(*relayTimer)
}{
{
msg: "release without stop",
f: func(rt *relayTimer) {
rt.Start(time.Hour, &relayItems{}, 0, false /* isOriginator */)
rt.Release()
},
},
{
msg: "start twice",
f: func(rt *relayTimer) {
rt.Start(time.Hour, &relayItems{}, 0, false /* isOriginator */)
rt.Start(time.Hour, &relayItems{}, 0, false /* isOriginator */)
},
},
{
msg: "underlying timer is already active",
f: func(rt *relayTimer) {
rt.timer.Reset(time.Hour)
rt.Start(time.Hour, &relayItems{}, 0, false /* isOriginator */)
},
},
{
msg: "use timer after releasing it",
f: func(rt *relayTimer) {
rt.Release()
rt.Stop()
},
},
}
for _, tt := range tests {
trigger := func(*relayItems, uint32, bool) {}
rtp := newRelayTimerPool(trigger, true /* verify */)
rt := rtp.Get()
assert.Panics(t, func() {
tt.f(rt)
}, tt.msg)
}
}