/*
 * The MIT License
 *
 * Copyright (c) 2009 The Broad Institute
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package htsjdk.samtools.util;

import htsjdk.HtsjdkTest;
import com.google.common.jimfs.Configuration;
import com.google.common.jimfs.Jimfs;

import java.io.*;
import java.net.URI;
import java.nio.file.FileSystem;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.spi.FileSystemProvider;

import htsjdk.samtools.SAMException;
import org.testng.Assert;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

import java.lang.IllegalArgumentException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;
import java.util.Random;
import java.util.zip.GZIPOutputStream;


public class IOUtilTest extends HtsjdkTest {


    private static final Path TEST_DATA_DIR = Paths.get ("src/test/resources/htsjdk/samtools/io/");
    private static final Path TEST_VARIANT_DIR = Paths.get("src/test/resources/htsjdk/variant/");
    private static final Path SLURP_TEST_FILE = TEST_DATA_DIR.resolve("slurptest.txt");
    private static final Path EMPTY_FILE = TEST_DATA_DIR.resolve("empty.txt");
    private static final Path FIVE_SPACES_THEN_A_NEWLINE_THEN_FIVE_SPACES_FILE = TEST_DATA_DIR.resolve("5newline5.txt");
    private static final List<String> SLURP_TEST_LINES = Arrays.asList("bacon   and rice   ", "for breakfast  ", "wont you join me");
    private static final String SLURP_TEST_LINE_SEPARATOR = "\n";
    private static final String TEST_FILE_PREFIX = "htsjdk-IOUtilTest";
    private static final String[] TEST_FILE_EXTENSIONS = {".txt", ".txt.gz"};
    private static final String TEST_STRING = "bar!";

    private File existingTempFile;
    private String systemUser;
    private String systemTempDir;
    private FileSystem inMemoryFileSystem;
    private static Path WORDS_LONG;

    @BeforeClass
    public void setUp() throws IOException {
        existingTempFile = File.createTempFile("FiletypeTest.", ".tmp");
        existingTempFile.deleteOnExit();
        systemTempDir = System.getProperty("java.io.tmpdir");
        final File tmpDir = new File(systemTempDir);
        inMemoryFileSystem = Jimfs.newFileSystem(Configuration.unix());;
        if (!tmpDir.isDirectory()) tmpDir.mkdir();
        if (!tmpDir.isDirectory())
            throw new RuntimeException("java.io.tmpdir (" + systemTempDir + ") is not a directory");
        systemUser = System.getProperty("user.name");
        //build long file of random words for compression testing
        WORDS_LONG = Files.createTempFile("words_long", ".txt");
        WORDS_LONG.toFile().deleteOnExit();
        final List<String> wordsList = Files.lines(TEST_DATA_DIR.resolve("dictionary_english_short.dic")).collect(Collectors.toList());
        final int numberOfWords = 300000;
        final int seed = 345987345;
        final Random rand = new Random(seed);
        try (final BufferedWriter writer = Files.newBufferedWriter(WORDS_LONG)) {
            for (int i = 0; i < numberOfWords; i++) {
                writer.write(wordsList.get(rand.nextInt(wordsList.size())));
            }
        }
    }

    @AfterClass
    public void tearDown() throws IOException {
        // reset java properties to original
        System.setProperty("java.io.tmpdir", systemTempDir);
        System.setProperty("user.name", systemUser);
        inMemoryFileSystem.close();
    }

    @Test
    public void testFileReadingAndWriting() throws IOException {
        String randomizedTestString = TEST_STRING + System.currentTimeMillis();
        for (String ext : TEST_FILE_EXTENSIONS) {
            File f = File.createTempFile(TEST_FILE_PREFIX, ext);
            f.deleteOnExit();

            OutputStream os = IOUtil.openFileForWriting(f);
            BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(os));
            writer.write(randomizedTestString);
            writer.close();

            InputStream is = IOUtil.openFileForReading(f);
            BufferedReader reader = new BufferedReader(new InputStreamReader(is));
            String line = reader.readLine();
            Assert.assertEquals(randomizedTestString, line);
        }
    }

    @Test(groups = {"unix"})
    public void testGetCanonicalPath() throws IOException {
        String tmpPath = System.getProperty("java.io.tmpdir");
        String userName = System.getProperty("user.name");

        if (tmpPath.endsWith(userName)) {
            tmpPath = tmpPath.substring(0, tmpPath.length() - userName.length());
        }

        File tmpDir = new File(tmpPath, userName);
        tmpDir.mkdir();
        tmpDir.deleteOnExit();
        File actual = new File(tmpDir, "actual.txt");
        actual.deleteOnExit();
        ProcessExecutor.execute(new String[]{"touch", actual.getAbsolutePath()});
        File symlink = new File(tmpDir, "symlink.txt");
        symlink.deleteOnExit();
        ProcessExecutor.execute(new String[]{"ln", "-s", actual.getAbsolutePath(), symlink.getAbsolutePath()});
        File lnDir = new File(tmpDir, "symLinkDir");
        lnDir.deleteOnExit();
        ProcessExecutor.execute(new String[]{"ln", "-s", tmpDir.getAbsolutePath(), lnDir.getAbsolutePath()});
        File lnToActual = new File(lnDir, "actual.txt");
        lnToActual.deleteOnExit();
        File lnToSymlink = new File(lnDir, "symlink.txt");
        lnToSymlink.deleteOnExit();

        File[] files = {actual, symlink, lnToActual, lnToSymlink};
        for (File f : files) {
            Assert.assertEquals(IOUtil.getFullCanonicalPath(f), actual.getCanonicalPath());
        }
    }

    @Test
    public void testUtfWriting() throws IOException {
        final String utf8 = new StringWriter().append((char) 168).append((char) 197).toString();
        for (String ext : TEST_FILE_EXTENSIONS) {
            final File f = File.createTempFile(TEST_FILE_PREFIX, ext);
            f.deleteOnExit();

            final BufferedWriter writer = IOUtil.openFileForBufferedUtf8Writing(f);
            writer.write(utf8);
            CloserUtil.close(writer);

            final BufferedReader reader = IOUtil.openFileForBufferedUtf8Reading(f);
            final String line = reader.readLine();
            Assert.assertEquals(utf8, line, f.getAbsolutePath());

            CloserUtil.close(reader);

        }
    }

    @Test
    public void slurpLinesTest() throws FileNotFoundException {
        Assert.assertEquals(IOUtil.slurpLines(SLURP_TEST_FILE.toFile()), SLURP_TEST_LINES);
    }

    @Test
    public void slurpWhitespaceOnlyFileTest() throws FileNotFoundException {
        Assert.assertEquals(IOUtil.slurp(FIVE_SPACES_THEN_A_NEWLINE_THEN_FIVE_SPACES_FILE.toFile()), "     \n     ");
    }

    @Test
    public void slurpEmptyFileTest() throws FileNotFoundException {
        Assert.assertEquals(IOUtil.slurp(EMPTY_FILE.toFile()), "");
    }

    @Test
    public void slurpTest() throws FileNotFoundException {
        Assert.assertEquals(IOUtil.slurp(SLURP_TEST_FILE.toFile()), CollectionUtil.join(SLURP_TEST_LINES, SLURP_TEST_LINE_SEPARATOR));
    }

    @Test(dataProvider = "fileTypeTestCases")
    public void testFileType(final String path, boolean expectedIsRegularFile) {
        final File file = new File(path);
        Assert.assertEquals(IOUtil.isRegularPath(file), expectedIsRegularFile);
        Assert.assertEquals(IOUtil.isRegularPath(file.toPath()), expectedIsRegularFile);
    }

    @Test(dataProvider = "unixFileTypeTestCases", groups = {"unix"})
    public void testFileTypeUnix(final String path, boolean expectedIsRegularFile) {
        final File file = new File(path);
        Assert.assertEquals(IOUtil.isRegularPath(file), expectedIsRegularFile);
        Assert.assertEquals(IOUtil.isRegularPath(file.toPath()), expectedIsRegularFile);
    }

    @Test
    public void testAddExtension() throws IOException {
        Path p = IOUtil.getPath("/folder/file");
        Assert.assertEquals(IOUtil.addExtension(p, ".ext"), IOUtil.getPath("/folder/file.ext"));
        p = IOUtil.getPath("folder/file");
        Assert.assertEquals(IOUtil.addExtension(p, ".ext"), IOUtil.getPath("folder/file.ext"));
        try (FileSystem jimfs = Jimfs.newFileSystem(Configuration.unix())) {
            p = jimfs.getPath("folder/sub/file");
            Assert.assertEquals(IOUtil.addExtension(p, ".ext"), jimfs.getPath("folder/sub/file.ext"));
            p = jimfs.getPath("folder/file");
            Assert.assertEquals(IOUtil.addExtension(p, ".ext"), jimfs.getPath("folder/file.ext"));
            p = jimfs.getPath("file");
            Assert.assertEquals(IOUtil.addExtension(p, ".ext"), jimfs.getPath("file.ext"));
        }
    }

    @Test
    public void testAddExtensionOnList() throws IOException {
        Path p = IOUtil.getPath("/folder/file");
        List<FileSystemProvider> fileSystemProviders = FileSystemProvider.installedProviders();

        List<Path> paths = new ArrayList<>();
        List<String> strings = new ArrayList<>();

        paths.add(IOUtil.addExtension(p, ".ext"));
        strings.add("/folder/file.ext");

        p = IOUtil.getPath("folder/file");
        paths.add(IOUtil.addExtension(p, ".ext"));
        strings.add("folder/file.ext");

        List<Path> expectedPaths = IOUtil.getPaths(strings);

        Assert.assertEquals(paths, expectedPaths);
    }


    @DataProvider(name = "fileTypeTestCases")
    private Object[][] fileTypeTestCases() {
        return new Object[][]{
                {existingTempFile.getAbsolutePath(), Boolean.TRUE},
                {systemTempDir, Boolean.FALSE}

        };
    }

    @DataProvider(name = "unixFileTypeTestCases")
    private Object[][] unixFileTypeTestCases() {
        return new Object[][]{
                {"/dev/null", Boolean.FALSE},
                {"/dev/stdout", Boolean.FALSE},
                {"/non/existent/file", Boolean.TRUE},
        };
    }

    @DataProvider
    public Object[][] getFiles(){
        final File file = new File("someFile");
        return new Object[][] {
                {null, null},
                {file, file.toPath()}
        };
    }

    @Test(dataProvider = "getFiles")
    public void testToPath(final File file, final Path expected){
        Assert.assertEquals(IOUtil.toPath(file), expected);
    }


    @DataProvider(name = "fileNamesForDelete")
    public Object[][] fileNamesForDelete() {
        return new Object[][] {
                {Collections.emptyList()},
                {Collections.singletonList("file1")},
                {Arrays.asList("file1", "file2")}
        };
    }

    @Test
    public void testGetDefaultTmpDirPath() throws Exception {
        try {
            Path testPath = IOUtil.getDefaultTmpDirPath();
            Assert.assertEquals(testPath.toFile().getAbsolutePath(), new File(systemTempDir).getAbsolutePath() + "/" + systemUser);

            // change the properties to test others
            final String newTempPath = Files.createTempDirectory("testGetDefaultTmpDirPath").toString();
            final String newUser = "my_user";
            System.setProperty("java.io.tmpdir", newTempPath);
            System.setProperty("user.name", newUser);
            testPath = IOUtil.getDefaultTmpDirPath();
            Assert.assertEquals(testPath.toFile().getAbsolutePath(), new File(newTempPath).getAbsolutePath() + "/" + newUser);

        } finally {
            // reset system properties
            System.setProperty("java.io.tmpdir", systemTempDir);
            System.setProperty("user.name", systemUser);
        }
    }

    @Test(dataProvider = "fileNamesForDelete")
    public void testDeletePathLocal(final List<String> fileNames) throws Exception {
        final Path tmpDir = IOUtil.createTempDir("testDeletePath");
        final List<Path> paths = createLocalFiles(tmpDir, fileNames);
        testDeletePaths(paths);
    }

    @Test
    public void testDeleteSinglePath() throws Exception {
        final Path toDelete = Files.createTempFile("file",".bad");
        Assert.assertTrue(Files.exists(toDelete));
        IOUtil.deletePath(toDelete);
        Assert.assertFalse(Files.exists(toDelete));
    }

    @Test
    public void testDeleteSingleWithDeletePaths() throws Exception {
        final Path toDelete = Files.createTempFile("file",".bad");
        Assert.assertTrue(Files.exists(toDelete));
        IOUtil.deletePaths(toDelete);
        Assert.assertFalse(Files.exists(toDelete));
    }

    @Test(dataProvider = "fileNamesForDelete")
    public void testDeletePathJims(final List<String> fileNames) throws Exception {
        final List<Path> paths = createJimfsFiles("testDeletePath", fileNames);
        testDeletePaths(paths);
    }

    @Test(dataProvider = "fileNamesForDelete")
    public void testDeleteArrayPathLocal(final List<String> fileNames) throws Exception {
        final Path tmpDir = IOUtil.createTempDir("testDeletePath");
        final List<Path> paths = createLocalFiles(tmpDir, fileNames);
        testDeletePathArray(paths);
    }

    @Test(dataProvider = "fileNamesForDelete")
    public void testDeleteArrayPathJims(final List<String> fileNames) throws Exception {
        final List<Path> paths = createJimfsFiles("testDeletePath", fileNames);
        testDeletePathArray(paths);
    }


    private static void testDeletePaths(final List<Path> paths) {
        paths.forEach(p -> Assert.assertTrue(Files.exists(p)));
        IOUtil.deletePaths(paths);
        paths.forEach(p -> Assert.assertFalse(Files.exists(p)));
    }

    private static void testDeletePathArray(final List<Path> paths) {
        paths.forEach(p -> Assert.assertTrue(Files.exists(p)));
        IOUtil.deletePaths(paths.toArray(new Path[paths.size()]));
        paths.forEach(p -> Assert.assertFalse(Files.exists(p)));
    }

    private static List<Path> createLocalFiles(final Path tmpDir, final List<String> fileNames) throws Exception {
        final List<Path> paths = new ArrayList<>(fileNames.size());
        for (final String f: fileNames) {
            final Path file = Files.createFile(tmpDir.resolve(f));
            paths.add(file);
        }
        return paths;
    }

    private List<Path> createJimfsFiles(final String folderName, final List<String> fileNames) throws Exception {
        final List<Path> paths = new ArrayList<>(fileNames.size());
        final Path folder = inMemoryFileSystem.getPath(folderName);
        if (Files.notExists(folder)) Files.createDirectory(folder);

        for (final String f: fileNames) {
            final Path p = inMemoryFileSystem.getPath(folderName, f);
            Files.createFile(p);
            paths.add(p);
        }

        return paths;
    }

    @DataProvider
    public Object[][] pathsForWritableDirectory() throws Exception {
        return new Object[][] {
                // non existent
                {inMemoryFileSystem.getPath("no_exists"), false},
                // non directory
                {Files.createFile(inMemoryFileSystem.getPath("testAssertDirectoryIsWritable_file")), false},
                // TODO - how to do in inMemoryFileSystem a non-writable directory?
                // writable directory
                {Files.createDirectory(inMemoryFileSystem.getPath("testAssertDirectoryIsWritable_directory")), true}
        };
    }

    @Test(dataProvider = "pathsForWritableDirectory")
    public void testAssertDirectoryIsWritablePath(final Path path, final boolean writable) {
        try {
            IOUtil.assertDirectoryIsWritable(path);
        } catch (SAMException e) {
            if (writable) {
                Assert.fail(e.getMessage());
            }
        }
    }

    @DataProvider
    public Object[][] filesForWritableDirectory() throws Exception {
        final File nonWritableFile = new File(systemTempDir, "testAssertDirectoryIsWritable_non_writable_dir");
        nonWritableFile.mkdir();
        nonWritableFile.setWritable(false);

        return new Object[][] {
                // non existent
                {new File("no_exists"), false},
                // non directory
                {existingTempFile, false},
                // non-writable directory
                {nonWritableFile, false},
                // writable directory
                {new File(systemTempDir), true},
        };
    }

    @Test(dataProvider = "filesForWritableDirectory")
    public void testAssertDirectoryIsWritableFile(final File file, final boolean writable) {
        try {
            IOUtil.assertDirectoryIsWritable(file);
        } catch (SAMException e) {
            if (writable) {
                Assert.fail(e.getMessage());
            }
        }
    }

    static final String level1 = "Level1.fofn";
    static final String level2 = "Level2.fofn";

    @DataProvider
    public Object[][] fofnData() throws IOException {

        Path fofnPath1 = inMemoryFileSystem.getPath(level1);
        Files.copy(TEST_DATA_DIR.resolve(level1), fofnPath1);

        Path fofnPath2 = inMemoryFileSystem.getPath(level2);
        Files.copy(TEST_DATA_DIR.resolve(level2), fofnPath2);

        return new Object[][]{
                {TEST_DATA_DIR + "/" + level1, new String[]{".vcf", ".vcf.gz"}, 2},
                {TEST_DATA_DIR + "/" + level2, new String[]{".vcf", ".vcf.gz"}, 4},
                {fofnPath1.toUri().toString(), new String[]{".vcf", ".vcf.gz"}, 2},
                {fofnPath2.toUri().toString(), new String[]{".vcf", ".vcf.gz"}, 4}
        };
    }

    @Test(dataProvider = "fofnData")
    public void testUnrollPaths(final String pathUri, final String[] extensions, final int expectedNumberOfUnrolledPaths) throws IOException {
        Path p = IOUtil.getPath(pathUri);
        List<Path> paths = IOUtil.unrollPaths(Collections.singleton(p), extensions);

        Assert.assertEquals(paths.size(), expectedNumberOfUnrolledPaths);
    }

    @DataProvider(name = "blockCompressedExtensionExtensionStrings")
    public static Object[][] createBlockCompressedExtensionStrings() {
        return new Object[][] {
                { "testzip.gz", true },
                { "test.gzip", true },
                { "test.bgz", true },
                { "test.bgzf", true },
                { "test.bzip2", false }
        };
    }

    @Test(dataProvider = "blockCompressedExtensionExtensionStrings")
    public void testBlockCompressionExtensionString(final String testString, final boolean expected) {
        Assert.assertEquals(IOUtil.hasBlockCompressedExtension(testString), expected);
    }

    @Test(dataProvider = "blockCompressedExtensionExtensionStrings")
    public void testBlockCompressionExtensionFile(final String testString, final boolean expected) {
        Assert.assertEquals(IOUtil.hasBlockCompressedExtension(new File(testString)), expected);
    }

    @DataProvider(name = "blockCompressedExtensionExtensionURIStrings")
    public static Object[][] createBlockCompressedExtensionURIs() {
        return new Object[][]{
                {"testzip.gz", true},
                {"test.gzip", true},
                {"test.bgz", true},
                {"test.bgzf", true},
                {"test", false},
                {"test.bzip2", false},

                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877.vcf.gz", true},
                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877.vcf.gzip", true},
                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877.vcf.bgz", true},
                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877.vcf.bgzf", true},
                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877.vcf.bzip2", false},
                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877", false},

                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877.vcf.gz?alt=media", true},
                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877.vcf.gzip?alt=media", true},
                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877.vcf.bgz?alt=media", true},
                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877.vcf.bgzf?alt=media", true},
                {"https://www.googleapis.com/download/storage/v1/b/deflaux-public-test/o/NA12877.vcf.bzip2?alt=media", false},

                {"ftp://ftp.broadinstitute.org/distribution/igv/TEST/cpgIslands.hg18.gz", true},
                {"ftp://ftp.broadinstitute.org/distribution/igv/TEST/cpgIslands.hg18.bed", false}
        };
    }

    @Test(dataProvider = "blockCompressedExtensionExtensionURIStrings")
    public void testBlockCompressionExtension(final String testURIString, final boolean expected) {
        URI testURI = URI.create(testURIString);
        Assert.assertEquals(IOUtil.hasBlockCompressedExtension(testURI), expected);
    }

    @Test(dataProvider = "blockCompressedExtensionExtensionURIStrings")
    public void testBlockCompressionExtensionStringVersion(final String testURIString, final boolean expected) {
        Assert.assertEquals(IOUtil.hasBlockCompressedExtension(testURIString), expected);
    }

    @DataProvider
    public static Object[][] blockCompressedFiles() {
        return new Object[][]{
                {TEST_DATA_DIR.resolve("ipsum.txt"), true, false},
                {TEST_DATA_DIR.resolve("ipsum.txt"), false, false},
                {TEST_DATA_DIR.resolve("ipsum.txt.gz"), true, false},
                {TEST_DATA_DIR.resolve("ipsum.txt.gz"), false, false},
                {TEST_DATA_DIR.resolve("ipsum.txt.bgz"), true, true},
                {TEST_DATA_DIR.resolve("ipsum.txt.bgz"), false, true},
                {TEST_DATA_DIR.resolve("ipsum.txt.bgz.wrongextension"), true, false},
                {TEST_DATA_DIR.resolve("ipsum.txt.bgz.wrongextension"), false, true},
                {TEST_DATA_DIR.resolve("ipsum.txt.bgzipped_with_gzextension.gz"), true, true},
                {TEST_DATA_DIR.resolve("ipsum.txt.bgzipped_with_gzextension.gz"), false, true},
                {TEST_DATA_DIR.resolve("example.bam"), true, false},
                {TEST_DATA_DIR.resolve("example.bam"), false, true}
        };
    }

    @Test(dataProvider = "blockCompressedFiles")
    public void testIsBlockCompressed(Path file, boolean checkExtension, boolean expected) throws IOException {
        Assert.assertEquals(IOUtil.isBlockCompressed(file, checkExtension), expected);
    }

    @Test(dataProvider = "blockCompressedFiles")
    public void testIsBlockCompressedOnJimfs(Path file, boolean checkExtension, boolean expected) throws IOException {
         try (FileSystem jimfs = Jimfs.newFileSystem(Configuration.unix())) {
             final Path jimfsRoot = jimfs.getRootDirectories().iterator().next();
             final Path jimfsFile = Files.copy(file, jimfsRoot.resolve(file.getFileName().toString()));
             Assert.assertEquals(IOUtil.isBlockCompressed(jimfsFile, checkExtension), expected);
         }
    }

    @DataProvider
    public static Object[][] filesToCompress() {
        return new Object[][]{
                {WORDS_LONG, ".gz", 8},
                {WORDS_LONG, ".bfq", 8},
                {TEST_VARIANT_DIR.resolve("test1.vcf"), ".gz", 7},
                {TEST_VARIANT_DIR.resolve("test1.vcf"), ".bfq", 7}
        };
    }

    @Test(dataProvider = "filesToCompress")
    public void testCompressionLevel(final Path file, final String extension, final int lastDifference) throws IOException {
        final long origSize = Files.size(file);
        long previousSize = origSize;
        for (int compressionLevel = 1; compressionLevel <= 9; compressionLevel++) {
            final Path outFile = Files.createTempFile("tmp", extension);
            outFile.toFile().deleteOnExit();
            IOUtil.setCompressionLevel(compressionLevel);
            Assert.assertEquals(IOUtil.getCompressionLevel(), compressionLevel);
            final InputStream inStream = IOUtil.openFileForReading(file);
            try (final OutputStream outStream = IOUtil.openFileForWriting(outFile.toFile())) {
                IOUtil.transferByStream(inStream, outStream, origSize);
            }
            final long newSize = Files.size(outFile);
            if (compressionLevel <= lastDifference) {
                Assert.assertTrue(previousSize > newSize);
            } else {
                Assert.assertTrue(previousSize >= newSize);
            }
            previousSize = newSize;
        }
    }

    @DataProvider
    public Object[][] getExtensions(){
        return new Object[][]{
                {".gz", true},
                {".bfq", true},
                {".txt", false}};
    }

    @Test(dataProvider = "getExtensions")
    public void testReadWriteJimfs(String extension, boolean gzipped) throws IOException {
        final Path jmfsRoot = inMemoryFileSystem.getRootDirectories().iterator().next();
        final Path tmp = Files.createTempFile(jmfsRoot, "test", extension);
        final String expected = "lorem ipswitch, nantucket, bucket";
        try (Writer out = IOUtil.openFileForBufferedWriting(tmp)){
            out.write(expected);
        }

        try (InputStream in = new BufferedInputStream(Files.newInputStream(tmp))){
               Assert.assertEquals(IOUtil.isGZIPInputStream(in), gzipped);
        }

        try (BufferedReader in = IOUtil.openFileForBufferedReading(tmp)){
            final String actual = in.readLine();
            Assert.assertEquals(actual, expected);
        }
    }

    @DataProvider
    public static Object[][] badCompressionLevels() {
        return new Object[][]{
                {-1},
                {10}
        };
    }

    @Test(dataProvider = "badCompressionLevels", expectedExceptions = {IllegalArgumentException.class})
    public void testCompressionLevelExceptions(final int compressionLevel) {
        IOUtil.setCompressionLevel(compressionLevel);
    }

    @DataProvider
    public static Object[][] filesToCopy() {
        return new Object[][]{
                {TEST_VARIANT_DIR.resolve("test1.vcf")},
                {TEST_DATA_DIR.resolve("ipsum.txt")}
        };
    }

    @Test(dataProvider = "filesToCopy")
    public void testCopyFile(final Path file) throws IOException {
        final Path outFile = Files.createTempFile("tmp", ".tmp");
        outFile.toFile().deleteOnExit();
        IOUtil.copyFile(file.toFile(), outFile.toFile());
        Assert.assertEquals(Files.lines(file).collect(Collectors.toList()), Files.lines(outFile).collect(Collectors.toList()));
    }

    @Test(dataProvider = "filesToCopy", expectedExceptions = {SAMException.class})
    public void testCopyFileReadException(final Path file) throws IOException {
        final Path outFile = Files.createTempFile("tmp", ".tmp");
        outFile.toFile().deleteOnExit();
        file.toFile().setReadable(false);
        try {
            IOUtil.copyFile(file.toFile(), outFile.toFile());
        } finally { //need to set input file permission back to readable so other unit tests can access it
            file.toFile().setReadable(true);
        }
    }

    @Test(dataProvider = "filesToCopy", expectedExceptions = {SAMException.class})
    public void testCopyFileWriteException(final Path file) throws IOException {
        final Path outFile = Files.createTempFile("tmp", ".tmp");
        outFile.toFile().deleteOnExit();
        outFile.toFile().setWritable(false);
        IOUtil.copyFile(file.toFile(), outFile.toFile());
    }

    @DataProvider
    public static Object[][] baseNameTests() {
        return new Object[][]{
                {TEST_DATA_DIR.resolve("ipsum.txt"), "ipsum"},
                {TEST_DATA_DIR.resolve("ipsum.txt.bgz.wrongextension"), "ipsum.txt.bgz"},
                {TEST_DATA_DIR.resolve("ipsum.txt.bgzipped_with_gzextension.gz"), "ipsum.txt.bgzipped_with_gzextension"},
                {TEST_VARIANT_DIR.resolve("utils"), "utils"},
                {TEST_VARIANT_DIR.resolve("not_real_file.txt"), "not_real_file"}
        };
    }

    @Test(dataProvider = "baseNameTests")
    public void testBasename(final Path file, final String expected) {
        final String result = IOUtil.basename(file.toFile());
        Assert.assertEquals(result, expected);
    }

    @DataProvider
    public static Object[][] regExpTests() {
        return new Object[][]{
                {"\\w+\\.txt", new String[]{"5newline5.txt", "empty.txt", "ipsum.txt", "slurptest.txt"}},
                {"^((?!txt).)*$", new String[]{"Level1.fofn", "Level2.fofn", "example.bam"}},
                {"^\\d+.*", new String[]{"5newline5.txt"}}
        };
    }

    @Test(dataProvider = "regExpTests")
    public void testRegExp(final String regexp, final String[] expected) throws IOException {
        final String[] allNames = {"5newline5.txt", "Level2.fofn", "example.bam", "ipsum.txt.bgz", "ipsum.txt.bgzipped_with_gzextension.gz", "slurptest.txt", "Level1.fofn", "empty.txt", "ipsum.txt", "ipsum.txt.bgz.wrongextension", "ipsum.txt.gz"};
        final Path regExpDir = Files.createTempDirectory("regExpDir");
        regExpDir.toFile().deleteOnExit();
        final List<String> listExpected = Arrays.asList(expected);
        final List<File> expectedFiles = new ArrayList<File>();
        for (String name : allNames) {
            final Path file = regExpDir.resolve(name);
            file.toFile().deleteOnExit();
            file.toFile().createNewFile();
            if (listExpected.contains(name)) {
                expectedFiles.add(file.toFile());
            }
        }
        final File[] result = IOUtil.getFilesMatchingRegexp(regExpDir.toFile(), regexp);
        Assert.assertEqualsNoOrder(result, expectedFiles.toArray());
    }

    @Test()
    public void testReadLines() throws IOException {
        final Path file = Files.createTempFile("tmp", ".txt");
        file.toFile().deleteOnExit();
        final int seed = 12394738;
        final Random rand = new Random(seed);
        final int nLines = 5;
        final List<String> lines = new ArrayList<String>();
        try (final PrintWriter writer = new PrintWriter(Files.newBufferedWriter(file))) {
            for (int i = 0; i < nLines; i++) {
                final String line = TEST_STRING + Integer.toString(rand.nextInt(100000000));
                lines.add(line);
                writer.println(line);
            }
        }
        final List<String> retLines = new ArrayList<String>();
        IOUtil.readLines(file.toFile()).forEachRemaining(retLines::add);
        Assert.assertEquals(retLines, lines);
    }

    @DataProvider
    public static Object[][] fileSuffixTests() {
        return new Object[][]{
                {TEST_DATA_DIR.resolve("ipsum.txt"), ".txt"},
                {TEST_DATA_DIR.resolve("ipsum.txt.bgz"), ".bgz"},
                {TEST_DATA_DIR, null}
        };
    }

    @Test(dataProvider = "fileSuffixTests")
    public void testSuffixTest(final Path file, final String expected) {
        final String ret = IOUtil.fileSuffix(file.toFile());
        Assert.assertEquals(ret, expected);
    }

    @Test
    public void testCopyDirectoryTree() throws IOException {
        final Path copyToDir = Files.createTempDirectory("copyToDir");
        copyToDir.toFile().deleteOnExit();
        IOUtil.copyDirectoryTree(TEST_VARIANT_DIR.toFile(), copyToDir.toFile());
        final List<Path> collect = Files.walk(TEST_VARIANT_DIR).filter(f -> !f.equals(TEST_VARIANT_DIR)).map(p -> p.getFileName()).collect(Collectors.toList());
        final List<Path> collectCopy = Files.walk(copyToDir).filter(f -> !f.equals(copyToDir)).map(p -> p.getFileName()).collect(Collectors.toList());
        Assert.assertEqualsNoOrder(collect.toArray(), collectCopy.toArray());
    }

    // Little utility to gzip a byte array
    private static byte[] gzipMessage(final byte[] message) throws IOException {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        GZIPOutputStream gzout = new GZIPOutputStream(bos);
        gzout.write(message);
        gzout.finish();
        gzout.close();
        return bos.toByteArray();
    }

    @DataProvider
    public static Object[][] gzipTests() throws IOException {
        final byte[] emptyMessage = "".getBytes();
        final byte[] message = "Hello World".getBytes();

        // Compressed version of the messages
        final byte[] gzippedMessage = gzipMessage(message);
        final byte[] gzippedEmptyMessage = gzipMessage(emptyMessage);

        return new Object[][]{
                {emptyMessage, false},
                {message, false},
                {gzippedMessage, true},
                {gzippedEmptyMessage, true}
        };
    }

    @Test(dataProvider = "gzipTests")
    public void isGZIPInputStreamTest(byte[] data, boolean isGzipped) throws IOException {
        try(ByteArrayInputStream inputStream = new ByteArrayInputStream(data)) {
            // test string without compression
            Assert.assertEquals(IOUtil.isGZIPInputStream(inputStream), isGzipped);
            // call twice to verify 'in.reset()' was called
            Assert.assertEquals(IOUtil.isGZIPInputStream(inputStream), isGzipped);
        }
    }
}

