managedconnectiontestcase.java

一个java方面的消息订阅发送的源码

/**
 * Redistribution and use of this software and associated documentation
 * ("Software"), with or without modification, are permitted provided
 * that the following conditions are met:
 *
 * 1. Redistributions of source code must retain copyright
 *    statements and notices.  Redistributions must also contain a
 *    copy of this document.
 *
 * 2. Redistributions in binary form must reproduce the
 *    above copyright notice, this list of conditions and the
 *    following disclaimer in the documentation and/or other
 *    materials provided with the distribution.
 *
 * 3. The name "Exolab" must not be used to endorse or promote
 *    products derived from this Software without prior written
 *    permission of Exoffice Technologies.  For written permission,
 *    please contact [e-mail].
 *
 * 4. Products derived from this Software may not be called "Exolab"
 *    nor may "Exolab" appear in their names without prior written
 *    permission of Exoffice Technologies. Exolab is a registered
 *    trademark of Exoffice Technologies.
 *
 * 5. Due credit should be given to the Exolab Project
 *    (http://www.exolab.org/).
 *
 * THIS SOFTWARE IS PROVIDED BY EXOFFICE TECHNOLOGIES AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT
 * NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
 * EXOFFICE TECHNOLOGIES OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 * OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * Copyright 2004-2005 (C) Exoffice Technologies Inc. All Rights Reserved.
 *
 * $Id: ManagedConnectionTestCase.java,v 1.2 2005/05/03 13:46:00 tanderson Exp $
 */
package org.exolab.jms.net.connector;

import java.security.Principal;

import junit.framework.TestCase;


/**
 * Tests the {@link ManagedConnectionFactory} interface.
 *
 * @version     $Revision: 1.2 $ $Date: 2005/05/03 13:46:00 $
 * @author      <a href="mailto:[e-mail]">Tim Anderson</a>
 */
public abstract class ManagedConnectionTestCase extends TestCase {

    /**
     * The managed connection factory.
     */
    private ManagedConnectionFactory _factory;


    /**
     * Construct an instance of this class for a specific test case.
     *
     * @param name the name of test case
     */
    public ManagedConnectionTestCase(String name) {
        super(name);
    }

    /**
     * Tests {@link ManagedConnection#getConnection}.
     *
     * @throws Exception for any error
     */
    public void testGetConnection() throws Exception {
        Principal principal = null;

        // set up an acceptor to handle the connection request
        ManagedConnectionAcceptor acceptor = createAcceptor(principal);
        InvocationHandler handler = new TestInvocationHandler();
        TestAcceptorEventListener listener = new TestAcceptorEventListener(
            handler);
        acceptor.accept(listener);

        // create the managed connection
        ConnectionRequestInfo info = getManagedConnectionRequestInfo();
        ManagedConnection managed = _factory.createManagedConnection(
            principal, info);

        // verify that getConnection() fails if no InvocationHandler
        // is registered
        try {
            managed.getConnection();
            fail("Expected " + IllegalStateException.class.getName()
                 + " to be thrown");
        } catch (IllegalStateException exception) {
            // the expected behaviour
        } catch (Exception exception) {
            fail("Expected " + IllegalStateException.class.getName()
                 + " to be thrown, but got exception="
                 + exception.getClass().getName() 
                 + ", message=" + exception.getMessage());
        }

        // register an InvocationHandler and verify that getConnection()
        // succeeds
        managed.setInvocationHandler(new TestInvocationHandler());

        Connection connection = managed.getConnection();
        assertNotNull(connection);        

        // clean up
        managed.destroy();
        listener.destroy();
        acceptor.close();
    }

    /**
     * Verifies that an <code>InvocationHandler</code> can be registered,
     * and that <code>IllegalStateException</code> is thrown if
     * {@link ManagedConnection#setInvocationHandler} is invoked more
     * than once.
     *
     * @throws Exception for any error
     */
    public void testSetInvocationHandler() throws Exception {
        Principal principal = null;

        // set up an acceptor to handle the connection request
        ManagedConnectionAcceptor acceptor = createAcceptor(principal);
        InvocationHandler handler = new TestInvocationHandler();
        TestAcceptorEventListener listener = new TestAcceptorEventListener(
            handler);
        acceptor.accept(listener);

        // create the managed connection
        ManagedConnection managed = createConnection(principal);
        try {
            managed.setInvocationHandler(null);
            fail("Expected " + IllegalStateException.class.getName()
                 + " to be thrown");
        } catch (IllegalStateException exception) {
            // the expected behaviour
        } catch (Exception exception) {
            fail("Expected " + IllegalStateException.class.getName()
                 + " to be thrown, but got exception="
                 + exception.getClass().getName() 
                 + ", message=" + exception.getMessage());
        }

        try {
            managed.setInvocationHandler(new TestInvocationHandler());
            fail("Expected " + IllegalStateException.class.getName()
                 + " to be thrown");
        } catch (IllegalStateException exception) {
            // the expected behaviour
        } catch (Exception exception) {
            fail("Expected " + IllegalStateException.class.getName()
                 + " to be thrown, but got exception="
                 + exception.getClass().getName() 
                 + ", message=" + exception.getMessage());
        }

        // clean up
        managed.destroy();
        listener.destroy();
        acceptor.close();
    }

    /**
     * Tests {@link ManagedConnection#isAlive}, from the client perspective.
     *
     * @throws Exception for any error
     */
    public void testClientIsAlive() throws Exception {
        Principal principal = null;

        // set up an acceptor to handle the connection request
        ManagedConnectionAcceptor acceptor = createAcceptor(principal);
        InvocationHandler handler = new TestInvocationHandler();
        TestAcceptorEventListener listener = new TestAcceptorEventListener(
            handler);
        acceptor.accept(listener);

        // create the client connection
        ManagedConnection client = createConnection(principal);
        assertTrue(client.isAlive());

        // delay to enable the listener to get notified
        Thread.sleep(1000);

        // destroy the server connection, and verify its dead from the
        // client perspective
        ManagedConnection server = listener.getConnection();
        assertNotNull(server);
        server.destroy();
        assertFalse(client.isAlive());

        // destroy the client
        client.destroy();
        assertFalse(client.isAlive());

        // clean up
        acceptor.close();
    }

    /**
     * Tests {@link ManagedConnection#isAlive}, from the server perspective.
     *
     * @throws Exception for any error
     */
    public void testServerIsAlive() throws Exception {
        Principal principal = null;

        // set up an acceptor to handle the connection request
        ManagedConnectionAcceptor acceptor = createAcceptor(principal);
        InvocationHandler handler = new TestInvocationHandler();
        TestAcceptorEventListener listener = new TestAcceptorEventListener(
            handler);
        acceptor.accept(listener);

        // create the client connection
        ManagedConnection client = createConnection(principal);

        // delay to enable the listener to get notified
        Thread.sleep(1000);

        // get the server connection
        ManagedConnection server = listener.getConnection();
        assertNotNull(server);
        assertTrue(server.isAlive());
        
        // destroy the client connection, and verify its dead from the
        // server perspective
        client.destroy();
        assertFalse(server.isAlive());

        // destroy the server connection
        server.destroy();
        assertFalse(server.isAlive());

        // clean up
        acceptor.close();
    }

    /**
     * Sets up the test case.
     *
     * @throws Exception for any error
     */
    protected void setUp() throws Exception {
        _factory = createManagedConnectionFactory();
    }

    /**
     * Creates a managed connection factory.
     *
     * @return the new managed connection factory
     * @throws Exception for any error
     */
    protected abstract ManagedConnectionFactory
        createManagedConnectionFactory() throws Exception;

    /**
     * Returns connection request info suitable for creating a managed
     * connection.
     *
     * @return connection request info for creating a managed connection
     * @throws Exception for any error
     */
    protected abstract ConnectionRequestInfo getManagedConnectionRequestInfo()
        throws Exception;

    /**
     * Returns connection request info suitable for creating a managed
     * connection acceptor.
     *
     * @return connection request info for creating a managed connection
     * acceptor
     * @throws Exception for any error
     */
    protected abstract ConnectionRequestInfo getAcceptorConnectionRequestInfo()
        throws Exception;

    /**
     * Helper to return the cached managed connection factory
     *
     * @return the cached managed connection factory
     */
    protected ManagedConnectionFactory getManagedConnectionFactory() {
        return _factory;
    }

    /**
     * Helper to create a managed connection.
     *
     * @param principal the principal to use. May be <code>null</code>
     * @throws Exception for any error
     */
    protected ManagedConnection createConnection(Principal principal)
        throws Exception {
        ConnectionRequestInfo info = getManagedConnectionRequestInfo();
        ManagedConnection connection = _factory.createManagedConnection(
            principal, info);
        connection.setInvocationHandler(new TestInvocationHandler());
        return connection;
    }

    /**
     * Helper to create a managed connection acceptor.
     *
     * @param principal the principal to use. May be <code>null</code>
     * @throws Exception for any error
     */
    protected ManagedConnectionAcceptor createAcceptor(Principal principal)
        throws Exception {
        ConnectionRequestInfo info = getAcceptorConnectionRequestInfo();
        Authenticator authenticator = new TestAuthenticator(principal);
        return _factory.createManagedConnectionAcceptor(authenticator, info);
    }

}