/*
* 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 com.facebook.presto.hive;
import com.facebook.presto.spi.type.Type;
import com.google.common.collect.ImmutableMap;
import org.testng.annotations.Test;
import java.util.HashMap;
import java.util.Map;
import static com.facebook.presto.hive.HiveTestUtils.TYPE_MANAGER;
import static com.facebook.presto.spi.type.TypeSignature.parseTypeSignature;
import static java.util.Objects.requireNonNull;
import static org.testng.Assert.assertEquals;
public class TestHiveTypeTranslator
{
private final TypeTranslator typeTranslator;
private final Map<String, HiveType> typeTranslationMap;
public TestHiveTypeTranslator()
{
this(new HiveTypeTranslator(), ImmutableMap.of());
}
protected TestHiveTypeTranslator(TypeTranslator typeTranslator, Map<String, HiveType> overwriteTranslation)
{
this.typeTranslator = requireNonNull(typeTranslator, "typeTranslator is null");
ImmutableMap<String, HiveType> hiveTypeTranslationMap = ImmutableMap.<String, HiveType>builder()
.put("bigint", HiveType.HIVE_LONG)
.put("integer", HiveType.HIVE_INT)
.put("smallint", HiveType.HIVE_SHORT)
.put("tinyint", HiveType.HIVE_BYTE)
.put("double", HiveType.HIVE_DOUBLE)
.put("varchar(3)", HiveType.valueOf("varchar(3)"))
.put("varchar", HiveType.HIVE_STRING)
.put("date", HiveType.HIVE_DATE)
.put("timestamp", HiveType.HIVE_TIMESTAMP)
.put("decimal(5,3)", HiveType.valueOf("decimal(5,3)"))
.put("varbinary", HiveType.HIVE_BINARY)
.put("array(timestamp)", HiveType.valueOf("array<timestamp>"))
.put("map(boolean,varbinary)", HiveType.valueOf("map<boolean,binary>"))
.put("row(col0 integer,col1 varbinary)", HiveType.valueOf("struct<col0:int,col1:binary>"))
.build();
typeTranslationMap = new HashMap<>();
typeTranslationMap.putAll(hiveTypeTranslationMap);
typeTranslationMap.putAll(overwriteTranslation);
}
@Test
public void testTypeTranslator()
{
for (Map.Entry<String, HiveType> entry : typeTranslationMap.entrySet()) {
assertTypeTranslation(entry.getKey(), entry.getValue());
}
}
private void assertTypeTranslation(String typeName, HiveType hiveType)
{
Type type = TYPE_MANAGER.getType(parseTypeSignature(typeName));
assertEquals(HiveType.toHiveType(typeTranslator, type), hiveType);
}
}