appserver/tests/amx/src/org/glassfish/admin/amxtest/TestRunner.java - glassfish - Git at Google

 /*
  * Copyright (c) 1997, 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 org.glassfish.admin.amxtest;

 import com.sun.appserv.management.DomainRoot;
 import com.sun.appserv.management.base.Util;
 import com.sun.appserv.management.client.AppserverConnectionSource;
 import com.sun.appserv.management.client.ConnectionSource;
 import com.sun.appserv.management.client.HandshakeCompletedListenerImpl;
 import com.sun.appserv.management.client.ProxyFactory;
 import com.sun.appserv.management.client.TLSParams;
 import com.sun.appserv.management.util.jmx.JMXUtil;
 import com.sun.appserv.management.util.jmx.ObjectNameComparator;
 import com.sun.appserv.management.util.stringifier.SmartStringifier;
 import junit.extensions.ActiveTestSuite;
 import junit.framework.TestSuite;
 import org.glassfish.admin.amxtest.JMXTestBase;

 import javax.management.JMException;
 import javax.management.MBeanInfo;
 import javax.management.MBeanServerConnection;
 import javax.management.ObjectName;
 import javax.net.ssl.X509TrustManager;
 import java.io.IOException;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;


 /**
  Class that supports running all the unit tests for AMX
  */
 public final class TestRunner {
     final ConnectionSource mConn;
     boolean mVerbose;

     public TestRunner(final ConnectionSource conn) {
         mConn = conn;
         mVerbose = false;
     }


     public int
     runSuite(
             String name,
             TestSuite suite) {
         System.out.println("*** testing " + name + " ***");

         junit.textui.TestRunner runner = new junit.textui.TestRunner();
         junit.framework.TestResult result = runner.doRun(suite, false);

         return (result.failureCount());
     }

     public int
     testClass(final Class<junit.framework.TestCase> theClass) {
         final TestSuite suite = new TestSuite(theClass);
         return (runSuite(theClass.getName(), suite));
     }


     public int
     testClassThreaded(final Class<junit.framework.TestCase> theClass) {
         final TestSuite suite = new ActiveTestSuite(theClass);

         return (runSuite(theClass.getName(), suite));
     }


     public void
     runTests(
             List<Class<junit.framework.TestCase>> testClasses,
             boolean threaded)
             throws Exception {
         for (final Class<junit.framework.TestCase> theClass : testClasses) {
             final int failureCount = threaded ?
                     testClassThreaded(theClass) :
                     testClass(theClass);

             if (failureCount != 0) {
                 println("Test " + theClass.getName() + " had failures: " + failureCount);
             }
         }
     }

     public long
     elapsed(final long start) {
         return (System.currentTimeMillis() - start);
     }

     private long
     testGetMBeanInfoSpeed(
             final MBeanServerConnection conn,
             final ObjectName[] objectNames)
             throws IOException, JMException {
         // sorting provides consistent, ordered output
         Arrays.sort(objectNames, ObjectNameComparator.INSTANCE);

         final long startAll = System.currentTimeMillis();
         for (int i = 0; i < objectNames.length; ++i) {
             final ObjectName objectName = objectNames[i];

             final long start = System.currentTimeMillis();

             final MBeanInfo mbeanInfo = conn.getMBeanInfo(objectName);

             final long elapsed = elapsed(start);

             String id = objectName.toString();
             String value;

             if ((value = objectName.getKeyProperty("type")) != null) {
                 id = value;
             } else if ((value = objectName.getKeyProperty("j2eeType")) != null) {
                 id = value;
             }

             if ((value = objectName.getKeyProperty("name")) != null) {
                 id = Util.concatenateProps(id, Util.makeNameProp(value));
             }

             //printVerbose( "GetMBeanInfo time for " + id + " = " + elapsed );
         }

         final long elapsed = System.currentTimeMillis() - startAll;
         return (elapsed);
     }


     protected void
     printVerbose(final Object o) {
         if (mVerbose) {
             println(o);
         }
     }

     private void
     println(final Object o) {
         System.out.println(toString(o));
     }

     private void
     print(final Object o) {
         System.out.print(toString(o));
     }

     private void
     testGetMBeanInfoSpeed(
             final MBeanServerConnection conn,
             final String domain,
             final String props)
             throws IOException, JMException {
         final ObjectName pattern = Util.newObjectNamePattern(domain, props);
         final Set<ObjectName> objectNameSet = JMXUtil.queryNames(conn, pattern, null);

         final ObjectName[] objectNames = new ObjectName[objectNameSet.size()];
         objectNameSet.toArray(objectNames);

         final long elapsed = testGetMBeanInfoSpeed(conn, objectNames);

         println("Time to getMBeanInfo on " + domain + ":" + props + " (" + objectNames.length + " MBeans)" +
                 " = " + elapsed + "ms");
     }

     public MBeanServerConnection
     getMBeanServerConnection()
             throws IOException {
         return (mConn == null ? null : mConn.getMBeanServerConnection(false));
     }

     public void
     testSpeed()
             throws IOException, JMException {
         final DomainRoot domainRootProxy = ProxyFactory.getInstance(mConn).
                 createDomainRoot();

         final MBeanServerConnection conn = getMBeanServerConnection();

         testGetMBeanInfoSpeed(conn, Util.getObjectName(domainRootProxy).getDomain(), JMXUtil.WILD_ALL);
     }


     /**
      Comment in call to this for hard-coded test.
      */
     private void
     testAppserverConnectionSource(
             final String host,
             final String user,
             final String password)
             throws IOException {
         MBeanServerConnection conn = null;

         final TestClientTrustStoreTrustManager tm = new TestClientTrustStoreTrustManager();
         final HandshakeCompletedListenerImpl hcl = new HandshakeCompletedListenerImpl();
         tm.setPrompt(true);
         final TLSParams tlsParams =
                 new TLSParams(new X509TrustManager[]{tm}, hcl);

         println("\ntestAppserverConnectionSource: testing: " + AppserverConnectionSource.PROTOCOL_RMI);

         final ConnectionSource rmiSource =
                 new AppserverConnectionSource(AppserverConnectionSource.PROTOCOL_RMI,
                                               host, 8686, user, password, null);
         conn = rmiSource.getMBeanServerConnection(true);
         conn.isRegistered(JMXUtil.getMBeanServerDelegateObjectName());

         println(AppserverConnectionSource.PROTOCOL_RMI + " OK using " + rmiSource);


         println("\ntestAppserverConnectionSource: testing: " + AppserverConnectionSource.PROTOCOL_HTTP);
         final Map<String, String> env = Collections.emptyMap();

         final ConnectionSource httpSource =
                 new AppserverConnectionSource(AppserverConnectionSource.PROTOCOL_HTTP,
                                               host, 1234, user, password, tlsParams, env);
         conn = httpSource.getMBeanServerConnection(true);
         assert conn.isRegistered(JMXUtil.getMBeanServerDelegateObjectName());

         println(AppserverConnectionSource.PROTOCOL_HTTP + " OK using " + httpSource);

     }


     public static String
     toString(Object o) {
         return (SmartStringifier.toString(o));
     }

     /**
      @param threaded if true, run the tests from each TestCase in separate threads
      @param env      arbitrary environment values for JMXTestBase
      */
     protected void
     runAll(
             final List<Class<junit.framework.TestCase>> testClasses,
             final boolean threaded,
             final Map<String, Object> env)
             throws Exception {
         mVerbose = Boolean.valueOf((String) env.get(PropertyKeys.VERBOSE_KEY)).booleanValue();

         //testSpeed();

         // use the current connection which must contain the AMX MBeans
         final MBeanServerConnection conn = getMBeanServerConnection();

         JMXTestBase.setGlobalConnection(conn);
         JMXTestBase.setEnvValues(env);

         println("\n--- " + testClasses.size() + " TEST CLASSES ---");

         for (final Class<junit.framework.TestCase> theClass : testClasses) {
             println(theClass.getName());
         }

         println("\n--- BEGIN TESTS ---");
         runTests(testClasses, threaded);
     }
 }
	/*
	* Copyright (c) 1997, 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 org.glassfish.admin.amxtest;

	import com.sun.appserv.management.DomainRoot;
	import com.sun.appserv.management.base.Util;
	import com.sun.appserv.management.client.AppserverConnectionSource;
	import com.sun.appserv.management.client.ConnectionSource;
	import com.sun.appserv.management.client.HandshakeCompletedListenerImpl;
	import com.sun.appserv.management.client.ProxyFactory;
	import com.sun.appserv.management.client.TLSParams;
	import com.sun.appserv.management.util.jmx.JMXUtil;
	import com.sun.appserv.management.util.jmx.ObjectNameComparator;
	import com.sun.appserv.management.util.stringifier.SmartStringifier;
	import junit.extensions.ActiveTestSuite;
	import junit.framework.TestSuite;
	import org.glassfish.admin.amxtest.JMXTestBase;

	import javax.management.JMException;
	import javax.management.MBeanInfo;
	import javax.management.MBeanServerConnection;
	import javax.management.ObjectName;
	import javax.net.ssl.X509TrustManager;
	import java.io.IOException;
	import java.util.Arrays;
	import java.util.Collections;
	import java.util.List;
	import java.util.Map;
	import java.util.Set;


	/**
	Class that supports running all the unit tests for AMX
	*/
	public final class TestRunner {
	final ConnectionSource mConn;
	boolean mVerbose;

	public TestRunner(final ConnectionSource conn) {
	mConn = conn;
	mVerbose = false;
	}


	public int
	runSuite(
	String name,
	TestSuite suite) {
	System.out.println("* testing " + name + " *");

	junit.textui.TestRunner runner = new junit.textui.TestRunner();
	junit.framework.TestResult result = runner.doRun(suite, false);

	return (result.failureCount());
	}

	public int
	testClass(final Class<junit.framework.TestCase> theClass) {
	final TestSuite suite = new TestSuite(theClass);
	return (runSuite(theClass.getName(), suite));
	}


	public int
	testClassThreaded(final Class<junit.framework.TestCase> theClass) {
	final TestSuite suite = new ActiveTestSuite(theClass);

	return (runSuite(theClass.getName(), suite));
	}


	public void
	runTests(
	List<Class<junit.framework.TestCase>> testClasses,
	boolean threaded)
	throws Exception {
	for (final Class<junit.framework.TestCase> theClass : testClasses) {
	final int failureCount = threaded ?
	testClassThreaded(theClass) :
	testClass(theClass);

	if (failureCount != 0) {
	println("Test " + theClass.getName() + " had failures: " + failureCount);
	}
	}
	}

	public long
	elapsed(final long start) {
	return (System.currentTimeMillis() - start);
	}

	private long
	testGetMBeanInfoSpeed(
	final MBeanServerConnection conn,
	final ObjectName[] objectNames)
	throws IOException, JMException {
	// sorting provides consistent, ordered output
	Arrays.sort(objectNames, ObjectNameComparator.INSTANCE);

	final long startAll = System.currentTimeMillis();
	for (int i = 0; i < objectNames.length; ++i) {
	final ObjectName objectName = objectNames[i];

	final long start = System.currentTimeMillis();

	final MBeanInfo mbeanInfo = conn.getMBeanInfo(objectName);

	final long elapsed = elapsed(start);

	String id = objectName.toString();
	String value;

	if ((value = objectName.getKeyProperty("type")) != null) {
	id = value;
	} else if ((value = objectName.getKeyProperty("j2eeType")) != null) {
	id = value;
	}

	if ((value = objectName.getKeyProperty("name")) != null) {
	id = Util.concatenateProps(id, Util.makeNameProp(value));
	}

	//printVerbose( "GetMBeanInfo time for " + id + " = " + elapsed );
	}

	final long elapsed = System.currentTimeMillis() - startAll;
	return (elapsed);
	}


	protected void
	printVerbose(final Object o) {
	if (mVerbose) {
	println(o);
	}
	}

	private void
	println(final Object o) {
	System.out.println(toString(o));
	}

	private void
	print(final Object o) {
	System.out.print(toString(o));
	}

	private void
	testGetMBeanInfoSpeed(
	final MBeanServerConnection conn,
	final String domain,
	final String props)
	throws IOException, JMException {
	final ObjectName pattern = Util.newObjectNamePattern(domain, props);
	final Set<ObjectName> objectNameSet = JMXUtil.queryNames(conn, pattern, null);

	final ObjectName[] objectNames = new ObjectName[objectNameSet.size()];
	objectNameSet.toArray(objectNames);

	final long elapsed = testGetMBeanInfoSpeed(conn, objectNames);

	println("Time to getMBeanInfo on " + domain + ":" + props + " (" + objectNames.length + " MBeans)" +
	" = " + elapsed + "ms");
	}

	public MBeanServerConnection
	getMBeanServerConnection()
	throws IOException {
	return (mConn == null ? null : mConn.getMBeanServerConnection(false));
	}

	public void
	testSpeed()
	throws IOException, JMException {
	final DomainRoot domainRootProxy = ProxyFactory.getInstance(mConn).
	createDomainRoot();

	final MBeanServerConnection conn = getMBeanServerConnection();

	testGetMBeanInfoSpeed(conn, Util.getObjectName(domainRootProxy).getDomain(), JMXUtil.WILD_ALL);
	}


	/**
	Comment in call to this for hard-coded test.
	*/
	private void
	testAppserverConnectionSource(
	final String host,
	final String user,
	final String password)
	throws IOException {
	MBeanServerConnection conn = null;

	final TestClientTrustStoreTrustManager tm = new TestClientTrustStoreTrustManager();
	final HandshakeCompletedListenerImpl hcl = new HandshakeCompletedListenerImpl();
	tm.setPrompt(true);
	final TLSParams tlsParams =
	new TLSParams(new X509TrustManager[]{tm}, hcl);

	println("\ntestAppserverConnectionSource: testing: " + AppserverConnectionSource.PROTOCOL_RMI);

	final ConnectionSource rmiSource =
	new AppserverConnectionSource(AppserverConnectionSource.PROTOCOL_RMI,
	host, 8686, user, password, null);
	conn = rmiSource.getMBeanServerConnection(true);
	conn.isRegistered(JMXUtil.getMBeanServerDelegateObjectName());

	println(AppserverConnectionSource.PROTOCOL_RMI + " OK using " + rmiSource);


	println("\ntestAppserverConnectionSource: testing: " + AppserverConnectionSource.PROTOCOL_HTTP);
	final Map<String, String> env = Collections.emptyMap();

	final ConnectionSource httpSource =
	new AppserverConnectionSource(AppserverConnectionSource.PROTOCOL_HTTP,
	host, 1234, user, password, tlsParams, env);
	conn = httpSource.getMBeanServerConnection(true);
	assert conn.isRegistered(JMXUtil.getMBeanServerDelegateObjectName());

	println(AppserverConnectionSource.PROTOCOL_HTTP + " OK using " + httpSource);

	}


	public static String
	toString(Object o) {
	return (SmartStringifier.toString(o));
	}

	/**
	@param threaded if true, run the tests from each TestCase in separate threads
	@param env arbitrary environment values for JMXTestBase
	*/
	protected void
	runAll(
	final List<Class<junit.framework.TestCase>> testClasses,
	final boolean threaded,
	final Map<String, Object> env)
	throws Exception {
	mVerbose = Boolean.valueOf((String) env.get(PropertyKeys.VERBOSE_KEY)).booleanValue();

	//testSpeed();

	// use the current connection which must contain the AMX MBeans
	final MBeanServerConnection conn = getMBeanServerConnection();

	JMXTestBase.setGlobalConnection(conn);
	JMXTestBase.setEnvValues(env);

	println("\n--- " + testClasses.size() + " TEST CLASSES ---");

	for (final Class<junit.framework.TestCase> theClass : testClasses) {
	println(theClass.getName());
	}

	println("\n--- BEGIN TESTS ---");
	runTests(testClasses, threaded);
	}
	}