Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* 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 io.trino.tests.product.hive;
import io.trino.tempto.ProductTest;
import io.trino.testng.services.Flaky;
import org.testng.annotations.Test;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Optional;
import static io.trino.plugin.hive.HiveTableProperties.TRANSACTIONAL;
import static io.trino.tempto.assertions.QueryAssert.Row.row;
import static io.trino.tests.product.TestGroups.HIVE_TRANSACTIONAL;
import static io.trino.tests.product.TestGroups.PROFILE_SPECIFIC_TESTS;
import static io.trino.tests.product.TestGroups.STORAGE_FORMATS;
import static io.trino.tests.product.utils.HadoopTestUtils.RETRYABLE_FAILURES_ISSUES;
import static io.trino.tests.product.utils.HadoopTestUtils.RETRYABLE_FAILURES_MATCH;
import static io.trino.tests.product.utils.QueryExecutors.onHive;
import static io.trino.tests.product.utils.QueryExecutors.onTrino;
import static java.util.Objects.requireNonNull;
import static org.assertj.core.api.Assertions.assertThat;
public class TestHiveCreateTable
extends ProductTest
{
@Test(groups = STORAGE_FORMATS)
@Flaky(issue = RETRYABLE_FAILURES_ISSUES, match = RETRYABLE_FAILURES_MATCH)
public void testCreateTable()
throws SQLException
{
onTrino().executeQuery("CREATE TABLE test_create_table(a bigint, b varchar, c smallint) WITH (format='ORC')");
onTrino().executeQuery("INSERT INTO test_create_table(a, b, c) VALUES " +
"(NULL, NULL, NULL), " +
"(-42, 'abc', SMALLINT '-127'), " +
"(9223372036854775807, 'abcdefghijklmnopqrstuvwxyz', SMALLINT '32767')");
assertThat(onTrino().executeQuery("SELECT * FROM test_create_table"))
.containsOnly(
row(null, null, null),
row(-42, "abc", -127),
row(9223372036854775807L, "abcdefghijklmnopqrstuvwxyz", 32767));
assertThat(getTableProperty("test_create_table", TRANSACTIONAL))
// Hive 3 removes "transactional" table property when it has value "false"
.isIn(Optional.empty(), Optional.of("false"));
onTrino().executeQuery("DROP TABLE test_create_table");
}
@Test(groups = STORAGE_FORMATS)
@Flaky(issue = RETRYABLE_FAILURES_ISSUES, match = RETRYABLE_FAILURES_MATCH)
public void testCreateTableAsSelect()
throws SQLException
{
onTrino().executeQuery("" +
"CREATE TABLE test_create_table_as_select WITH (format='ORC') AS " +
"SELECT * FROM (VALUES " +
" (NULL, NULL, NULL), " +
" (-42, 'abc', SMALLINT '-127'), " +
" (9223372036854775807, 'abcdefghijklmnopqrstuvwxyz', SMALLINT '32767')" +
") t(a, b, c)");
assertThat(onTrino().executeQuery("SELECT * FROM test_create_table_as_select"))
.containsOnly(
row(null, null, null),
row(-42, "abc", -127),
row(9223372036854775807L, "abcdefghijklmnopqrstuvwxyz", 32767));
assertThat(getTableProperty("test_create_table_as_select", TRANSACTIONAL))
// Hive 3 removes "transactional" table property when it has value "false"
.isIn(Optional.empty(), Optional.of("false"));
onTrino().executeQuery("DROP TABLE test_create_table_as_select");
}
@Test(groups = {HIVE_TRANSACTIONAL, PROFILE_SPECIFIC_TESTS})
public void testVerifyEnvironmentHiveTransactionalByDefault()
throws SQLException
{
onHive().executeQuery("CREATE TABLE test_hive_transactional_by_default(a bigint) STORED AS ORC");
assertThat(getTableProperty("test_hive_transactional_by_default", TRANSACTIONAL))
.contains("true");
onHive().executeQuery("DROP TABLE test_hive_transactional_by_default");
}
private static Optional getTableProperty(String tableName, String propertyName)
throws SQLException
{
requireNonNull(tableName, "tableName is null");
requireNonNull(propertyName, "propertyName is null");
try (Statement statement = onHive().getConnection().createStatement();
ResultSet resultSet = statement.executeQuery("SHOW TBLPROPERTIES " + tableName)) {
while (resultSet.next()) {
if (propertyName.equals(resultSet.getString("prpt_name"))) {
// We need to distinguish between a property that is not set and a property that has NULL value.
return Optional.of(resultSet.getString("prpt_value"));
}
}
}
return Optional.empty();
}
}
