/*
* JBoss, Home of Professional Open Source
* Copyright 2009, Red Hat Middleware LLC, and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* Licensed 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.jboss.as.arquillian.protocol.jmx;
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.jar.Attributes;
import java.util.jar.JarFile;
import java.util.jar.Manifest;
import org.jboss.arquillian.container.test.spi.RemoteLoadableExtension;
import org.jboss.arquillian.container.test.spi.TestDeployment;
import org.jboss.arquillian.container.test.spi.client.deployment.DeploymentPackager;
import org.jboss.arquillian.container.test.spi.client.deployment.ProtocolArchiveProcessor;
import org.jboss.arquillian.protocol.jmx.AbstractJMXProtocol;
import org.jboss.as.arquillian.protocol.jmx.JMXProtocolAS7.ServiceArchiveHolder;
import org.jboss.as.arquillian.service.ArquillianService;
import org.jboss.as.arquillian.service.JMXProtocolEndpointExtension;
import org.jboss.logging.Logger;
import org.jboss.msc.service.ServiceActivator;
import org.jboss.osgi.spi.util.BundleInfo;
import org.jboss.osgi.testing.ManifestBuilder;
import org.jboss.shrinkwrap.api.Archive;
import org.jboss.shrinkwrap.api.ArchivePath;
import org.jboss.shrinkwrap.api.ArchivePaths;
import org.jboss.shrinkwrap.api.Node;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.asset.Asset;
import org.jboss.shrinkwrap.api.asset.UrlAsset;
import org.jboss.shrinkwrap.api.container.ManifestContainer;
import org.jboss.shrinkwrap.api.spec.JavaArchive;
/**
* A {@link DeploymentPackager} for the JMXProtocol.
*
* It dynamically generates the arquillian-archive, which is deployed to the
* target container before the test run.
*
* @see ArquillianServiceDeployer
*
* @author Thomas.Diesler@jboss.com
* @since 01-Jul-2011
*/
public class JMXProtocolPackager implements DeploymentPackager {
private static final List<String> defaultDependencies = new ArrayList<String>();
static {
defaultDependencies.add("deployment.arquillian-service");
defaultDependencies.add("org.jboss.modules");
defaultDependencies.add("org.jboss.msc");
}
private static final Logger log = Logger.getLogger(JMXProtocolPackager.class);
private ServiceArchiveHolder archiveHolder;
JMXProtocolPackager(ServiceArchiveHolder archiveHolder) {
this.archiveHolder = archiveHolder;
}
@Override
public Archive<?> generateDeployment(TestDeployment testDeployment, Collection<ProtocolArchiveProcessor> protocolProcessors) {
final Archive<?> appArchive = testDeployment.getApplicationArchive();
if (archiveHolder.getArchive() == null) {
final Collection<Archive<?>> auxArchives = testDeployment.getAuxiliaryArchives();
JavaArchive archive = generateArquillianServiceArchive(auxArchives);
archiveHolder.setArchive(archive);
}
addModulesManifestDependencies(appArchive);
archiveHolder.addPreparedDeployment(appArchive);
return appArchive;
}
private JavaArchive generateArquillianServiceArchive(Collection<Archive<?>> auxArchives) {
final JavaArchive archive = ShrinkWrap.create(JavaArchive.class, "arquillian-service");
log.debugf("Generating: %s", archive.getName());
archive.addPackage(ArquillianService.class.getPackage());
archive.addPackage(AbstractJMXProtocol.class.getPackage());
// Merge the auxilliary archives and collect the loadable extensions
final Set<String> loadableExtensions = new HashSet<String>();
final String loadableExtentionsPath = "META-INF/services/" + RemoteLoadableExtension.class.getName();
for (Archive<?> aux : auxArchives) {
Node node = aux.get(loadableExtentionsPath);
if (node != null) {
BufferedReader br = new BufferedReader(new InputStreamReader(node.getAsset().openStream()));
try {
String line = br.readLine();
while (line != null) {
loadableExtensions.add(line);
line = br.readLine();
}
} catch (IOException ex) {
// ignore
}
}
log.debugf("Merging archive: %s", aux);
archive.merge(aux);
}
loadableExtensions.add(JMXProtocolEndpointExtension.class.getName());
// Generate the manifest with it's dependencies
archive.setManifest(new Asset() {
public InputStream openStream() {
ManifestBuilder builder = ManifestBuilder.newInstance();
StringBuffer dependencies = new StringBuffer();
dependencies.append("org.jboss.as.jmx,");
dependencies.append("org.jboss.as.server,");
dependencies.append("org.jboss.as.osgi,");
dependencies.append("org.jboss.jandex,");
dependencies.append("org.jboss.logging,");
dependencies.append("org.jboss.modules,");
dependencies.append("org.jboss.msc,");
dependencies.append("org.jboss.osgi.framework,");
dependencies.append("org.osgi.core");
builder.addManifestHeader("Dependencies", dependencies.toString());
return builder.openStream();
}
});
// Add the ServiceActivator
String serviceActivatorPath = "META-INF/services/" + ServiceActivator.class.getName();
final URL serviceActivatorURL = this.getClass().getClassLoader().getResource("arquillian-service/" + serviceActivatorPath);
if (serviceActivatorURL == null) {
throw new RuntimeException("No arquillian-service/" + serviceActivatorPath + " found by classloader: " + this.getClass().getClassLoader());
}
archive.addAsResource(new UrlAsset(serviceActivatorURL), serviceActivatorPath);
// Replace the loadable extensions with the collected set
archive.delete(ArchivePaths.create(loadableExtentionsPath));
archive.addAsResource(new Asset() {
@Override
public InputStream openStream() {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
PrintWriter pw = new PrintWriter(new OutputStreamWriter(baos));
for (String line : loadableExtensions) {
pw.println(line);
}
pw.close();
return new ByteArrayInputStream(baos.toByteArray());
}
}, loadableExtentionsPath);
log.debugf("Loadable extensions: %s", loadableExtensions);
log.tracef("Archive content: %s\n%s", archive, archive.toString(true));
return archive;
}
/**
* Adds the Manifest Attribute "Dependencies" with the required dependencies for JBoss Modules to depend on the Arquillian Service.
*
* @param appArchive The Archive to deploy
*/
private void addModulesManifestDependencies(Archive<?> appArchive) {
if (appArchive instanceof ManifestContainer<?> == false)
throw new IllegalArgumentException("ManifestContainer expected " + appArchive);
final Manifest manifest = ManifestUtils.getOrCreateManifest(appArchive);
// Don't enrich with Modules Dependencies if this is a OSGi bundle
if(BundleInfo.isValidBundleManifest(manifest)) {
return;
}
Attributes attributes = manifest.getMainAttributes();
if (attributes.getValue(Attributes.Name.MANIFEST_VERSION.toString()) == null) {
attributes.putValue(Attributes.Name.MANIFEST_VERSION.toString(), "1.0");
}
String value = attributes.getValue("Dependencies");
StringBuffer moduleDeps = new StringBuffer(value != null && value.trim().length() > 0 ? value : "org.jboss.modules");
for (String dep : defaultDependencies) {
if (moduleDeps.indexOf(dep) < 0)
moduleDeps.append("," + dep);
}
log.debugf("Add dependencies: %s", moduleDeps);
attributes.putValue("Dependencies", moduleDeps.toString());
// Add the manifest to the archive
ArchivePath manifestPath = ArchivePaths.create(JarFile.MANIFEST_NAME);
appArchive.delete(manifestPath);
appArchive.add(new Asset() {
public InputStream openStream() {
try {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
manifest.write(baos);
return new ByteArrayInputStream(baos.toByteArray());
} catch (IOException ex) {
throw new IllegalStateException("Cannot write manifest", ex);
}
}
}, manifestPath);
}
}