@Test public void testAwaitInputInBuffer() throws Exception { final ByteArrayInputStream instream = Mockito.spy(new ByteArrayInputStream(new byte[] {1, 2, 3, 4, 5})); Mockito.when(socket.getInputStream()).thenReturn(instream); conn.bind(socket); conn.ensureOpen(); conn.getSessionInputBuffer().read(); Assert.assertTrue(conn.awaitInput(432)); Mockito.verify(socket, Mockito.never()).setSoTimeout(Mockito.anyInt()); Mockito.verify(instream, Mockito.times(1)) .read(Mockito.<byte[]>any(), Mockito.anyInt(), Mockito.anyInt()); }