/*
 * Copyright (c) 2017, 2020 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.sfsb.passivateactivate.client;

import java.util.ArrayList;
import javax.naming.*;
import jakarta.jms.*;
import jakarta.ejb.*;
import javax.rmi.PortableRemoteObject;

import com.sun.s1asdev.ejb.sfsb.passivateactivate.ejb.SFSBHome;
import com.sun.s1asdev.ejb.sfsb.passivateactivate.ejb.SFSB;

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

public class Client {

    // consts
    public static String kTestNotRun    = "TEST NOT RUN";
    public static String kTestPassed    = "TEST PASSED";
    public static String kTestFailed    = "TEST FAILED";

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

    private static final int    MAX_SFSBS = 40;

    private SFSBHome home;
    private ArrayList sfsbList = new ArrayList();

    private String _sfsbPrefix = "SFSB_" + System.currentTimeMillis() + "_";

    public static void main (String[] args) {

        stat.addDescription("passivateactivate");
        Client client = new Client(args);
        System.out.println("[passivateactivateClient] doTest()...");
        client.doTest();
        System.out.println("[passivateactivateClient] DONE doTest()...");
        stat.printSummary("passivateactivate");
    }

    public Client (String[] args) {
    }

    public void doTest() {
        initSFSBList();     //create SFSBs
        accessSFSB();       //access the SFBS
        removeTest();
    }

    private void initSFSBList() {
        System.out.println("[passivateactivateClient] Inside init....");
        try {
            Context ic = new InitialContext();
            Object objref = ic.lookup("java:comp/env/ejb/SFSB");
            home = (SFSBHome)PortableRemoteObject.narrow
                (objref, SFSBHome.class);
            for (int i=0; i < MAX_SFSBS; i++) {

                //Creating these many SFSBs will cause passivation
                SFSB sfsb = (SFSB) home.create(_sfsbPrefix + i);
                sfsb.createSFSBChild();
                sfsbList.add(sfsb);
            }
            System.out.println("[passivateactivate] Initalization done");
            stat.addStatus("ejbclient initSFSBList", stat.PASS);
        } catch(Exception e) {
            e.printStackTrace();
            //stat.addStatus("ejbclient localEntityGetEJBObject(-)" , stat.PASS);
            System.out.println("[passivateactivateClient] Exception in init....");
            e.printStackTrace();
            stat.addStatus("ejbclient initSFSBList", stat.FAIL);
        }
    }

    public void accessSFSB() {
        try {
            System.out.println("Waiting for 10 seconds before accessing...");
            for (int i=0; i<10; i++) {
                System.out.println("" + (10 - i) + " seconds left...");
                try {
                    Thread.currentThread().sleep(1*1000);
                } catch (Exception ex) {
                }
            }

            boolean passed = true;
            for (int i=0; i < MAX_SFSBS; i++) {
                SFSB sfsb = (SFSB) sfsbList.get(i);
                String sfsbName = _sfsbPrefix+i;
                String retrievedName = sfsb.getName();

                boolean sessionCtxTest = sfsb.checkSessionContext();
                boolean initialCtxTest = sfsb.checkInitialContext();
                boolean entityHomeTest = sfsb.checkEntityHome();
                boolean entityLocalHomeTest = sfsb.checkEntityLocalHome();
                boolean entityRemoteTest = sfsb.checkEntityRemoteRef();
                boolean entityLocalTest = sfsb.checkEntityLocalRef();
                boolean homeHandleTest = sfsb.checkHomeHandle();
                boolean handleTest = sfsb.checkHandle();
                boolean utTest = sfsb.checkUserTransaction();
                boolean activationTest = (sfsb.getActivationCount() != 0);
                boolean passivationTest = (sfsb.getPassivationCount() != 0);

                int actCount = sfsb.getActivationCount();
                int pasCount = sfsb.getPassivationCount();

                System.out.println(sessionCtxTest + "; " + initialCtxTest
                    + "; " + entityHomeTest + "; " + entityLocalHomeTest
                    + "; " + entityRemoteTest + "; " + entityLocalTest
                    + "; " + homeHandleTest + "; " + handleTest
                    + "; " + utTest
                    + "; " + activationTest + " (" + actCount + ")"
                    + "; " + passivationTest + " (" + pasCount + ")"
                );

                passed = sessionCtxTest && initialCtxTest
                    && entityHomeTest && entityLocalHomeTest
                    && entityRemoteTest && entityLocalTest
                    && homeHandleTest && handleTest && utTest
                    && activationTest && passivationTest;

                if (! passed) {
                    break;
                }

                sfsb.sleepForSeconds(2);
            }

            if (passed) {
                stat.addStatus("ejbclient accessSFSB", stat.PASS);
            } else {
                stat.addStatus("ejbclient accessSFSB", stat.FAIL);
            }

            for (int i=0; i < MAX_SFSBS; i++) {
                SFSB sfsb = (SFSB) sfsbList.get(i);
                String sfsbName = _sfsbPrefix+i;

                sfsb.makeStateNonSerializable();
            }

            //Creating these many SFSBs should force passivation of the above
            //        non-serializable beans
            for (int i=0; i < MAX_SFSBS; i++) {
                home.create(_sfsbPrefix + (i+1)*1000);
            }

            System.out.println("Waiting for 10 seconds for passivation to complete...");

            for (int i=0; i<10; i++) {
                System.out.println("" + (10 - i) + " seconds left...");
                try {
                    Thread.currentThread().sleep(1*1000);
                } catch (Exception ex) {
                }
            }

            for (int i=0; i < MAX_SFSBS; i++) {
                SFSB sfsb = (SFSB) sfsbList.get(i);
                String sfsbName = _sfsbPrefix+i;

                try {
                    System.out.print("Expecting exception for: " + sfsbName);
                    String nm = sfsb.getName();
                    System.out.println("ERROR. Didn't get expected exception. "
                            + "Got: " + nm);
                    passed = false;
                    break;
                } catch (Exception ex) {
                    System.out.println("[**Got Exception**]");
                }
            }
            if (passed) {
                stat.addStatus("ejbclient non-serializable-state", stat.PASS);
            } else {
                stat.addStatus("ejbclient non-serializable-state", stat.FAIL);
            }
        } catch (Exception ex) {
            stat.addStatus("ejbclient accessSFSB", stat.FAIL);

        }
    }


    public void removeTest() {
        SFSB sfsb = null;
        try {
            String myName = "_2_" + _sfsbPrefix + "_2_";
            sfsb = (SFSB) home.create(myName);
            String retrievedName = sfsb.getName();
            boolean nameOK = myName.equalsIgnoreCase(retrievedName);
            boolean gotException = false;

            // Test BMT SFSB remove called in a transaction
            jakarta.transaction.UserTransaction ut = (jakarta.transaction.UserTransaction)
                     (new InitialContext()).lookup("java:comp/UserTransaction");
            ut.begin();
            sfsb.remove();
            ut.commit();
            try {
                sfsb.getName();
                gotException = false;            //Expecting an exception
            } catch (Exception ex) {
                gotException = true;
            }

            String resultStr = "(" + nameOK + " @@@ " + gotException + ")";
            if (nameOK && gotException) {
                stat.addStatus("ejbclient removeTest " + resultStr, stat.PASS);
            } else {
                System.err.println("======> FAIL because: nameOK: " + nameOK + " gotException: " + gotException);
                stat.addStatus("ejbclient removeTest " + resultStr, stat.FAIL);
            }

        } catch (Exception ex) {
            ex.printStackTrace();
            stat.addStatus("ejbclient removeTest", stat.FAIL);
        }
    }

} //Client{}