com.izforge.izpack.installer.multiunpacker.MultiVolumeFileUnpackerTest.java Source code

Introduction

Here is the source code for com.izforge.izpack.installer.multiunpacker.MultiVolumeFileUnpackerTest.java

Source

/*
 * IzPack - Copyright 2001-2012 Julien Ponge, All Rights Reserved.
 *
 * http://izpack.org/
 * http://izpack.codehaus.org/
 *
 * Copyright 2012 Tim Anderson
 *
 * 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 com.izforge.izpack.installer.multiunpacker;

import com.izforge.izpack.api.data.Blockable;
import com.izforge.izpack.api.data.OverrideType;
import com.izforge.izpack.api.data.PackFile;
import com.izforge.izpack.api.data.XPackFile;
import com.izforge.izpack.api.exception.InstallerException;
import com.izforge.izpack.core.io.FileSpanningInputStream;
import com.izforge.izpack.core.io.FileSpanningOutputStream;
import com.izforge.izpack.core.io.VolumeLocator;
import com.izforge.izpack.installer.unpacker.AbstractFileUnpackerTest;
import com.izforge.izpack.installer.unpacker.FileQueueFactory;
import com.izforge.izpack.installer.unpacker.FileUnpacker;
import com.izforge.izpack.util.Platforms;
import com.izforge.izpack.util.os.FileQueue;
import org.apache.commons.io.IOUtils;
import org.junit.Test;

import java.io.*;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

/**
 * Tests the {@link MultiVolumeFileUnpacker} class.
 *
 * @author Tim Anderson
 */
public class MultiVolumeFileUnpackerTest extends AbstractFileUnpackerTest {

    /**
     * The first volume.
     */
    private File volume;

    /**
     * The no. of volumes.
     */
    private int volumeCount;

    /**
     * Verifies that the {@link VolumeLocator#getVolume(String, boolean)} method is invoked to prompt
     * for missing media.
     *
     * @throws IOException        for any I/O error
     * @throws InstallerException for any installer error
     */
    @Test
    public void testPromptForNextMedia() throws IOException, InstallerException {
        File baseDir = temporaryFolder.getRoot();
        File source = createSourceFile(baseDir);
        File target = getTargetFile(baseDir);

        // rename the first volume so that it is prompted for
        final File volume1 = new File(volume.getPath() + ".1");
        assertTrue(volume1.exists());
        final File renamed = new File(volume1.getPath() + ".bak");
        assertTrue(volume1.renameTo(renamed));

        VolumeLocator locator = new VolumeLocator() {
            @Override
            public File getVolume(String path, boolean corrupt) throws IOException {
                // rename the file back
                assertTrue(renamed.renameTo(volume1));
                return volume1;
            }
        };

        FileSpanningInputStream stream = new FileSpanningInputStream(volume, volumeCount);
        stream.setLocator(locator);

        FileQueue queue = new FileQueueFactory(Platforms.WINDOWS, getLibrarian()).create();
        FileUnpacker unpacker = new MultiVolumeFileUnpacker(stream, getCancellable(), queue);

        PackFile file = createPackFile(baseDir, source, target, Blockable.BLOCKABLE_NONE);
        assertFalse(target.exists());

        InputStream packStream = createPackStream(source);
        unpacker.unpack(file, packStream, target);
        assertTrue(queue.isEmpty()); // file should not have been queued

        // verify the file unpacked successfully
        checkTarget(source, target);
    }

    /**
     * Creates a new source file.
     *
     * @param baseDir the base directory
     * @return the source file
     * @throws java.io.IOException for any I/O error
     */
    @Override
    protected File createSourceFile(File baseDir) throws IOException {
        File source = new File(baseDir, "source.txt");
        PrintWriter stream = new PrintWriter(source);
        for (int i = 0; i < 20000; ++i) {
            stream.println(i);
        }
        assertFalse(stream.checkError());
        stream.close();

        volume = new File(temporaryFolder.getRoot(), "volume");
        FileSpanningOutputStream out = new FileSpanningOutputStream(volume, 8192);
        FileInputStream in = new FileInputStream(source);
        IOUtils.copy(in, out);

        // verify there is more than one volume
        out.close();
        volumeCount = out.getVolumes();
        assertTrue(volumeCount > 1);
        in.close();
        return source;
    }

    /**
     * Helper to create an unpacker.
     *
     * @param sourceDir the source directory. May be {@code null}
     * @param queue     the file queue
     * @return a new unpacker
     */
    protected FileUnpacker createUnpacker(File sourceDir, FileQueue queue) throws IOException {
        FileSpanningInputStream stream = new FileSpanningInputStream(volume, volumeCount);
        return new MultiVolumeFileUnpacker(stream, getCancellable(), queue);
    }

    /**
     * Helper to create a new pack file.
     *
     * @param baseDir   the base directory
     * @param source    the source file
     * @param target    the target file
     * @param blockable the blockable type
     * @return a new pack file
     * @throws IOException if the source file doesn't exist
     */
    @Override
    protected PackFile createPackFile(File baseDir, File source, File target, Blockable blockable)
            throws IOException {
        // XPackFile required for the Archivefileposition attribute.
        XPackFile result = new XPackFile(baseDir, source, target.getName(), null, OverrideType.OVERRIDE_TRUE, null,
                blockable);
        return result;
    }
}