org.apache.hive.hplsql.TestHplsqlOffline.java Source code

Introduction

Here is the source code for org.apache.hive.hplsql.TestHplsqlOffline.java

Source

/**
 * 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.hive.hplsql;

import java.io.BufferedReader;
import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import java.io.StringReader;
import org.apache.commons.io.FileUtils;
import org.junit.Assert;
import org.junit.Test;

/**
 * Unit tests for HPL/SQL that generate SQL but do not run them (Hive connection no required)
 */
public class TestHplsqlOffline {

    private final ByteArrayOutputStream out = new ByteArrayOutputStream();

    @Test
    public void testCreateTableDb2() throws Exception {
        run("create_table_db2");
    }

    @Test
    public void testCreateTableMssql() throws Exception {
        run("create_table_mssql");
    }

    @Test
    public void testCreateTableMssql2() throws Exception {
        run("create_table_mssql2");
    }

    @Test
    public void testCreateTableMysql() throws Exception {
        run("create_table_mysql");
    }

    @Test
    public void testCreateTableOra() throws Exception {
        run("create_table_ora");
    }

    @Test
    public void testCreateTableOra2() throws Exception {
        run("create_table_ora2");
    }

    @Test
    public void testCreateTablePg() throws Exception {
        run("create_table_pg");
    }

    @Test
    public void testCreateTableTd() throws Exception {
        run("create_table_td");
    }

    @Test
    public void testDeleteAll() throws Exception {
        run("delete_all");
    }

    @Test
    public void testInsertMysql() throws Exception {
        run("insert_mysql");
    }

    @Test
    public void testSelect() throws Exception {
        run("select");
    }

    @Test
    public void testSelectDb2() throws Exception {
        run("select_db2");
    }

    @Test
    public void testSelectTeradata() throws Exception {
        run("select_teradata");
    }

    @Test
    public void testUpdate() throws Exception {
        run("update");
    }

    /**
     * Run a test file
     */
    void run(String testFile) throws Exception {
        System.setOut(new PrintStream(out));
        Exec exec = new Exec();
        String[] args = { "-f", "src/test/queries/offline/" + testFile + ".sql", "-trace", "-offline" };
        exec.run(args);
        String s = getTestOutput(out.toString()).trim();
        FileUtils.writeStringToFile(new java.io.File("target/tmp/log/" + testFile + ".out.txt"), s);
        String t = FileUtils
                .readFileToString(new java.io.File("src/test/results/offline/" + testFile + ".out.txt"), "utf-8")
                .trim();
        System.setOut(null);
        Assert.assertEquals(s, t);
    }

    /**
     * Get test output
     */
    String getTestOutput(String s) throws Exception {
        StringBuilder sb = new StringBuilder();
        BufferedReader reader = new BufferedReader(new StringReader(s));
        String line = null;
        while ((line = reader.readLine()) != null) {
            if (!line.startsWith("log4j:") && !line.contains("INFO Log4j")) {
                sb.append(line);
                sb.append("\n");
            }
        }
        return sb.toString();
    }
}