UberizeMojoTest.java

package org.fusesource.mvnplugins.uberize.mojo;

/*
 * 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.
 */

import java.io.File;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;

import org.fusesource.mvnplugins.uberize.Uberizer;
import org.fusesource.mvnplugins.uberize.relocation.PackageRelocation;
import org.fusesource.mvnplugins.uberize.transformer.PlexusComponents;
import org.fusesource.mvnplugins.uberize.transformer.ClassShader;
import org.codehaus.plexus.PlexusTestCase;

/**
 * @author Jason van Zyl
 * @author Mauro Talevi
 */
public class UberizeMojoTest
    extends PlexusTestCase
{
    public void testShaderWithDefaultShadedPattern()
        throws Exception
    {
        shaderWithPattern(null, new File( "target/foo-default.jar" ));
    }

    public void testShaderWithCustomShadedPattern()
        throws Exception
    {
        shaderWithPattern("org/shaded/plexus/util", new File( "target/foo-custom.jar" ));
    }

    public void testShaderWithExclusions()
        throws Exception
    {
        File workDir = new File( getBasedir(), "target/unit/uberize" );
        File jarFile = new File( getBasedir(), "target/unit/foo-bar.jar" );

        Uberizer s = (Uberizer) lookup( Uberizer.ROLE );

        Set sources = new LinkedHashSet();
        sources.add( new File( getBasedir(), "src/test/jars/test-artifact-1.0-SNAPSHOT.jar" ) );

        List relocators = new ArrayList();

        ClassShader shader = new ClassShader();
        shader.relocations = new PackageRelocation[] {
            new PackageRelocation("org.codehaus.plexus.util", "hidden", Arrays.asList( new String[] {
            "org.codehaus.plexus.util.xml.Xpp3Dom", "org.codehaus.plexus.util.xml.pull.*" } ))
        };

        List transformers = new ArrayList();
        transformers.add(shader);

        List filters = new ArrayList();

        s.uberize(workDir,  sources, jarFile, filters, transformers );

        final URL url = jarFile.toURI().toURL();
        ClassLoader cl = new URLClassLoader( new URL[] {url} );
        Class c = cl.loadClass( "org.fusesource.mvnplugins.uberize.Lib" );

        Field field = c.getDeclaredField( "CLASS_REALM_PACKAGE_IMPORT" );
        assertEquals( "org.codehaus.plexus.util.xml.pull", field.get( null ) );

        Method method = c.getDeclaredMethod( "getClassRealmPackageImport", new Class[0] );
        assertEquals( "org.codehaus.plexus.util.xml.pull", method.invoke( null, new Object[0] ) );
    }

    public void shaderWithPattern(String shadedPattern, File jar)
        throws Exception
    {
        File workDir = new File( getBasedir(), "target/uberize" );
        Uberizer s = (Uberizer) lookup( Uberizer.ROLE );

        Set sources = new LinkedHashSet();
        sources.add( new File( getBasedir(), "src/test/jars/test-project-1.0-SNAPSHOT.jar" ) );
        sources.add( new File( getBasedir(), "src/test/jars/plexus-utils-1.4.1.jar" ) );

        List relocators = new ArrayList();

        ClassShader shader = new ClassShader();
        shader.relocations = new PackageRelocation[] {
            new PackageRelocation("org/codehaus/plexus/util", shadedPattern, Arrays.asList(
            new String[]{"org/codehaus/plexus/util/xml/Xpp3Dom", "org/codehaus/plexus/util/xml/pull.*"}))
        };

        List transformers = new ArrayList();
        transformers.add(shader);
        transformers.add( new PlexusComponents() );

        List filters = new ArrayList();

        s.uberize(workDir,  sources, jar, filters, transformers );
    }

}