eu.unifiedviews.plugins.transformer.gunzipper.GunzipperTest.java Source code

Introduction

Here is the source code for eu.unifiedviews.plugins.transformer.gunzipper.GunzipperTest.java

Source

package eu.unifiedviews.plugins.transformer.gunzipper;

import java.io.File;
import java.io.FileOutputStream;
import java.net.URI;
import java.util.Set;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.junit.Assert;
import org.junit.Test;

import cz.cuni.mff.xrg.odcs.dpu.test.TestEnvironment;
import eu.unifiedviews.dataunit.files.FilesDataUnit;
import eu.unifiedviews.dataunit.files.WritableFilesDataUnit;
import eu.unifiedviews.helpers.dataunit.files.FilesHelper;
import eu.unifiedviews.helpers.dataunit.virtualpath.VirtualPathHelpers;
import eu.unifiedviews.helpers.dpu.test.config.ConfigurationBuilder;

public class GunzipperTest {
    @Test
    public void testSmallFile() throws Exception {
        GunzipperConfig_V1 config = new GunzipperConfig_V1();

        // Prepare DPU.
        Gunzipper dpu = new Gunzipper();
        dpu.configure((new ConfigurationBuilder()).setDpuConfiguration(config).toString());

        // Prepare test environment.
        TestEnvironment environment = new TestEnvironment();

        // Prepare data unit.
        WritableFilesDataUnit filesOutput = environment.createFilesOutput("filesOutput");
        WritableFilesDataUnit filesInput = environment.createFilesInput("filesInput");

        File inputFile = new File(URI.create(filesInput.addNewFile("LICENSE.gz")));
        try (FileOutputStream fout = new FileOutputStream(inputFile)) {
            IOUtils.copy(Thread.currentThread().getContextClassLoader().getResourceAsStream("LICENSE.gz"), fout);
        }
        try {
            // Run.
            environment.run(dpu);

            // Get file iterator.
            Set<FilesDataUnit.Entry> outputFiles = FilesHelper.getFiles(filesOutput);
            Assert.assertEquals(1, outputFiles.size());

            FilesDataUnit.Entry entry = outputFiles.iterator().next();
            String outputContent = IOUtils.toString(new URI(entry.getFileURIString()), "US-ASCII");
            String expectedContent = IOUtils.toString(
                    Thread.currentThread().getContextClassLoader().getResourceAsStream("LICENSE"), "US-ASCII");

            Assert.assertEquals(expectedContent, outputContent);
            Assert.assertEquals("LICENSE", VirtualPathHelpers.getVirtualPath(filesOutput, "LICENSE.gz"));
        } finally {
            // Release resources.
            environment.release();
        }
    }
}