io.trino.tests.product.hive.TestHiveTransactionalTableInsert Maven / Gradle / Ivy
/*
* 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 org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
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.utils.QueryExecutors.onHive;
import static io.trino.tests.product.utils.QueryExecutors.onTrino;
import static java.util.Locale.ENGLISH;
import static java.util.stream.Collectors.joining;
import static org.assertj.core.api.Assertions.assertThat;
public class TestHiveTransactionalTableInsert
extends HiveProductTest
{
@Test(dataProvider = "transactionalTableType", groups = {HIVE_TRANSACTIONAL, PROFILE_SPECIFIC_TESTS})
public void testInsertIntoTransactionalTable(TransactionalTableType type)
{
String tableName = "test_insert_into_transactional_table_" + type.name().toLowerCase(ENGLISH);
onHive().executeQuery("" +
"CREATE TABLE " + tableName + "(a bigint)" +
"CLUSTERED BY(a) INTO 4 BUCKETS STORED AS ORC " +
hiveTableProperties(type));
try {
onTrino().executeQuery("INSERT INTO " + tableName + " (a) VALUES (42)");
assertThat(onTrino().executeQuery("SELECT * FROM " + tableName))
.containsOnly(row(42));
}
finally {
onHive().executeQuery("DROP TABLE " + tableName);
}
}
@DataProvider
public Object[][] transactionalTableType()
{
return new Object[][] {
{TransactionalTableType.ACID},
{TransactionalTableType.INSERT_ONLY},
};
}
private static String hiveTableProperties(TransactionalTableType transactionalTableType)
{
return transactionalTableType.getHiveTableProperties().stream()
.collect(joining(",", "TBLPROPERTIES (", ")"));
}
}