adding quickfix
This commit is contained in:
Ramiro Paz
221
quickfix/resend_state_test.go
Normal file
221
quickfix/resend_state_test.go
Normal file
@ -0,0 +1,221 @@
|
||||
// Copyright (c) quickfixengine.org All rights reserved.
|
||||
//
|
||||
// This file may be distributed under the terms of the quickfixengine.org
|
||||
// license as defined by quickfixengine.org and appearing in the file
|
||||
// LICENSE included in the packaging of this file.
|
||||
//
|
||||
// This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING
|
||||
// THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A
|
||||
// PARTICULAR PURPOSE.
|
||||
//
|
||||
// See http://www.quickfixengine.org/LICENSE for licensing information.
|
||||
//
|
||||
// Contact ask@quickfixengine.org if any conditions of this licensing
|
||||
// are not clear to you.
|
||||
|
||||
package quickfix
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
"quantex.com/qfixdpl/quickfix/internal"
|
||||
)
|
||||
|
||||
type resendStateTestSuite struct {
|
||||
SessionSuiteRig
|
||||
}
|
||||
|
||||
func TestResendStateTestSuite(t *testing.T) {
|
||||
suite.Run(t, new(resendStateTestSuite))
|
||||
}
|
||||
|
||||
func (s *resendStateTestSuite) SetupTest() {
|
||||
s.Init()
|
||||
s.session.State = resendState{}
|
||||
}
|
||||
|
||||
func (s *resendStateTestSuite) TestIsLoggedOn() {
|
||||
s.True(s.session.IsLoggedOn())
|
||||
}
|
||||
|
||||
func (s *resendStateTestSuite) TestIsConnected() {
|
||||
s.True(s.session.IsConnected())
|
||||
}
|
||||
|
||||
func (s *resendStateTestSuite) TestIsSessionTime() {
|
||||
s.True(s.session.IsSessionTime())
|
||||
}
|
||||
|
||||
func (s *resendStateTestSuite) TestTimeoutPeerTimeout() {
|
||||
s.MockApp.On("ToAdmin")
|
||||
s.session.Timeout(s.session, internal.PeerTimeout)
|
||||
|
||||
s.MockApp.AssertExpectations(s.T())
|
||||
s.State(pendingTimeout{resendState{}})
|
||||
}
|
||||
|
||||
func (s *resendStateTestSuite) TestTimeoutUnchangedIgnoreLogonLogoutTimeout() {
|
||||
tests := []internal.Event{internal.LogonTimeout, internal.LogoutTimeout}
|
||||
|
||||
for _, event := range tests {
|
||||
s.session.Timeout(s.session, event)
|
||||
s.State(resendState{})
|
||||
}
|
||||
}
|
||||
|
||||
func (s *resendStateTestSuite) TestTimeoutUnchangedNeedHeartbeat() {
|
||||
s.MockApp.On("ToAdmin")
|
||||
s.session.Timeout(s.session, internal.NeedHeartbeat)
|
||||
|
||||
s.MockApp.AssertExpectations(s.T())
|
||||
s.State(resendState{})
|
||||
}
|
||||
|
||||
func (s *resendStateTestSuite) TestFixMsgIn() {
|
||||
s.session.State = inSession{}
|
||||
|
||||
// In session expects seq number 1, send too high.
|
||||
s.MessageFactory.SetNextSeqNum(2)
|
||||
s.MockApp.On("ToAdmin")
|
||||
|
||||
msgSeqNum2 := s.NewOrderSingle()
|
||||
s.fixMsgIn(s.session, msgSeqNum2)
|
||||
|
||||
s.MockApp.AssertExpectations(s.T())
|
||||
s.State(resendState{})
|
||||
s.LastToAdminMessageSent()
|
||||
s.MessageType(string(msgTypeResendRequest), s.MockApp.lastToAdmin)
|
||||
s.FieldEquals(tagBeginSeqNo, 1, s.MockApp.lastToAdmin.Body)
|
||||
s.NextTargetMsgSeqNum(1)
|
||||
|
||||
msgSeqNum3 := s.NewOrderSingle()
|
||||
s.fixMsgIn(s.session, msgSeqNum3)
|
||||
s.State(resendState{})
|
||||
s.NextTargetMsgSeqNum(1)
|
||||
|
||||
msgSeqNum4 := s.NewOrderSingle()
|
||||
s.fixMsgIn(s.session, msgSeqNum4)
|
||||
|
||||
s.State(resendState{})
|
||||
s.NextTargetMsgSeqNum(1)
|
||||
|
||||
s.MessageFactory.SetNextSeqNum(1)
|
||||
s.MockApp.On("FromApp").Return(nil)
|
||||
s.fixMsgIn(s.session, s.NewOrderSingle())
|
||||
|
||||
s.MockApp.AssertNumberOfCalls(s.T(), "FromApp", 4)
|
||||
s.State(inSession{})
|
||||
s.NextTargetMsgSeqNum(5)
|
||||
}
|
||||
|
||||
func (s *resendStateTestSuite) TestFixMsgInSequenceReset() {
|
||||
s.session.State = inSession{}
|
||||
|
||||
// In session expects seq number 1, send too high.
|
||||
s.MessageFactory.SetNextSeqNum(3)
|
||||
s.MockApp.On("ToAdmin")
|
||||
|
||||
msgSeqNum3 := s.NewOrderSingle()
|
||||
s.fixMsgIn(s.session, msgSeqNum3)
|
||||
|
||||
s.MockApp.AssertExpectations(s.T())
|
||||
s.State(resendState{})
|
||||
s.LastToAdminMessageSent()
|
||||
s.MessageType(string(msgTypeResendRequest), s.MockApp.lastToAdmin)
|
||||
s.FieldEquals(tagBeginSeqNo, 1, s.MockApp.lastToAdmin.Body)
|
||||
s.NextTargetMsgSeqNum(1)
|
||||
|
||||
s.MessageFactory.SetNextSeqNum(1)
|
||||
s.MockApp.On("FromAdmin").Return(nil)
|
||||
s.fixMsgIn(s.session, s.SequenceReset(2))
|
||||
s.NextTargetMsgSeqNum(2)
|
||||
s.State(resendState{})
|
||||
|
||||
s.MockApp.On("FromApp").Return(nil)
|
||||
s.fixMsgIn(s.session, s.NewOrderSingle())
|
||||
|
||||
s.MockApp.AssertNumberOfCalls(s.T(), "FromApp", 2)
|
||||
s.NextTargetMsgSeqNum(4)
|
||||
s.State(inSession{})
|
||||
}
|
||||
|
||||
func (s *resendStateTestSuite) TestFixMsgInResendChunk() {
|
||||
s.session.State = inSession{}
|
||||
s.ResendRequestChunkSize = 2
|
||||
|
||||
// In session expects seq number 1, send too high.
|
||||
s.MessageFactory.SetNextSeqNum(4)
|
||||
s.MockApp.On("ToAdmin")
|
||||
|
||||
msgSeqNum4 := s.NewOrderSingle()
|
||||
s.fixMsgIn(s.session, msgSeqNum4)
|
||||
|
||||
s.MockApp.AssertExpectations(s.T())
|
||||
s.State(resendState{})
|
||||
s.LastToAdminMessageSent()
|
||||
s.MessageType(string(msgTypeResendRequest), s.MockApp.lastToAdmin)
|
||||
s.FieldEquals(tagBeginSeqNo, 1, s.MockApp.lastToAdmin.Body)
|
||||
s.FieldEquals(tagEndSeqNo, 2, s.MockApp.lastToAdmin.Body)
|
||||
s.NextTargetMsgSeqNum(1)
|
||||
|
||||
msgSeqNum5 := s.NewOrderSingle()
|
||||
s.fixMsgIn(s.session, msgSeqNum5)
|
||||
s.State(resendState{})
|
||||
s.NextTargetMsgSeqNum(1)
|
||||
|
||||
msgSeqNum6 := s.NewOrderSingle()
|
||||
s.fixMsgIn(s.session, msgSeqNum6)
|
||||
|
||||
s.State(resendState{})
|
||||
s.NextTargetMsgSeqNum(1)
|
||||
|
||||
s.MessageFactory.SetNextSeqNum(1)
|
||||
s.MockApp.On("FromApp").Return(nil)
|
||||
s.fixMsgIn(s.session, s.NewOrderSingle())
|
||||
|
||||
s.MockApp.AssertNumberOfCalls(s.T(), "FromApp", 1)
|
||||
s.State(resendState{})
|
||||
s.NextTargetMsgSeqNum(2)
|
||||
|
||||
s.fixMsgIn(s.session, s.NewOrderSingle())
|
||||
s.MockApp.AssertNumberOfCalls(s.T(), "FromApp", 2)
|
||||
s.State(resendState{})
|
||||
s.NextTargetMsgSeqNum(3)
|
||||
|
||||
s.LastToAdminMessageSent()
|
||||
s.MessageType(string(msgTypeResendRequest), s.MockApp.lastToAdmin)
|
||||
s.FieldEquals(tagBeginSeqNo, 3, s.MockApp.lastToAdmin.Body)
|
||||
s.FieldEquals(tagEndSeqNo, 0, s.MockApp.lastToAdmin.Body)
|
||||
}
|
||||
|
||||
// TestFixMsgResendWithOldSendingTime tests that we suspend staleness checks during replay
|
||||
// as a replayed message may be arbitrarily old.
|
||||
func (s *resendStateTestSuite) TestFixMsgResendWithOldSendingTime() {
|
||||
s.session.State = inSession{}
|
||||
s.ResendRequestChunkSize = 2
|
||||
|
||||
// In session expects seq number 1, send too high.
|
||||
s.MessageFactory.SetNextSeqNum(4)
|
||||
s.MockApp.On("ToAdmin")
|
||||
|
||||
msgSeqNum4 := s.NewOrderSingle()
|
||||
s.fixMsgIn(s.session, msgSeqNum4)
|
||||
|
||||
s.MockApp.AssertExpectations(s.T())
|
||||
s.State(resendState{})
|
||||
s.LastToAdminMessageSent()
|
||||
s.MessageType(string(msgTypeResendRequest), s.MockApp.lastToAdmin)
|
||||
s.FieldEquals(tagBeginSeqNo, 1, s.MockApp.lastToAdmin.Body)
|
||||
s.FieldEquals(tagEndSeqNo, 2, s.MockApp.lastToAdmin.Body)
|
||||
s.NextTargetMsgSeqNum(1)
|
||||
|
||||
msgSeqNum5 := s.NewOrderSingle()
|
||||
// Set the sending time far enough in the past to trip the staleness check.
|
||||
msgSeqNum5.Header.SetField(tagSendingTime, FIXUTCTimestamp{Time: time.Now().Add(-s.MaxLatency)})
|
||||
s.fixMsgIn(s.session, msgSeqNum5)
|
||||
s.State(resendState{})
|
||||
s.NextTargetMsgSeqNum(1)
|
||||
}
|
||||
Reference in New Issue
Block a user