TestTPConversation.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2008, Red Hat, Inc., and others contributors as indicated
 * by the @authors tag. All rights reserved.
 * See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 * This copyrighted material is made available to anyone wishing to use,
 * modify, copy, or redistribute it subject to the terms and conditions
 * of the GNU Lesser General Public License, v. 2.1.
 * This program is distributed in the hope that it will be useful, but WITHOUT A
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
 * PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more details.
 * You should have received a copy of the GNU Lesser General Public License,
 * v.2.1 along with this distribution; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA  02110-1301, USA.
 */
package org.jboss.narayana.blacktie.jatmibroker.xatmi;

import junit.framework.TestCase;

import org.apache.log4j.LogManager;
import org.apache.log4j.Logger;
import org.jboss.narayana.blacktie.jatmibroker.RunServer;
import org.jboss.narayana.blacktie.jatmibroker.core.conf.ConfigurationException;

public class TestTPConversation extends TestCase {
    private static final Logger log = LogManager.getLogger(TestTPConversation.class);
    private RunServer server = new RunServer();

    private Connection connection;
    private int sendlen;
    private X_OCTET sendbuf;
    private Session cd;

    public static int interationCount = 100;

    public void setUp() throws ConnectionException, ConfigurationException {
        server.serverinit();

        ConnectionFactory connectionFactory = ConnectionFactory.getConnectionFactory();
        connection = connectionFactory.getConnection();
        sendlen = 11;
        sendbuf = (X_OCTET) connection.tpalloc("X_OCTET", null);
    }

    public void tearDown() throws ConnectionException, ConfigurationException {
        server.serverdone();
        connection.close();
    }

    public void test_conversation() throws ConnectionException, ConfigurationException {
        log.info("test_conversation");
        server.tpadvertiseTestTPConversation();

        sendbuf.setByteArray("conversate".getBytes());
        cd = connection.tpconnect(RunServer.getServiceNameTestTPConversation(), sendbuf, Connection.TPRECVONLY);
        long revent = 0;
        log.info("Started conversation");
        for (int i = 0; i < interationCount; i++) {
            try {
                Buffer result = cd.tprecv(0);
                fail("Did not get sendonly event");
            } catch (ResponseException e) {
                assertTrue(e.getTperrno() == Connection.TPEEVENT);
                assertTrue(e.getEvent() == Connection.TPEV_SENDONLY);
                Buffer rcvbuf = e.getReceived();
                String expectedResult = ("hi" + i);
                assertTrue(strcmp(expectedResult, rcvbuf) == 0);

                sendbuf.setByteArray(("yo" + i).getBytes());
                // btlogger((char*) "test_conversation:%s:", sendbuf);
                int result = cd.tpsend(sendbuf, Connection.TPRECVONLY);
                assertTrue(result != -1);
            }
        }
        log.info("Conversed");
        try {
            cd.tprecv(0);
            fail("Expected event");
        } catch (ResponseException e) {
            assertTrue(e.getTperrno() == Connection.TPEEVENT);
            Buffer rcvbuf = e.getReceived();
            String expectedResult = ("hi" + interationCount);
            log.info("Expected: " + expectedResult + " Received: " + new String(((X_OCTET) rcvbuf).getByteArray()));
            assertTrue(strcmp(expectedResult, rcvbuf) == 0);
        }
    }

    public void test_short_conversation() throws ConnectionException, ConfigurationException {
        server.tpadvertiseTestTPConversa2();

        log.info("test_short_conversation");
        cd = connection.tpconnect(RunServer.getServiceNameTestTPConversa2(), null, Connection.TPRECVONLY);
        assertTrue(cd != null);

        Buffer rcvbuf = cd.tprecv(0);
        assertTrue(rcvbuf != null);
        assertTrue(strcmp("hi0", rcvbuf) == 0);

        try {
            cd.tprecv(0);
            fail("Expected event");
        } catch (ResponseException e) {
            assertTrue(e.getTperrno() == Connection.TPEEVENT);
            rcvbuf = e.getReceived();
            String expectedResult = ("hi1");
            log.info("Expected: " + expectedResult + " Received: " + new String(((X_OCTET) rcvbuf).getByteArray()));
            assertTrue(strcmp(expectedResult, rcvbuf) == 0);
        }
    }

    public static int strcmp(String string, Buffer buffer) {
        byte[] expected = string.getBytes();
        byte[] received = ((X_OCTET) buffer).getByteArray();
        if (received.length < expected.length) {
            return -1;
        }
        for (int i = 0; i < expected.length; i++) {
            if (expected[i] != received[i]) {
                return -1;
            }
        }

        for (int i = expected.length; i < received.length; i++) {
            if (received[i] != '\0') {
                return -1;
            }
        }
        return 0;
    }

    public static int strcmp(Buffer buffer, String string) {
        return strcmp(string, buffer);
    }

}