URIClassLoaderTest.java

/*
 * See the NOTICE file distributed with this work for additional
 * information regarding copyright ownership.
 *
 * 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.xwiki.classloader;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.net.URI;
import java.net.URLStreamHandlerFactory;
import java.util.jar.JarEntry;
import java.util.jar.JarOutputStream;

import org.junit.Assert;

import org.jmock.Expectations;
import org.junit.Test;
import org.xwiki.bridge.DocumentAccessBridge;
import org.xwiki.model.reference.AttachmentReference;
import org.xwiki.model.reference.AttachmentReferenceResolver;
import org.xwiki.model.reference.DocumentReference;
import org.xwiki.test.jmock.AbstractComponentTestCase;

/**
 * Unit tests for {@link URIClassLoader}.
 * 
 * @version $Id: ac0c0d64e76ac0691fc06b0b2de24f9a598b55cb $
 */
public class URIClassLoaderTest extends AbstractComponentTestCase
{
    private AttachmentReferenceResolver<String> arf;

    private DocumentAccessBridge dab;

    @SuppressWarnings("unchecked")
    @Override
    protected void registerComponents() throws Exception
    {
        super.registerComponents();

        this.arf = registerMockComponent(AttachmentReferenceResolver.TYPE_STRING, "current");
        this.dab = registerMockComponent(DocumentAccessBridge.class);
    }

    /**
     * Verify that resource located in a URI with an attachmentjar protocol can be found.
     */
    @Test
    public void testFindResource() throws Exception
    {
        URLStreamHandlerFactory urlStreamHandlerFactory =
            getComponentManager().getInstance(URLStreamHandlerFactory.class);
        URIClassLoader cl =
            new URIClassLoader(new URI[] {new URI("attachmentjar://page%40filename1"), new URI("http://some/url"),
            new URI("attachmentjar://filename2")}, urlStreamHandlerFactory);

        Assert.assertEquals(3, cl.getURLs().length);
        Assert.assertEquals("attachmentjar://page%40filename1", cl.getURLs()[0].toString());
        Assert.assertEquals("http://some/url", cl.getURLs()[1].toString());
        Assert.assertEquals("attachmentjar://filename2", cl.getURLs()[2].toString());

        final AttachmentReference attachmentName1 = new AttachmentReference("filename1",
            new DocumentReference("wiki", "space", "page"));
        final AttachmentReference attachmentName2 = new AttachmentReference("filename2",
            new DocumentReference("wiki", "space", "page"));

        getMockery().checking(new Expectations()
        {
            {
                allowing(URIClassLoaderTest.this.arf).resolve("page@filename1");
                will(returnValue(attachmentName1));
                oneOf(URIClassLoaderTest.this.dab).getAttachmentContent(attachmentName1);
                will(returnValue(new ByteArrayInputStream(createJarFile("/nomatch"))));
                allowing(URIClassLoaderTest.this.arf).resolve("filename2");
                will(returnValue(attachmentName2));
                oneOf(URIClassLoaderTest.this.dab).getAttachmentContent(attachmentName2);
                will(returnValue(new ByteArrayInputStream(createJarFile("/something"))));
            }
        });

        Assert.assertEquals("jar:attachmentjar://filename2!/something", cl.findResource("/something").toString());
    }

    private byte[] createJarFile(String resourceName) throws Exception
    {
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        JarOutputStream jos = new JarOutputStream(baos);
        JarEntry entry = new JarEntry(resourceName);
        jos.putNextEntry(entry);
        jos.write("whatever".getBytes("UTF-8"));
        jos.closeEntry();
        jos.close();
        return baos.toByteArray();
    }
}