com.splicemachine.derby.impl.sql.execute.operations.export.ExportFileTest.java Source code

Introduction

Here is the source code for com.splicemachine.derby.impl.sql.execute.operations.export.ExportFileTest.java

Source

/*
 * Copyright 2012 - 2016 Splice Machine, Inc.
 *
 * 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.splicemachine.derby.impl.sql.execute.operations.export;

import com.splicemachine.access.api.DistributedFileSystem;
import com.splicemachine.si.impl.TestingFileSystem;
import com.splicemachine.si.testenv.ArchitectureIndependent;
import com.splicemachine.si.testenv.SITestDataEnv;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.RandomStringUtils;
import org.junit.*;
import org.junit.experimental.categories.Category;
import org.junit.rules.TemporaryFolder;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.file.FileSystem;
import java.nio.file.FileSystems;

import static org.junit.Assert.*;

@Category(ArchitectureIndependent.class)
public class ExportFileTest {

    @Rule
    public TemporaryFolder temporaryFolder = new TemporaryFolder();

    @BeforeClass
    public static void setupConfig() {
        // necessary for mapr
        //        SpliceConstants.config.set("fs.default.name", "file:///");
    }

    private DistributedFileSystem dfs = new TestingFileSystem(FileSystems.getDefault().provider());

    @Test
    public void getOutputStream_createsStreamConnectedToExpectedFile() throws IOException {

        // given
        ExportParams exportParams = ExportParams.withDirectory(temporaryFolder.getRoot().getAbsolutePath());
        ExportFile exportFile = new ExportFile(exportParams, testTaskId(), dfs);
        final byte[] EXPECTED_CONTENT = ("splice for the win" + RandomStringUtils.randomAlphanumeric(1000))
                .getBytes("utf-8");

        // when
        OutputStream outputStream = exportFile.getOutputStream();
        outputStream.write(EXPECTED_CONTENT);
        outputStream.close();

        // then
        File expectedFile = new File(temporaryFolder.getRoot(), "export_82010203042A060708.csv");
        assertTrue(expectedFile.exists());
        Assert.assertArrayEquals(EXPECTED_CONTENT, IOUtils.toByteArray(new FileInputStream(expectedFile)));
    }

    @Test
    public void buildFilenameFromTaskId() throws IOException {
        ExportFile streamSetup = new ExportFile(new ExportParams(), testTaskId(), dfs);
        byte[] taskId = testTaskId();
        assertEquals("export_82010203042A060708.csv", streamSetup.buildFilenameFromTaskId(taskId));
    }

    @Test
    public void delete() throws IOException {
        // given
        ExportParams exportParams = ExportParams.withDirectory(temporaryFolder.getRoot().getAbsolutePath());
        ExportFile exportFile = new ExportFile(exportParams, testTaskId(), dfs);

        exportFile.getOutputStream();
        assertTrue("export file should exist in temp dir", temporaryFolder.getRoot().list().length > 0);

        // when
        exportFile.delete();

        // then
        assertTrue("export file should be deleted", temporaryFolder.getRoot().list().length == 0);
    }

    @Test
    public void createDirectory() throws IOException {
        String testDir = temporaryFolder.getRoot().getAbsolutePath() + "/" + RandomStringUtils.randomAlphabetic(9);
        ExportParams exportParams = ExportParams.withDirectory(testDir);
        ExportFile exportFile = new ExportFile(exportParams, testTaskId(), dfs);

        assertTrue(exportFile.createDirectory());

        assertTrue(new File(testDir).exists());
        assertTrue(new File(testDir).isDirectory());
    }

    @Test
    public void createDirectory_returnsFalseWhenCannotCreate() throws IOException {
        String testDir = "/noPermissionToCreateFolderInRoot";
        ExportParams exportParams = ExportParams.withDirectory(testDir);
        ExportFile exportFile = new ExportFile(exportParams, testTaskId(), dfs);

        assertFalse(exportFile.createDirectory());

        assertFalse(new File(testDir).exists());
        assertFalse(new File(testDir).isDirectory());
    }

    // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

    private static byte[] testTaskId() {
        return new byte[] { -126, 1, 2, 3, 4, 42, 6, 7, 8 };
    }

}