hadoop-gremlin/src/test/java/org/apache/tinkerpop/gremlin/hadoop/process/computer/AbstractHadoopGraphComputerTest.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.
 */

package org.apache.tinkerpop.gremlin.hadoop.process.computer;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.fs.Path;
import org.apache.tinkerpop.gremlin.util.Gremlin;
import org.junit.Test;

import java.io.File;
import java.util.stream.Stream;

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

/**
 * @author Marko A. Rodriguez (http://markorodriguez.com)
 */
public class AbstractHadoopGraphComputerTest {

    @Test
    public void shouldCopyDirectoriesCorrectly() throws Exception {
        final String hdfsName = this.getClass().getSimpleName() + "-hdfs";
        final String localName = this.getClass().getSimpleName() + "-local";
        final FileSystem fs = FileSystem.get(new Configuration());
        if (!new File(System.getProperty("java.io.tmpdir") + "/" + localName).exists())
            assertTrue(new File(System.getProperty("java.io.tmpdir") + "/" + localName).mkdir());
        File tempFile1 = new File(System.getProperty("java.io.tmpdir") + "/" + localName + "/test1.txt");
        File tempFile2 = new File(System.getProperty("java.io.tmpdir") + "/" + localName + "/test2.txt");
        assertTrue(tempFile1.createNewFile());
        assertTrue(tempFile2.createNewFile());
        assertTrue(tempFile1.exists());
        assertTrue(tempFile2.exists());
        if (fs.exists(new Path("target/" + hdfsName)))
            assertTrue(fs.delete(new Path("target/" + hdfsName), true));
        fs.copyFromLocalFile(true, new Path(tempFile1.getAbsolutePath()), new Path("target/" + hdfsName + "/test1.dat"));
        fs.copyFromLocalFile(true, new Path(tempFile2.getAbsolutePath()), new Path("target/" + hdfsName + "/test2.dat"));
        assertTrue(fs.exists(new Path("target/" + hdfsName + "/test1.dat")));
        assertTrue(fs.exists(new Path("target/" + hdfsName + "/test2.dat")));
        assertTrue(fs.exists(new Path("target/" + hdfsName)));
        assertTrue(fs.isDirectory(new Path("target/" + hdfsName)));
        assertFalse(tempFile1.exists());
        assertFalse(tempFile2.exists());
        assertTrue(new File(System.getProperty("java.io.tmpdir") + "/" + localName).exists());
        assertTrue(new File(System.getProperty("java.io.tmpdir") + "/" + localName).delete());
        assertTrue(fs.exists(new Path("target/" + hdfsName + "/test1.dat")));
        assertTrue(fs.exists(new Path("target/" + hdfsName + "/test2.dat")));
        assertTrue(fs.exists(new Path("target/" + hdfsName)));
        assertTrue(fs.isDirectory(new Path("target/" + hdfsName)));
        /////
        final String hadoopGremlinLibsRemote = "hadoop-gremlin-" + Gremlin.version() + "-libs";
        final File localDirectory = new File(System.getProperty("java.io.tmpdir") + "/" + hadoopGremlinLibsRemote);
        final File localLibDirectory = new File(localDirectory.getAbsolutePath() + "/" + hdfsName);
        if (localLibDirectory.exists()) {
            Stream.of(localLibDirectory.listFiles()).forEach(File::delete);
            assertTrue(localLibDirectory.delete());
        }
        assertFalse(localLibDirectory.exists());
        assertEquals(localLibDirectory, AbstractHadoopGraphComputer.copyDirectoryIfNonExistent(fs, "target/" + hdfsName));
        assertTrue(localLibDirectory.exists());
        assertTrue(localLibDirectory.isDirectory());
        assertEquals(2, Stream.of(localLibDirectory.listFiles()).filter(file -> file.getName().endsWith(".dat")).count());
    }
}