From efef0ac671a376286acfc9a8c4f22ec8cbff06c3 Mon Sep 17 00:00:00 2001 From: "U-IGI\\gibbsd" Date: Fri, 6 Oct 2017 21:19:19 +0100 Subject: [PATCH] change implementation to max scheduled write bytes --- .../java/quickfix/mina/IoSessionResponderTest.java | 30 ++++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/quickfixj-core/src/test/java/quickfix/mina/IoSessionResponderTest.java b/quickfixj-core/src/test/java/quickfix/mina/IoSessionResponderTest.java index cb680c3d..0062de11 100644 --- a/quickfixj-core/src/test/java/quickfix/mina/IoSessionResponderTest.java +++ b/quickfixj-core/src/test/java/quickfix/mina/IoSessionResponderTest.java @@ -20,9 +20,12 @@ package quickfix.mina; import junit.framework.TestCase; +import quickfix.Session; + import org.apache.mina.core.future.WriteFuture; import org.apache.mina.core.session.IoSession; +import java.io.IOException; import java.net.InetSocketAddress; import static org.mockito.Mockito.*; @@ -121,4 +124,31 @@ public class IoSessionResponderTest extends TestCase { verify(mockProtocolSession).getRemoteAddress(); verifyNoMoreInteractions(mockProtocolSession); } + + public void testMaxScheduledWriteBtyesExceeded() throws IOException { + IoSession mockProtocolSession = mock(IoSession.class); + Session session = mock(Session.class); + stub(mockProtocolSession.getAttribute(SessionConnector.QF_SESSION)).toReturn(session); + stub(mockProtocolSession.getScheduledWriteBytes()).toReturn(10L); + + IoSessionResponder responder = new IoSessionResponder(mockProtocolSession, false, 0, 5); + String data = "Should fail"; + assertFalse(responder.send(data)); + + verify(session).disconnect(anyString(), eq(true)); + } + + public void testMaxScheduledWriteBtyesNotExceeded() throws IOException { + IoSession mockProtocolSession = mock(IoSession.class); + Session session = mock(Session.class); + stub(mockProtocolSession.getAttribute(SessionConnector.QF_SESSION)).toReturn(session); + stub(mockProtocolSession.getScheduledWriteBytes()).toReturn(10L); + + IoSessionResponder responder = new IoSessionResponder(mockProtocolSession, false, 0, 50); + String data = "Should succeed"; + assertTrue(responder.send(data)); + + verify(mockProtocolSession).write(data); + } + } -- 2.14.1