Examples with TestInputStream com.unboundid.util.TestInputStream used on opensource projects

Example 1 with TestInputStream

use of com.unboundid.util.TestInputStream in project ldapsdk by pingidentity.

the class LDAPListenerClientConnectionTestCase method testThrowOnGetInputStream.

/**
 * Tests the behavior when a client connection has been created without a
 * listener and with a socket that throws an exception when trying to get the
 * input stream.
 *
 * @throws  Exception  If an unexpected problem occurs.
 */
@Test(expectedExceptions = { LDAPException.class })
public void testThrowOnGetInputStream() throws Exception {
    final TestSocket s = new TestSocket(new TestInputStream(new ByteArrayInputStream(new byte[1024]), new IOException("foo"), 100, true), new TestOutputStream(new ByteArrayOutputStream(), new IOException("bar"), 1024, true));
    s.setThrowOnGetInputStream(true);
    s.setThrowOnClose(true);
    new LDAPListenerClientConnection(null, s, new CannedResponseRequestHandler(), null);
}

Example 2 with TestInputStream

use of com.unboundid.util.TestInputStream in project ldapsdk by pingidentity.

the class LDAPListenerClientConnectionTestCase method testThrowOnClose.

/**
 * Tests the behavior when a client connection has been created without a
 * listener and with elements that throw exceptions on closure.
 *
 * @throws  Exception  If an unexpected problem occurs.
 */
@Test()
public void testThrowOnClose() throws Exception {
    final TestSocket s = new TestSocket(new TestInputStream(new ByteArrayInputStream(new byte[1024]), new IOException("foo"), 100, true), new TestOutputStream(new ByteArrayOutputStream(), new IOException("bar"), 1024, true));
    s.setThrowOnClose(true);
    final LDAPListenerClientConnection conn = new LDAPListenerClientConnection(null, s, new CannedResponseRequestHandler(), new TestLDAPListenerExceptionHandler());
    conn.removeSearchEntryTransformer(null);
    conn.removeSearchReferenceTransformer(null);
    conn.removeIntermediateResponseTransformer(null);
    conn.close(new LDAPException(ResultCode.LOCAL_ERROR, "foo"));
}

Example 3 with TestInputStream

use of com.unboundid.util.TestInputStream in project ldapsdk by pingidentity.

the class LDAPMessageTestCase method testReadFromIOExceptionAfterFirst.

/**
 * Tests the {@code LDAPMessage.readFrom} method with an input stream that
 * throws an exception after the first byte has been read.
 *
 * @throws  Exception  If an unexpected problem occurs.
 */
@Test(expectedExceptions = { LDAPException.class })
public void testReadFromIOExceptionAfterFirst() throws Exception {
    ByteArrayInputStream baos = new ByteArrayInputStream(new byte[10]);
    TestInputStream is = new TestInputStream(baos, new IOException(), 1, false);
    LDAPMessage.readFrom(new ASN1StreamReader(is), false);
}

Example 4 with TestInputStream

use of com.unboundid.util.TestInputStream in project ldapsdk by pingidentity.

the class LDIFEntrySourceTestCase method testReaderThrowsIOException.

/**
 * Tests with an LDIF reader that will always throw an I/O exception.
 *
 * @throws  Exception  If an unexpected problem occurs.
 */
@Test(expectedExceptions = { EntrySourceException.class })
public void testReaderThrowsIOException() throws Exception {
    InputStream is = new TestInputStream(new FileInputStream(ldifFile), new IOException(), 0, true);
    LDIFEntrySource s = new LDIFEntrySource(new LDIFReader(is));
    try {
        s.nextEntry();
    } finally {
        s.close();
    }
}

Example 5 with TestInputStream

use of com.unboundid.util.TestInputStream in project ldapsdk by pingidentity.

the class LDAPListenerClientConnectionTestCase method testThrowOnNewRequestHandlerInstance.

/**
 * Tests the behavior when a client connection has been created without a
 * listener and with a request handler that throws an exception when trying to
 * create a new instance.
 *
 * @throws  Exception  If an unexpected problem occurs.
 */
@Test(expectedExceptions = { LDAPException.class })
public void testThrowOnNewRequestHandlerInstance() throws Exception {
    final TestSocket s = new TestSocket(new TestInputStream(new ByteArrayInputStream(new byte[1024]), new IOException("foo"), 100, true), new TestOutputStream(new ByteArrayOutputStream(), new IOException("bar"), 1024, true));
    s.setThrowOnClose(true);
    TestRequestHandler.setThrowOnNewInstance(true);
    try {
        new LDAPListenerClientConnection(null, s, new TestRequestHandler(), null);
    } finally {
        TestRequestHandler.setThrowOnNewInstance(false);
    }
}