/*
* JBoss, Home of Professional Open Source.
* Copyright 2010, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.arquillian.service;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import org.jboss.arquillian.testenricher.msc.ServiceTargetAssociation;
import org.jboss.arquillian.testenricher.osgi.BundleAssociation;
import org.jboss.as.osgi.OSGiConstants;
import org.jboss.as.server.deployment.AttachmentKey;
import org.jboss.as.server.deployment.Attachments;
import org.jboss.as.server.deployment.DeploymentUnit;
import org.jboss.modules.Module;
import org.jboss.msc.service.Service;
import org.jboss.msc.service.ServiceBuilder;
import org.jboss.msc.service.ServiceBuilder.DependencyType;
import org.jboss.msc.service.ServiceController;
import org.jboss.msc.service.ServiceController.Mode;
import org.jboss.msc.service.ServiceName;
import org.jboss.msc.service.ServiceTarget;
import org.jboss.msc.service.StartContext;
import org.jboss.msc.service.StartException;
import org.jboss.msc.service.StopContext;
import org.jboss.msc.value.InjectedValue;
import org.jboss.osgi.framework.Services;
import org.jboss.osgi.resolver.XBundle;
import org.osgi.framework.BundleContext;
/**
* The ArquillianConfig represents an Arquillian deployment.
*
* @author Thomas.Diesler@jboss.com
*/
class ArquillianConfig implements Service<ArquillianConfig> {
static final AttachmentKey<ArquillianConfig> KEY = AttachmentKey.create(ArquillianConfig.class);
private final ArquillianService arqService;
private final DeploymentUnit depUnit;
private final ServiceName serviceName;
private final List<String> testClasses = new ArrayList<String>();
private final InjectedValue<BundleContext> injectedBundleContext = new InjectedValue<BundleContext>();
private ServiceTarget serviceTarget;
static ServiceName getServiceName(DeploymentUnit depUnit) {
return ServiceName.JBOSS.append("arquillian", "config", depUnit.getName());
}
ArquillianConfig(ArquillianService arqService, DeploymentUnit depUnit, Set<String> testClasses) {
this.arqService = arqService;
this.depUnit = depUnit;
this.serviceName = getServiceName(depUnit);
this.testClasses.addAll(testClasses);
}
ServiceBuilder<ArquillianConfig> buildService(ServiceTarget serviceTarget, ServiceController<?> depController) {
ServiceBuilder<ArquillianConfig> builder = serviceTarget.addService(getServiceName(), this);
builder.addDependency(DependencyType.OPTIONAL, Services.FRAMEWORK_CREATE, BundleContext.class, injectedBundleContext);
builder.addDependency(depController.getName());
return builder;
}
ServiceName getServiceName() {
return serviceName;
}
DeploymentUnit getDeploymentUnit() {
return depUnit;
}
List<String> getTestClasses() {
return Collections.unmodifiableList(testClasses);
}
BundleContext getBundleContext() {
return injectedBundleContext.getOptionalValue();
}
Class<?> loadClass(String className) throws ClassNotFoundException {
if (testClasses.contains(className) == false)
throw new ClassNotFoundException("Class '" + className + "' not found in: " + testClasses);
XBundle bundle = depUnit.getAttachment(OSGiConstants.BUNDLE_KEY);
Module module = depUnit.getAttachment(Attachments.MODULE);
if (bundle == null && module == null)
throw new IllegalStateException("Cannot determine deployment type: " + depUnit);
Class<?> testClass;
if (bundle != null) {
testClass = bundle.loadClass(className);
BundleAssociation.setBundle(bundle);
} else {
testClass = module.getClassLoader().loadClass(className);
}
return testClass;
}
@Override
public synchronized void start(StartContext context) throws StartException {
serviceTarget = context.getChildTarget();
arqService.registerArquillianConfig(this);
for(String testClass : testClasses) {
ServiceTargetAssociation.setServiceTarget(testClass, serviceTarget);
}
}
@Override
public synchronized void stop(StopContext context) {
context.getController().setMode(Mode.REMOVE);
arqService.unregisterArquillianConfig(this);
for(String testClass : testClasses) {
ServiceTargetAssociation.clearServiceTarget(testClass);
}
}
@Override
public synchronized ArquillianConfig getValue() {
return this;
}
@Override
public String toString() {
String uname = depUnit.getName();
String sname = serviceName.getCanonicalName();
return "ArquillianConfig[service=" + sname + ",unit=" + uname + ",tests=" + testClasses + "]";
}
}