fop-core/src/test/java/org/apache/fop/afp/AFPResourceUtilTestCase.java

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

/* $Id$ */

package org.apache.fop.afp;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Arrays;

import org.junit.Test;

import static org.junit.Assert.assertTrue;

import org.apache.commons.io.IOUtils;

import org.apache.fop.afp.util.AFPResourceUtil;

/**
 * Tests the {@link AFPResourceUtil} class.
 */
public class AFPResourceUtilTestCase {

    private static final String RESOURCE_FILENAME = "expected_resource.afp";
    private static final String NAMED_RESOURCE_FILENAME = "expected_named_resource.afp";

    private static final String RESOURCE_ANY_NAME = "resource_any_name.afp";
    private static final String RESOURCE_NAME_MATCH = "resource_name_match.afp";
    private static final String RESOURCE_NAME_MISMATCH = "resource_name_mismatch.afp";
    private static final String RESOURCE_NO_END_NAME = "resource_no_end_name.afp";

    private static final String PSEG_A = "XFEATHER";
    private static final String PSEG_B = "S1CODEQR";

    /**
     * Tests copyResourceFile()
     * @throws Exception -
     */
    @Test
    public void testCopyResourceFile() throws Exception {
        compareResources(new ResourceCopier() {
            public void copy(InputStream in, OutputStream out) throws IOException {
                AFPResourceUtil.copyResourceFile(in, out);
            }
        }, RESOURCE_FILENAME, RESOURCE_FILENAME);
    }

    /**
     * Tests copyNamedResource()
     * @throws Exception -
     */
    @Test
    public void testCopyNamedResource() throws Exception {
        compareResources(new ResourceCopier() {
            public void copy(InputStream in, OutputStream out) throws IOException {
                AFPResourceUtil.copyNamedResource(PSEG_A, in, out);
            }
        }, RESOURCE_FILENAME, NAMED_RESOURCE_FILENAME);
    }

    private void compareResources(ResourceCopier copyResource, String resourceA, String resourceB)
            throws IOException {
        ByteArrayOutputStream baos = copyResource(resourceA, copyResource);
        byte[] expectedBytes = resourceAsByteArray(resourceB);
        assertTrue(Arrays.equals(expectedBytes, baos.toByteArray()));
    }

    private ByteArrayOutputStream copyResource(String resource, ResourceCopier resourceCopier)
            throws IOException {
        final ByteArrayOutputStream baos = new ByteArrayOutputStream();
        InputStream in = null;
        try {
            in = getClass().getResourceAsStream(resource);
            resourceCopier.copy(in, baos);
        } finally {
            in.close();
        }
        return baos;
    }

    private byte[] resourceAsByteArray(String resource) throws IOException {
        InputStream in = null;
        byte[] expectedBytes = null;
        try {
            in = getClass().getResourceAsStream(resource);
            expectedBytes = IOUtils.toByteArray(in);
        } finally {
            in.close();
        }
        return expectedBytes;
    }

    /**
     * Tests the validity of a closing structured field having an FF FF name which
     * allows it to match any existing matching starting field
     * @throws Exception -
     */
    @Test
    public void testResourceAnyName() throws Exception {
        testResource(RESOURCE_ANY_NAME, PSEG_B);
    }

    /**
     * Tests a matching end structured field name
     * @throws Exception -
     */
    @Test
    public void testResourceNameMatch() throws Exception {
        testResource(RESOURCE_NAME_MATCH, PSEG_B);
    }

    /**
     * Tests to see whether a matching structured field pair with mismatching
     * names fails.
     * @throws Exception -
     */
    @Test(expected = Exception.class)
    public void testResourceNameMismatch() throws Exception {
        testResource(RESOURCE_NAME_MISMATCH, PSEG_B);
    }

    /**
     * Tests a matching structured end field with no name
     * @throws Exception -
     */
    @Test
    public void testResourceNoEndName() throws Exception {
        testResource(RESOURCE_NO_END_NAME, PSEG_B);
    }

    private void testResource(String resource, final String pseg) throws Exception {
        copyResource(resource, new ResourceCopier() {
            public void copy(InputStream in, OutputStream out) throws IOException {
                AFPResourceUtil.copyNamedResource(pseg, in, out);
            }
        });
    }

    private interface ResourceCopier {
        void copy(InputStream in, OutputStream out) throws IOException;
    }
}