9
9
)
10
10
11
11
type InSessionTestSuite struct {
12
- SessionSuite
12
+ SessionSuiteRig
13
13
}
14
14
15
15
func TestInSessionTestSuite (t * testing.T ) {
@@ -34,84 +34,84 @@ func (s *InSessionTestSuite) TestIsSessionTime() {
34
34
}
35
35
36
36
func (s * InSessionTestSuite ) TestLogout () {
37
- s .mockApp .On ("FromAdmin" ).Return (nil )
38
- s .mockApp .On ("ToAdmin" )
39
- s .mockApp .On ("OnLogout" )
37
+ s .MockApp .On ("FromAdmin" ).Return (nil )
38
+ s .MockApp .On ("ToAdmin" )
39
+ s .MockApp .On ("OnLogout" )
40
40
s .session .FixMsgIn (s .session , s .Logout ())
41
41
42
- s .mockApp .AssertExpectations (s .T ())
42
+ s .MockApp .AssertExpectations (s .T ())
43
43
s .State (latentState {})
44
44
45
45
s .LastToAdminMessageSent ()
46
- s .MessageType (enum .MsgType_LOGOUT , s .mockApp .lastToAdmin )
46
+ s .MessageType (enum .MsgType_LOGOUT , s .MockApp .lastToAdmin )
47
47
s .NextTargetMsgSeqNum (2 )
48
48
s .NextSenderMsgSeqNum (2 )
49
49
}
50
50
51
51
func (s * InSessionTestSuite ) TestLogoutResetOnLogout () {
52
52
s .session .resetOnLogout = true
53
53
54
- s .mockApp .On ("ToApp" ).Return (nil )
54
+ s .MockApp .On ("ToApp" ).Return (nil )
55
55
s .Nil (s .queueForSend (s .NewOrderSingle ()))
56
- s .mockApp .AssertExpectations (s .T ())
56
+ s .MockApp .AssertExpectations (s .T ())
57
57
58
- s .mockApp .On ("FromAdmin" ).Return (nil )
59
- s .mockApp .On ("ToAdmin" )
60
- s .mockApp .On ("OnLogout" )
58
+ s .MockApp .On ("FromAdmin" ).Return (nil )
59
+ s .MockApp .On ("ToAdmin" )
60
+ s .MockApp .On ("OnLogout" )
61
61
s .session .FixMsgIn (s .session , s .Logout ())
62
62
63
- s .mockApp .AssertExpectations (s .T ())
63
+ s .MockApp .AssertExpectations (s .T ())
64
64
s .State (latentState {})
65
65
s .LastToAppMessageSent ()
66
66
s .LastToAdminMessageSent ()
67
- s .MessageType (enum .MsgType_LOGOUT , s .mockApp .lastToAdmin )
67
+ s .MessageType (enum .MsgType_LOGOUT , s .MockApp .lastToAdmin )
68
68
69
69
s .NextTargetMsgSeqNum (1 )
70
70
s .NextSenderMsgSeqNum (1 )
71
71
s .NoMessageQueued ()
72
72
}
73
73
74
74
func (s * InSessionTestSuite ) TestTimeoutNeedHeartbeat () {
75
- s .mockApp .On ("ToAdmin" ).Return (nil )
75
+ s .MockApp .On ("ToAdmin" ).Return (nil )
76
76
s .session .Timeout (s .session , internal .NeedHeartbeat )
77
77
78
- s .mockApp .AssertExpectations (s .T ())
78
+ s .MockApp .AssertExpectations (s .T ())
79
79
s .State (inSession {})
80
80
s .LastToAdminMessageSent ()
81
- s .MessageType (enum .MsgType_HEARTBEAT , s .mockApp .lastToAdmin )
81
+ s .MessageType (enum .MsgType_HEARTBEAT , s .MockApp .lastToAdmin )
82
82
s .NextSenderMsgSeqNum (2 )
83
83
}
84
84
85
85
func (s * InSessionTestSuite ) TestTimeoutPeerTimeout () {
86
- s .mockApp .On ("ToAdmin" ).Return (nil )
86
+ s .MockApp .On ("ToAdmin" ).Return (nil )
87
87
s .session .Timeout (s .session , internal .PeerTimeout )
88
88
89
- s .mockApp .AssertExpectations (s .T ())
89
+ s .MockApp .AssertExpectations (s .T ())
90
90
s .State (pendingTimeout {inSession {}})
91
91
s .LastToAdminMessageSent ()
92
- s .MessageType (enum .MsgType_TEST_REQUEST , s .mockApp .lastToAdmin )
92
+ s .MessageType (enum .MsgType_TEST_REQUEST , s .MockApp .lastToAdmin )
93
93
s .NextSenderMsgSeqNum (2 )
94
94
}
95
95
96
96
func (s * InSessionTestSuite ) TestDisconnected () {
97
- s .mockApp .On ("OnLogout" ).Return (nil )
97
+ s .MockApp .On ("OnLogout" ).Return (nil )
98
98
s .session .Disconnected (s .session )
99
- s .mockApp .AssertExpectations (s .T ())
99
+ s .MockApp .AssertExpectations (s .T ())
100
100
s .State (latentState {})
101
101
}
102
102
103
103
func (s * InSessionTestSuite ) TestStop () {
104
- s .mockApp .On ("ToAdmin" )
104
+ s .MockApp .On ("ToAdmin" )
105
105
s .session .Stop (s .session )
106
106
107
- s .mockApp .AssertExpectations (s .T ())
107
+ s .MockApp .AssertExpectations (s .T ())
108
108
s .State (logoutState {})
109
109
s .LastToAdminMessageSent ()
110
- s .MessageType (enum .MsgType_LOGOUT , s .mockApp .lastToAdmin )
110
+ s .MessageType (enum .MsgType_LOGOUT , s .MockApp .lastToAdmin )
111
111
112
- s .mockApp .On ("OnLogout" )
112
+ s .MockApp .On ("OnLogout" )
113
113
s .session .Timeout (s .session , <- s .sessionEvent )
114
- s .mockApp .AssertExpectations (s .T ())
114
+ s .MockApp .AssertExpectations (s .T ())
115
115
s .Stopped ()
116
116
s .Disconnected ()
117
117
}
0 commit comments