embulk-output-oracle 0.7.10 → 0.7.11

data/src/test/java/org/embulk/output/oracle/DummyFileSystem.java

@@ -0,0 +1,96 @@
+package org.embulk.output.oracle;
+
+import java.io.IOException;
+import java.nio.file.FileStore;
+import java.nio.file.FileSystem;
+import java.nio.file.Path;
+import java.nio.file.PathMatcher;
+import java.nio.file.WatchService;
+import java.nio.file.attribute.UserPrincipalLookupService;
+import java.nio.file.spi.FileSystemProvider;
+import java.util.Set;
+
+public class DummyFileSystem extends FileSystem
+{
+    private final FileSystem original;
+    private final String fileContents;
+
+    public DummyFileSystem(FileSystem original, String fileContents)
+    {
+        this.original = original;
+        this.fileContents = fileContents;
+    }
+
+    @Override
+    public FileSystemProvider provider()
+    {
+        return new DummyFileSystemProvider(original.provider(), fileContents);
+    }
+
+    @Override
+    public void close() throws IOException
+    {
+        original.close();
+    }
+
+    @Override
+    public boolean isOpen()
+    {
+        return original.isOpen();
+    }
+
+    @Override
+    public boolean isReadOnly()
+    {
+        return original.isReadOnly();
+    }
+
+    @Override
+    public String getSeparator()
+    {
+        return original.getSeparator();
+    }
+
+    @Override
+    public Iterable<Path> getRootDirectories()
+    {
+        return original.getRootDirectories();
+    }
+
+    @Override
+    public Iterable<FileStore> getFileStores()
+    {
+        return original.getFileStores();
+    }
+
+    @Override
+    public Set<String> supportedFileAttributeViews()
+    {
+        return original.supportedFileAttributeViews();
+    }
+
+    @Override
+    public Path getPath(String first, String... more)
+    {
+        return original.getPath(first, more);
+    }
+
+    @Override
+    public PathMatcher getPathMatcher(String syntaxAndPattern)
+    {
+        return original.getPathMatcher(syntaxAndPattern);
+    }
+
+    @Override
+    public UserPrincipalLookupService getUserPrincipalLookupService()
+    {
+        return original.getUserPrincipalLookupService();
+    }
+
+    @Override
+    public WatchService newWatchService() throws IOException
+    {
+        return original.newWatchService();
+    }
+
+}

data/src/test/java/org/embulk/output/oracle/DummyFileSystemProvider.java

@@ -0,0 +1,154 @@
+package org.embulk.output.oracle;
+
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URI;
+import java.nio.channels.SeekableByteChannel;
+import java.nio.file.AccessMode;
+import java.nio.file.CopyOption;
+import java.nio.file.DirectoryStream;
+import java.nio.file.DirectoryStream.Filter;
+import java.nio.file.FileStore;
+import java.nio.file.FileSystem;
+import java.nio.file.LinkOption;
+import java.nio.file.OpenOption;
+import java.nio.file.Path;
+import java.nio.file.attribute.BasicFileAttributes;
+import java.nio.file.attribute.FileAttribute;
+import java.nio.file.attribute.FileAttributeView;
+import java.nio.file.spi.FileSystemProvider;
+import java.util.Map;
+import java.util.Set;
+
+public class DummyFileSystemProvider extends FileSystemProvider
+{
+    private final FileSystemProvider original;
+    private final String fileContents;
+
+    public DummyFileSystemProvider(FileSystemProvider original, String fileContents)
+    {
+        this.original = original;
+        this.fileContents = fileContents;
+    }
+
+    @Override
+    public String getScheme()
+    {
+        return original.getScheme();
+    }
+
+    @Override
+    public FileSystem newFileSystem(URI uri, Map<String, ?> env) throws IOException
+    {
+        return original.newFileSystem(uri, env);
+    }
+
+    @Override
+    public FileSystem getFileSystem(URI uri)
+    {
+        return original.getFileSystem(uri);
+    }
+
+    @Override
+    public Path getPath(URI uri)
+    {
+        return original.getPath(uri);
+    }
+
+    @Override
+    public SeekableByteChannel newByteChannel(Path path, Set<? extends OpenOption> options, FileAttribute<?>... attrs)
+            throws IOException
+    {
+        return original.newByteChannel(path, options, attrs);
+    }
+
+    @Override
+    public DirectoryStream<Path> newDirectoryStream(Path dir, Filter<? super Path> filter)
+            throws IOException
+    {
+        return original.newDirectoryStream(dir, filter);
+    }
+
+    @Override
+    public void createDirectory(Path dir, FileAttribute<?>... attrs) throws IOException
+    {
+        original.createDirectory(dir, attrs);
+    }
+
+    @Override
+    public void delete(Path path) throws IOException
+    {
+        original.delete(path);
+    }
+
+    @Override
+    public void copy(Path source, Path target, CopyOption... options) throws IOException
+    {
+        original.copy(source, target, options);
+    }
+
+    @Override
+    public void move(Path source, Path target, CopyOption... options) throws IOException
+    {
+        original.move(source, target, options);
+    }
+
+    @Override
+    public boolean isSameFile(Path path, Path path2) throws IOException
+    {
+        return original.isSameFile(path, path2);
+    }
+
+    @Override
+    public boolean isHidden(Path path) throws IOException
+    {
+        return original.isHidden(path);
+    }
+
+    @Override
+    public FileStore getFileStore(Path path) throws IOException
+    {
+        return original.getFileStore(path);
+    }
+
+    @Override
+    public void checkAccess(Path path, AccessMode... modes) throws IOException
+    {
+        original.checkAccess(path, modes);
+    }
+
+    @Override
+    public <V extends FileAttributeView> V getFileAttributeView(Path path, Class<V> type, LinkOption... options)
+    {
+        return original.getFileAttributeView(path, type, options);
+    }
+
+    @Override
+    public <A extends BasicFileAttributes> A readAttributes(Path path, Class<A> type, LinkOption... options)
+            throws IOException
+    {
+        return original.readAttributes(path, type, options);
+    }
+
+    @Override
+    public Map<String, Object> readAttributes(Path path, String attributes, LinkOption... options)
+            throws IOException
+    {
+        return original.readAttributes(path, attributes, options);
+    }
+
+    @Override
+    public void setAttribute(Path path, String attribute, Object value, LinkOption... options)
+            throws IOException
+    {
+        original.setAttribute(path, attribute, value, options);
+    }
+
+    @Override
+    public InputStream newInputStream(Path path, OpenOption... options) throws IOException
+    {
+        return new ByteArrayInputStream(fileContents.getBytes());
+    }
+
+}

data/src/test/java/org/embulk/output/oracle/DummyPath.java

@@ -0,0 +1,188 @@
+package org.embulk.output.oracle;
+
+import java.io.File;
+import java.io.IOException;
+import java.net.URI;
+import java.nio.file.FileSystem;
+import java.nio.file.LinkOption;
+import java.nio.file.Path;
+import java.nio.file.WatchEvent.Kind;
+import java.nio.file.WatchEvent.Modifier;
+import java.nio.file.WatchKey;
+import java.nio.file.WatchService;
+import java.util.Iterator;
+
+public class DummyPath implements Path
+{
+    private final Path original;
+    private final String fileContents;
+
+    public DummyPath(Path original, String fileContents)
+    {
+        this.original = original;
+        this.fileContents = fileContents;
+    }
+
+    @Override
+    public int compareTo(Path other)
+    {
+        return original.compareTo(other);
+    }
+
+    @Override
+    public boolean endsWith(Path other)
+    {
+        return original.endsWith(other);
+    }
+
+    @Override
+    public boolean endsWith(String other)
+    {
+        return original.endsWith(other);
+    }
+
+    @Override
+    public Path getFileName()
+    {
+        return original.getFileName();
+    }
+
+    @Override
+    public FileSystem getFileSystem()
+    {
+        return new DummyFileSystem(original.getFileSystem(), fileContents);
+    }
+
+    @Override
+    public Path getName(int index)
+    {
+        return original.getName(index);
+    }
+
+    @Override
+    public int getNameCount()
+    {
+        return original.getNameCount();
+    }
+
+    @Override
+    public Path getParent()
+    {
+        return original.getParent();
+    }
+
+    @Override
+    public Path getRoot()
+    {
+        return original.getRoot();
+    }
+
+    @Override
+    public boolean isAbsolute()
+    {
+        return original.isAbsolute();
+    }
+
+    @Override
+    public Iterator<Path> iterator()
+    {
+        return original.iterator();
+    }
+
+    @Override
+    public Path normalize()
+    {
+        return original.normalize();
+    }
+
+    @Override
+    public WatchKey register(WatchService watcher, Kind<?>... events) throws IOException
+    {
+        return original.register(watcher, events);
+    }
+
+    @Override
+    public WatchKey register(WatchService watcher, Kind<?>[] events, Modifier... modifiers) throws IOException
+    {
+        return original.register(watcher, events, modifiers);
+    }
+
+    @Override
+    public Path relativize(Path other)
+    {
+        return original.relativize(other);
+    }
+
+    @Override
+    public Path resolve(Path other)
+    {
+        return original.resolve(other);
+    }
+
+    @Override
+    public Path resolve(String other)
+    {
+        return original.resolve(other);
+    }
+
+    @Override
+    public Path resolveSibling(Path other)
+    {
+        return original.resolveSibling(other);
+    }
+
+    @Override
+    public Path resolveSibling(String other)
+    {
+        return original.resolveSibling(other);
+    }
+
+    @Override
+    public boolean startsWith(Path other)
+    {
+        return original.startsWith(other);
+    }
+
+    @Override
+    public boolean startsWith(String other)
+    {
+        return original.startsWith(other);
+    }
+
+    @Override
+    public Path subpath(int beginIndex, int endIndex)
+    {
+        return original.subpath(beginIndex, endIndex);
+    }
+
+    @Override
+    public Path toAbsolutePath()
+    {
+        return original.toAbsolutePath();
+    }
+
+    @Override
+    public File toFile()
+    {
+        return original.toFile();
+    }
+
+    @Override
+    public Path toRealPath(LinkOption... options) throws IOException
+    {
+        return original.toRealPath(options);
+    }
+
+    @Override
+    public URI toUri()
+    {
+        return original.toUri();
+    }
+
+    @Override
+    public String toString()
+    {
+        return original.toString();
+    }
+
+}

data/src/test/java/org/embulk/output/oracle/OracleTests.java

@@ -0,0 +1,90 @@
+package org.embulk.output.oracle;
+
+import com.google.common.base.Throwables;
+import com.google.common.io.ByteStreams;
+
+import org.embulk.config.ConfigSource;
+import org.embulk.test.EmbulkTests;
+import org.embulk.test.TestingEmbulk;
+
+import java.io.IOException;
+import java.nio.charset.Charset;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.Arrays;
+
+import static java.util.Locale.ENGLISH;
+
+public class OracleTests
+{
+    public static ConfigSource baseConfig()
+    {
+        return EmbulkTests.config("EMBULK_OUTPUT_ORACLE_TEST_CONFIG");
+    }
+
+    public static void execute(TestingEmbulk embulk, String sql) throws IOException
+    {
+        try {
+            Class.forName("oracle.jdbc.OracleDriver");
+        } catch (ClassNotFoundException e) {
+            throw new RuntimeException("You should put 'ojdbc7.jar' in 'embulk-output-oracle/driver' directory in order to test.");
+        }
+
+        Path sqlFile = embulk.createTempFile("sql");
+        Files.write(sqlFile, Arrays.asList(sql), Charset.forName("UTF8"));
+
+        ConfigSource config = baseConfig();
+        String host = config.get(String.class, "host");
+        String port = config.get(String.class, "port", "1521");
+        String user = config.get(String.class, "user");
+        String password = config.get(String.class, "password");
+        String database = config.get(String.class, "database");
+
+        ProcessBuilder pb = new ProcessBuilder(
+                "sql",
+                user + "/" + password + "@" + host + ":" + port + "/" + database,
+                "@" + sqlFile.toFile().getAbsolutePath());
+        pb.environment().put("JAVA_TOOL_OPTIONS", "-Dfile.encoding=UTF8");
+        pb.redirectErrorStream(true);
+        int code;
+        try {
+            Process process = pb.start();
+            ByteStreams.copy(process.getInputStream(), System.out);
+            code = process.waitFor();
+        } catch (IOException | InterruptedException ex) {
+            throw Throwables.propagate(ex);
+        }
+        if (code != 0) {
+            throw new RuntimeException(String.format(ENGLISH,
+                        "Command finished with non-zero exit code. Exit code is %d.", code));
+        }
+    }
+
+    public static String selectRecords(TestingEmbulk embulk, String tableName) throws IOException
+    {
+        Path temp = embulk.createTempFile("txt");
+        Files.delete(temp);
+
+        StringBuilder sql = new StringBuilder();
+        sql.append("SPOOL '" + temp.toString() + "'");
+        sql.append(System.lineSeparator());
+        sql.append("SET FEEDBACK OFF");
+        sql.append(System.lineSeparator());
+        sql.append("SET HEADING OFF");
+        sql.append(System.lineSeparator());
+        sql.append("SET PAGESIZE 0");
+        sql.append(System.lineSeparator());
+        sql.append("SET LINESIZE 1000");
+        sql.append(System.lineSeparator());
+        sql.append("SET COLSEP ','");
+        sql.append(System.lineSeparator());
+        sql.append("SET TRIMSPOOL ON");
+        sql.append(System.lineSeparator());
+        sql.append("SELECT * FROM " + tableName + ";");
+        sql.append(System.lineSeparator());
+        sql.append("EXIT;");
+        execute(embulk, sql.toString());
+
+        return EmbulkTests.readSortedFile(temp);
+    }
+}