appserver/tests/appserv-tests/devtests/ejb/stubs/stubser/client/Client.java

/*
 * Copyright (c) 2017, 2018 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package com.sun.s1asdev.ejb.stubs.stubser.client;

import java.io.*;
import java.util.*;
import jakarta.ejb.*;
import javax.naming.*;
import javax.rmi.PortableRemoteObject;
import org.omg.CORBA.ORB;
import com.sun.s1asdev.ejb.stubs.stubser.FooHome;
import com.sun.s1asdev.ejb.stubs.stubser.Foo;
import com.sun.ejte.ccl.reporter.SimpleReporterAdapter;

public class Client {

    private static final String HOME_HANDLE_FILE = "homehandle";
    private static final String FOO_HANDLE_FILE = "foohandle";

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

    public static void main (String[] args) {

        System.out.println("cp = " + System.getProperty("java.class.path"));

        Client client = new Client(args);
        System.out.println("args.length = " + args.length);
        for(int i = 0; i < args.length; i++) {
            System.out.println("arg " + i + " = " + args[0]);
        }


        if( args.length == 1) {
            stat.addDescription("ejb-stubs-stubser2");
            client.doRestartTest();
            stat.printSummary("ejb-stubs-stubser2ID");
        } else {
            stat.addDescription("ejb-stubs-stubser1");
            client.doTest();
            stat.printSummary("ejb-stubs-stubser1ID");
        }

    }

    public Client (String[] args) {
    }

    public void doRestartTest() {
        try {
            // when we lookup a FooHome *before* deserializing the home reference,
            // everything works
            // @@@
            System.out.println("Adding lookup before reconstitution ...");
            lookupHome();

            System.out.println("Attempting to reconstruct foo handle");
            Handle fh = (Handle) readFromFile(FOO_HANDLE_FILE);
            Foo f = (Foo) fh.getEJBObject();
            f.callHello();

            System.out.println("successfully invoked foo via reconsituted handle");

            System.out.println("Attempting to reconstruct home handle as first" +
                               " operation in client");
            HomeHandle hh = (HomeHandle) readFromFile(HOME_HANDLE_FILE);
            invoke((FooHome) hh.getEJBHome());

            System.out.println("successfully invoked ejb via reconsituted home");


            stat.addStatus("ejbclient restart2", stat.PASS);

        } catch(Exception e) {
            e.printStackTrace();
            stat.addStatus("ejbclient restart2" , stat.FAIL);
        }

    }

    public void doTest() {

        try {


            FooHome  home = lookupHome();

            System.err.println("Narrowed home!!");



            // create home handle
            HomeHandle hh = home.getHomeHandle();
            serializeToFile(HOME_HANDLE_FILE, hh);
            HomeHandle hh2 = (HomeHandle) readFromFile(HOME_HANDLE_FILE);
            home = (FooHome) hh2.getEJBHome();

            Foo f = invoke(home);
            System.out.println("successfully invoked ejb via reconsituted home");

            // create SFSB instance handle file
            Handle fh = f.getHandle();
            serializeToFile(FOO_HANDLE_FILE, fh);
            Handle fh2 = (Handle) readFromFile(FOO_HANDLE_FILE);
            Foo f2 = (Foo) fh2.getEJBObject();
            f2.callHello();

            System.out.println("successfully invoked ejb");

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

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

            return;
    }

    private FooHome lookupHome() throws Exception {
        Context ic = new InitialContext();

        System.out.println("Looking up ejb ref ");
        // create EJB using factory from container
        Object objref = ic.lookup("java:global/ejb-stubs-stubserApp/ejb-stubs-stubser-ejb/FooBean");
        System.out.println("objref = " + objref);
        System.err.println("Looked up home!!");

        FooHome  home = (FooHome)PortableRemoteObject.narrow
            (objref, FooHome.class);

        System.err.println("Narrowed home!!");

        return home;
    }

    private Foo invoke(FooHome fooHome) throws Exception {

        Foo f = fooHome.create();
        System.err.println("Got the EJB!!");

        // invoke method on the EJB
        System.out.println("invoking ejb");
        f.callHello();

        return f;

    }

    private void serializeToFile(String filename, Serializable s) throws Exception {
        FileOutputStream fos = new FileOutputStream(filename);
        ObjectOutputStream os = new ObjectOutputStream(fos);
        os.writeObject(s);
        os.close();
        System.out.println("serialized " + filename);
    }

    private Serializable readFromFile(String filename) throws Exception {
        FileInputStream fis = new FileInputStream(filename);
        ObjectInputStream ois = new ObjectInputStream(fis);
        Serializable s = (Serializable) ois.readObject();
        ois.close();
        System.out.println("Reconstituted " + filename);

        return s;
    }

}