ListenerTest.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2006, Red Hat Middleware LLC, and individual contributors
 * as indicated by the @author tags.
 * 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.
 *
 * (C) 2005-2006,
 * @author JBoss Inc.
 */
package com.hp.mwtests.ts.arjuna.recovery;

/*
 * Copyright (C) 1999-2001 by HP Bluestone Software, Inc. All rights Reserved.
 *
 * HP Arjuna Labs,
 * Newcastle upon Tyne,
 * Tyne and Wear,
 * UK.
 *
 * $Id: ListenerTest.java 2342 2006-03-30 13:06:17Z  $
 */

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.SocketException;

import org.junit.Test;

import com.arjuna.ats.arjuna.recovery.Service;
import com.arjuna.ats.internal.arjuna.recovery.Listener;

class ListenerTestService implements Service
{
    public void doWork(InputStream is, OutputStream os)
            throws IOException
    {
        BufferedReader in = new BufferedReader(new InputStreamReader(is));
        PrintWriter out = new PrintWriter(new OutputStreamWriter(os));

        try {
            String test_string = in.readLine();
            out.println(test_string);
            out.flush();
        }
        catch (SocketException ex) {
            ; // socket closed
        }
        catch (IOException ex) {
            System.err.println("testService: failed");
        }
    }
}

public class ListenerTest
{
    @Test
    public void test()
    {
        assertTrue(test_setup());

        test1();

        assertEquals(0, _tests_failed);
        assertEquals(1, _tests_passed);
    }

    /**
     * Pre-test setup.
     */
    private static boolean test_setup()
    {
        boolean setupOk = false;

        try {
            _tests_passed = 0;
            _tests_failed = 0;

            _test_service = new ListenerTestService();

            _test_host = InetAddress.getLocalHost().getHostAddress();
            _test_service_socket = new ServerSocket(_test_port);

            _test_socket = new Socket(_test_host, _test_port);
            _from_test_service = new BufferedReader(new InputStreamReader
                    (_test_socket.getInputStream()));
            _to_test_service = new PrintWriter(new OutputStreamWriter
                    (_test_socket.getOutputStream()));

            setupOk = true;
        }
        catch (Exception ex) {
            System.err.println("test_setup: Failed " + ex);
        }

        return setupOk;
    }

    /**
     * Check that listener can be created and a simlple service ran.
     */
    private static void test1()
    {
        try {
            Listener testListener = new Listener(_test_service_socket,
                    _test_service);
            try {
                testListener.start();

                Thread.sleep(1000); // allow testListener to startup.

                String testString = "testString";
                _to_test_service.println(testString);
                _to_test_service.flush();
                String returnString = _from_test_service.readLine();

                if (testString.equals(returnString)) {
                    System.out.println(_unit_test + "test1: passed");
                    _tests_passed++;
                } else {
                    System.out.println(_unit_test + "test1: failed");
                    _tests_failed++;
                }
            }
            catch (Exception ex) {
                System.err.println(_unit_test + " test1 " + ex);
                _tests_failed++;
            }

            testListener.stopListener();
        }
        catch (IOException ex) {
            System.err.println(_unit_test + " test1 " + ex);
            _tests_failed++;
        }
    }

    private static final String _unit_test = "com.hp.mwtests.ts.arjuna.recovery.ListenerTest: ";

    private static String _test_host;
    private static final int _test_port = 4321;

    private static ListenerTestService _test_service;

    private static Socket _test_socket;
    private static ServerSocket _test_service_socket;

    private static BufferedReader _from_test_service;
    private static PrintWriter _to_test_service;

    private static int _tests_passed = 0;
    private static int _tests_failed = 0;
}