3434import static org .mockito .Mockito .verifyNoMoreInteractions ;
3535import static org .mockito .Mockito .when ;
3636
37+ import quickfix .Log ;
38+ import quickfix .Session ;
39+ import quickfix .SessionID ;
3740
3841public class IoSessionResponderTest {
3942
43+ private final SessionID sessionID = new SessionID ("FIX.4.4:SENDER->TARGET" );
44+
4045 @ Test
4146 public void testAsynchronousSend () throws Exception {
4247 IoSession mockIoSession = mock (IoSession .class );
@@ -48,6 +53,7 @@ public void testAsynchronousSend() throws Exception {
4853 boolean result = responder .send ("abcd" );
4954
5055 assertTrue (result );
56+ verify (mockIoSession ).getAttribute (SessionConnector .QF_SESSION );
5157 verify (mockIoSession ).write ("abcd" );
5258 verifyNoMoreInteractions (mockWriteFuture );
5359 verifyNoMoreInteractions (mockIoSession );
@@ -65,6 +71,7 @@ public void testSynchronousSend() throws Exception {
6571 boolean result = responder .send ("abcd" );
6672
6773 assertTrue (result );
74+ verify (mockIoSession ).getAttribute (SessionConnector .QF_SESSION );
6875 verify (mockIoSession ).write ("abcd" );
6976 verify (mockWriteFuture ).awaitUninterruptibly (timeout );
7077 verifyNoMoreInteractions (mockWriteFuture );
@@ -75,7 +82,10 @@ public void testSynchronousSend() throws Exception {
7582 public void testSynchronousSendWithJoinException () throws Exception {
7683 int timeout = 123 ;
7784 IoSession mockIoSession = mock (IoSession .class );
78-
85+ Session mockSession = mock (Session .class );
86+ when (mockIoSession .getAttribute (SessionConnector .QF_SESSION )).thenReturn (mockSession );
87+ when (mockSession .getSessionID ()).thenReturn (sessionID );
88+
7989 WriteFuture mockWriteFuture = mock (WriteFuture .class );
8090 when (mockIoSession .write ("abcd" )).thenReturn (mockWriteFuture );
8191 doThrow (new RuntimeException ("TEST" )).when (mockWriteFuture ).awaitUninterruptibly (timeout );
@@ -84,6 +94,7 @@ public void testSynchronousSendWithJoinException() throws Exception {
8494 boolean result = responder .send ("abcd" );
8595
8696 assertFalse (result );
97+ verify (mockIoSession ).getAttribute (SessionConnector .QF_SESSION );
8798 verify (mockIoSession ).write ("abcd" );
8899 verify (mockWriteFuture ).awaitUninterruptibly (timeout );
89100 verifyNoMoreInteractions (mockWriteFuture );
@@ -94,6 +105,11 @@ public void testSynchronousSendWithJoinException() throws Exception {
94105 public void testSynchronousSendWithJoinTimeout () throws Exception {
95106 int timeout = 123 ;
96107 IoSession mockIoSession = mock (IoSession .class );
108+ Session mockSession = mock (Session .class );
109+ Log log = mock (Log .class );
110+ when (mockIoSession .getAttribute (SessionConnector .QF_SESSION )).thenReturn (mockSession );
111+ when (mockSession .getSessionID ()).thenReturn (sessionID );
112+ when (mockSession .getLog ()).thenReturn (log );
97113
98114 WriteFuture mockWriteFuture = mock (WriteFuture .class );
99115 when (mockIoSession .write ("abcd" )).thenReturn (mockWriteFuture );
@@ -103,6 +119,7 @@ public void testSynchronousSendWithJoinTimeout() throws Exception {
103119 boolean result = responder .send ("abcd" );
104120
105121 assertFalse (result );
122+ verify (mockIoSession ).getAttribute (SessionConnector .QF_SESSION );
106123 verify (mockIoSession ).write ("abcd" );
107124 verify (mockWriteFuture ).awaitUninterruptibly (timeout );
108125 verifyNoMoreInteractions (mockWriteFuture );
0 commit comments