/*
* 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.junit.jee.statement;
import org.apache.openejb.OpenEJBException;
import org.apache.openejb.OpenEjbContainer;
import org.apache.openejb.junit.jee.config.Properties;
import org.apache.openejb.junit.jee.config.Property;
import org.apache.openejb.junit.jee.config.PropertyFile;
import org.apache.openejb.osgi.client.LocalInitialContextFactory;
import org.apache.openejb.util.Classes;
import org.junit.runners.model.Statement;
import javax.ejb.embeddable.EJBContainer;
import javax.naming.Context;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
public class StartingStatement extends DecoratingStatement {
private final Class<?> clazz;
private java.util.Properties properties;
private EJBContainer container;
public StartingStatement(final Statement statement, final Class<?> clazz) {
super(statement);
this.clazz = clazz;
}
@Override
protected void before() throws Exception {
properties = new java.util.Properties();
{ // set caller first to let it be overridable by @Property
final StringBuilder b = new StringBuilder();
for (final Class<?> c : Classes.ancestors(clazz)) {
if (c != Object.class) {
b.append(c.getName()).append(",");
}
}
b.setLength(b.length() - 1);
properties.put(OpenEjbContainer.Provider.OPENEJB_ADDITIONNAL_CALLERS_KEY, b.toString());
}
// default implicit config
{
final InputStream is = clazz.getClassLoader().getResourceAsStream("openejb-junit.properties");
if (is != null) {
properties.load(is);
}
}
final PropertyFile propertyFile = clazz.getAnnotation(PropertyFile.class);
if (propertyFile != null) {
final String path = propertyFile.value();
if (!path.isEmpty()) {
InputStream is = clazz.getClassLoader().getResourceAsStream(path);
if (is == null) {
final File file = new File(path);
if (file.exists()) {
is = new FileInputStream(file);
} else {
throw new OpenEJBException("properties resource '" + path + "' not found");
}
}
properties.load(is);
}
}
final Properties annotationConfig = clazz.getAnnotation(Properties.class);
if (annotationConfig != null) {
for (final Property property : annotationConfig.value()) {
properties.put(property.key(), property.value());
}
}
if (!properties.containsKey(Context.INITIAL_CONTEXT_FACTORY)) {
properties.setProperty(Context.INITIAL_CONTEXT_FACTORY, LocalInitialContextFactory.class.getName());
}
container = EJBContainer.createEJBContainer(properties);
}
public java.util.Properties getProperties() {
return properties;
}
public EJBContainer getContainer() {
return container;
}
}