org.apache.openejb.test.TestManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of openejb-itests-client Show documentation
There is a newer version: 4.7.5
Show newest version
/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.openejb.test;

import java.io.File;
import java.io.FileInputStream;
import java.util.Properties;
import java.security.PrivilegedAction;
import java.security.AccessController;

/**
 * 
 * @version $Rev$ $Date$
 */
public class TestManager {
    
    private static TestServer server;
    private static TestDatabase database;
    private static TestJms jms;
    private static boolean warn = true;

    // TODO: Move it to a central place where all system properties are managed in a unified way
    final static String TESTSUITE_PROPERTY_FILENAME = "openejb.testsuite.properties";
    final static String TEST_SERVER_CLASSNAME = "openejb.test.server";
    final static String TEST_DATABASE_CLASSNAME = "openejb.test.database";
    final static String TEST_JMS_CLASSNAME = "openejb.test.jms";

    public static void init(String propertiesFileName) throws Exception{
        Properties props = null;

        try{
            props = new Properties(System.getProperties());
            warn = props.getProperty("openejb.test.nowarn") == null;
        } catch (final SecurityException e){
            throw new IllegalArgumentException("Cannot access the system properties: "+e.getClass().getName()+" "+e.getMessage());
        }
        
        if (propertiesFileName == null) {
            try{
                propertiesFileName = System.getProperty(TESTSUITE_PROPERTY_FILENAME);
                if (propertiesFileName != null) {
                    props.putAll( getProperties( propertiesFileName ) );
                }

            } catch (final SecurityException e){
                throw new IllegalArgumentException("Cannot access the system property \"" + TESTSUITE_PROPERTY_FILENAME + "\": "+e.getClass().getName()+" "+e.getMessage());
            }
        } else {
            props.putAll( getProperties( propertiesFileName ) );
        }
        initServer(props);
        initDatabase(props);
        initJms(props);
    }
    
    public static void start() throws Exception{
        try{
            if (server != null) {
                server.start();
            }
        } catch (final Exception e){
            if (warn) System.out.println("Cannot start the test server: "+e.getClass().getName()+" "+e.getMessage());
            throw e;
        }
        try{
            if (database != null) {
                database.start();
            }
        } catch (final Exception e){
            if (warn) System.out.println("Cannot start the test database: "+e.getClass().getName()+" "+e.getMessage());
            throw e;
        }
    }

    public static void stop() throws Exception{
        try{
            if (database != null) {
                database.stop();
            }
        } catch (final Exception e){
            if (warn) System.out.println("Cannot stop the test database: "+e.getClass().getName()+" "+e.getMessage());
            throw e;
        }
        try{
            if (server != null) {
                server.stop();
            }
        } catch (final Exception e){
            if (warn) System.out.println("Cannot stop the test server 2: "+e.getClass().getName()+" "+e.getMessage());
            throw e;
        }
    }
            
    private static Properties getProperties(final String fileName) throws Exception{
        File file = new File(fileName);
        file = file.getAbsoluteFile();
        final Properties props = (Properties)System.getProperties().clone();
        props.load(new FileInputStream(file));
        return props;
    }

    private static ClassLoader getContextClassLoader() {
        return AccessController.doPrivileged(new PrivilegedAction() {
            public ClassLoader run() {
                return Thread.currentThread().getContextClassLoader();
            }
        });
    }

    private static void initServer(final Properties props){
        try{

            final String className = props.getProperty(TEST_SERVER_CLASSNAME);
            if (className == null) {
                throw new IllegalArgumentException(
                        "Must specify a test server by setting its class name using the system property \"" + TEST_SERVER_CLASSNAME + "\"");
            }
            final ClassLoader cl = getContextClassLoader();
            final Class testServerClass = Class.forName( className, true, cl );
            server = (TestServer)testServerClass.newInstance();
            server.init( props );
        } catch (final Exception e){
            if (warn) e.printStackTrace();
            if (warn) System.out.println("Cannot instantiate or initialize the test server: "+e.getClass().getName()+" "+e.getMessage());
            throw new RuntimeException("Cannot instantiate or initialize the test server: "+e.getClass().getName()+" "+e.getMessage(),e);
        }
    }

    private static void initDatabase(final Properties props){
        try{
            final String className = props.getProperty(TEST_DATABASE_CLASSNAME);
            if (className == null) throw new IllegalArgumentException("Must specify a test database by setting its class name  using the system property \"" + TEST_DATABASE_CLASSNAME + "\"");
            final ClassLoader cl = getContextClassLoader();
            final Class testDatabaseClass = Class.forName( className , true, cl);
            database = (TestDatabase)testDatabaseClass.newInstance();
            database.init( props );
        } catch (final Exception e){
            if (warn) System.out.println("Cannot instantiate or initialize the test database: "+e.getClass().getName()+" "+e.getMessage());
            throw new RuntimeException("Cannot instantiate or initialize the test database: "+e.getClass().getName()+" "+e.getMessage(),e);
        }
    }

    private static void initJms(final Properties props){
        try{
            String className = props.getProperty(TEST_JMS_CLASSNAME);
            if (className == null) className = "org.apache.openejb.test.ActiveMqTestJms";
            final ClassLoader cl = getContextClassLoader();
            final Class testJmsClass = Class.forName( className , true, cl);
            jms = (TestJms)testJmsClass.newInstance();
            jms.init( props );
        } catch (final Exception e){
            if (warn) System.out.println("Cannot instantiate or initialize the test jms: "+e.getClass().getName()+" "+e.getMessage());
            throw new RuntimeException("Cannot instantiate or initialize the test jms: "+e.getClass().getName()+" "+e.getMessage(),e);
        }
    }


    public static TestServer getServer(){
        return server;
    }

    public static TestDatabase getDatabase(){
        return database;
    }

    public static TestJms getJms() {
        return jms;
    }

    public static Properties getContextEnvironment(){
        return server.getContextEnvironment();
    }
}