@Test public void testDoOneMessageNotifiesClientHealthMonitor() throws IOException { AcceptorImpl acceptorStub = mock(AcceptorImpl.class); ClientProtocolProcessor clientProtocolProcessor = mock(ClientProtocolProcessor.class); ServerConnection serverConnection = getServerConnection(clientProtocolProcessor, acceptorStub); serverConnection.doOneMessage(); ArgumentCaptor<ClientProxyMembershipID> clientProxyMembershipIDArgumentCaptor = ArgumentCaptor.forClass(ClientProxyMembershipID.class); verify(clientHealthMonitorMock).receivedPing(clientProxyMembershipIDArgumentCaptor.capture()); assertEquals("identity(localhost<ec>:0,connection=1", clientProxyMembershipIDArgumentCaptor.getValue().toString()); }
@Test public void testEOFDoesNotCauseWarningMessage() throws IOException, IncompatibleVersionException { Socket socketMock = mock(Socket.class); when(socketMock.getInetAddress()).thenReturn(InetAddress.getByName("localhost")); when(socketMock.isClosed()).thenReturn(true); ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); when(socketMock.getOutputStream()).thenReturn(outputStream); AcceptorImpl acceptorStub = mock(AcceptorImpl.class); ClientProtocolProcessor clientProtocolProcessor = mock(ClientProtocolProcessor.class); doThrow(new IOException()).when(clientProtocolProcessor).processMessage(any(), any()); ServerConnection serverConnection = getServerConnection(socketMock, clientProtocolProcessor, acceptorStub); String expectedMessage = "invoking doOneMessage"; String unexpectedMessage = "IOException"; // Create some stdout content so we can tell that the capture worked. System.out.println(expectedMessage); serverConnection.doOneMessage(); // verify that an IOException wasn't logged String stdoutCapture = systemOutRule.getLog(); assertTrue(stdoutCapture.contains(expectedMessage)); assertFalse(stdoutCapture.contains(unexpectedMessage)); }
@Test public void testProcessFlag() throws Exception { ServerConnection serverConnection = IOExceptionThrowingServerConnection(); Assert.assertTrue(serverConnection.processMessages); serverConnection.doOneMessage(); Assert.assertTrue(!serverConnection.processMessages); }
if (!isTerminated()) { getAcceptor().setTLCommBuffer(); doOneMessage(); if (this.processMessages && !(this.crHelper.isShutdown())) { while (this.processMessages && !(this.crHelper.isShutdown())) { try { doOneMessage(); } catch (CancelException e) {