Client.java

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) 2017 Oracle and/or its affiliates. All rights reserved.
 *
 * The contents of this file are subject to the terms of either the GNU
 * General Public License Version 2 only ("GPL") or the Common Development
 * and Distribution License("CDDL") (collectively, the "License").  You
 * may not use this file except in compliance with the License.  You can
 * obtain a copy of the License at
 * https://oss.oracle.com/licenses/CDDL+GPL-1.1
 * or LICENSE.txt.  See the License for the specific
 * language governing permissions and limitations under the License.
 *
 * When distributing the software, include this License Header Notice in each
 * file and include the License file at LICENSE.txt.
 *
 * GPL Classpath Exception:
 * Oracle designates this particular file as subject to the "Classpath"
 * exception as provided by Oracle in the GPL Version 2 section of the License
 * file that accompanied this code.
 *
 * Modifications:
 * If applicable, add the following below the License Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyright [year] [name of copyright owner]"
 *
 * Contributor(s):
 * If you wish your version of this file to be governed by only the CDDL or
 * only the GPL Version 2, indicate your decision by adding "[Contributor]
 * elects to include this software in this distribution under the [CDDL or GPL
 * Version 2] license."  If you don't indicate a single choice of license, a
 * recipient has the option to distribute your version of this file under
 * either the CDDL, the GPL Version 2 or to extend the choice of license to
 * its licensees as provided above.  However, if you add GPL Version 2 code
 * and therefore, elected the GPL Version 2 license, then the option applies
 * only if the new code is made subject to such option by the copyright
 * holder.
 */

package com.acme;

import javax.ejb.*;
import javax.annotation.*;

import javax.naming.InitialContext;

import java.util.concurrent.*;


import com.sun.ejte.ccl.reporter.SimpleReporterAdapter;

public class Client {

    private static SimpleReporterAdapter stat = 
        new SimpleReporterAdapter("appserv-tests");

    private static String appName;

    private RemoteAsync remoteAsync;

    private StatefulCncSuperIntf serializedBean;

    private StatefulCncSuperIntf nonSerializedBean;

    private int num;

    public static void main(String args[]) {

	appName = args[0]; 
	stat.addDescription(appName);
	Client client = new Client(args);       
        client.doTest();	
        stat.printSummary(appName + "ID");
    }

    public Client(String[] args) {
	num = Integer.valueOf(args[1]);
    }

    public void doTest() {

	try {

	    serializedBean = (StatefulCncSuperIntf) new InitialContext().lookup("java:global/" + appName + "/StatefulBeanSerialized!com.acme.StatefulCncRemote");

	    nonSerializedBean = (StatefulCncSuperIntf) new InitialContext().lookup("java:global/" + appName + "/StatefulBeanNotSerialized!com.acme.StatefulCncRemote");


	    nonSerializedBean.sleep(3);


	    try {
		System.out.println("Sleep a small amount to make sure the async method" +
				   "is dispatched before we make the next call...");
		Thread.sleep(500);
		nonSerializedBean.hello();
		throw new EJBException("Did not receive concurrent access exception");
	    } catch(ConcurrentAccessException cae) {
		System.out.println("Successfully received concurrent access exception for " +
				   " concurrent access attempt on non-serialized sfsb");
	    }

	    try {
		serializedBean.sleep(3);
		try {
		    System.out.println("Sleep a small amount to make sure the async method" +
				   "is dispatched before we make the next call...");
		    Thread.sleep(100);
		    serializedBean.helloWait();
		    throw new EJBException("Should have received timeout exception");
		} catch(ConcurrentAccessTimeoutException cate) {
		    System.out.println("Successfully received timeout exception");
		}

		System.out.println("Attempt loopback call");
		serializedBean.attemptLoopback();

		// Invoking the same bean must still work since the concurrent access
		// exceptions do not result in the bean being destroyed
		String result = serializedBean.hello();
		System.out.println("Serialized bean says " + result);

	    } catch(ConcurrentAccessException cae) {
		 throw new EJBException("Shouldn't have received concurrent access exception");
	    }

	    nonSerializedBean.attemptLoopback();

	    remoteAsync = (RemoteAsync) new InitialContext().lookup("java:global/" + appName + "/SingletonBean");
	    remoteAsync.startTest();
	    
	    ExecutorService executor = Executors.newCachedThreadPool();

	    IncrementCount ics[] = new IncrementCount[num];
	    for(int i = 0; i < ics.length; i++) {

		ics[i] = new IncrementCount();

		executor.execute( ics[i] );
	    }

	    executor.shutdown();

	    executor.awaitTermination(15, TimeUnit.SECONDS);

	    int count = serializedBean.getCount();

	    if( count == num ) {
		System.out.println("Total count of " + count + " invocations is correct");
	    } else {
		throw new EJBException("Wrong count total = " + count);
	    }

	    // Now put the bean to sleep and make a blocking 2nd call.  The first method
	    // is a @Remove method so when the blocking call awakes we should get the
	    // appropriate exception.
	    serializedBean.sleepAndRemove(3);

	    try {
		System.out.println("Sleep a small amount to make sure the async method" +
				   "is dispatched before we make the next call...");
		Thread.sleep(100);
		serializedBean.hello();
		throw new EJBException("Did not receive NoSuchEJBException");
	    } catch(NoSuchEJBException nse) {
		System.out.println("Successfully received NoSuchEJBException for " +
				   " concurrent access attempt on removed sfsb");
	    }


	    stat.addStatus("local main", stat.PASS);

	} catch(Exception e) {
	    stat.addStatus("local main", stat.FAIL);
	    e.printStackTrace();
	}
    }


     class IncrementCount implements Runnable {
	public void run() {
	    serializedBean.incrementCount(1);
	}
    }

}