JingsongLi commented on a change in pull request #1393:
URL: https://github.com/apache/iceberg/pull/1393#discussion_r479869458
##########
File path:
flink/src/test/java/org/apache/iceberg/flink/TestFlinkCatalogTable.java
##########
@@ -88,4 +105,172 @@ public void testRenameTable() {
Collections.singletonList(TableColumn.of("id", DataTypes.BIGINT())),
tEnv.from("tl2").getSchema().getTableColumns());
}
+
+ @Test
+ public void testCreateTable() throws TableNotExistException {
+ tEnv.executeSql("CREATE TABLE tl(id BIGINT)");
+
+ Table table = table("tl");
+ Assert.assertEquals(
+ new Schema(Types.NestedField.optional(1, "id",
Types.LongType.get())).asStruct(),
+ table.schema().asStruct());
+ Assert.assertEquals(Maps.newHashMap(), table.properties());
+
+ CatalogTable catalogTable = catalogTable("tl");
+ Assert.assertEquals(TableSchema.builder().field("id",
DataTypes.BIGINT()).build(), catalogTable.getSchema());
+ Assert.assertEquals(Maps.newHashMap(), catalogTable.getOptions());
+ }
+
+ @Test
+ public void testCreateTableLocation() {
+ Assume.assumeFalse("HadoopCatalog does not support creating table with
location", isHadoopCatalog);
+
+ tEnv.executeSql("CREATE TABLE tl(id BIGINT) WITH
('location'='/tmp/location')");
+
+ Table table = table("tl");
+ Assert.assertEquals(
+ new Schema(Types.NestedField.optional(1, "id",
Types.LongType.get())).asStruct(),
+ table.schema().asStruct());
+ Assert.assertEquals("/tmp/location", table.location());
+ Assert.assertEquals(Maps.newHashMap(), table.properties());
+ }
+
+ @Test
+ public void testCreatePartitionTable() throws TableNotExistException {
+ tEnv.executeSql("CREATE TABLE tl(id BIGINT, dt STRING) PARTITIONED
BY(dt)");
+
+ Table table = table("tl");
+ Assert.assertEquals(
+ new Schema(
+ Types.NestedField.optional(1, "id", Types.LongType.get()),
+ Types.NestedField.optional(2, "dt",
Types.StringType.get())).asStruct(),
+ table.schema().asStruct());
+
Assert.assertEquals(PartitionSpec.builderFor(table.schema()).identity("dt").build(),
table.spec());
+ Assert.assertEquals(Maps.newHashMap(), table.properties());
+
+ CatalogTable catalogTable = catalogTable("tl");
+ Assert.assertEquals(
+ TableSchema.builder().field("id", DataTypes.BIGINT()).field("dt",
DataTypes.STRING()).build(),
+ catalogTable.getSchema());
+ Assert.assertEquals(Maps.newHashMap(), catalogTable.getOptions());
+ Assert.assertEquals(Collections.singletonList("dt"),
catalogTable.getPartitionKeys());
+ }
+
+ @Test
+ public void testLoadTransformPartitionTable() throws TableNotExistException {
+ Schema schema = new Schema(Types.NestedField.optional(0, "id",
Types.LongType.get()));
+ validationCatalog.createTable(
+ TableIdentifier.of(icebergNamespace, "tl"), schema,
+ PartitionSpec.builderFor(schema).bucket("id", 100).build());
+
+ CatalogTable catalogTable = catalogTable("tl");
+ Assert.assertEquals(
+ TableSchema.builder().field("id", DataTypes.BIGINT()).build(),
+ catalogTable.getSchema());
+ Assert.assertEquals(Maps.newHashMap(), catalogTable.getOptions());
+ Assert.assertEquals(Collections.emptyList(),
catalogTable.getPartitionKeys());
+ }
+
+ @Test
+ public void testAlterTable() throws TableNotExistException {
+ tEnv.executeSql("CREATE TABLE tl(id BIGINT) WITH ('oldK'='oldV')");
+ Map<String, String> properties = Maps.newHashMap();
+ properties.put("oldK", "oldV");
+
+ // new
+ tEnv.executeSql("ALTER TABLE tl SET('newK'='newV')");
+ properties.put("newK", "newV");
+ Assert.assertEquals(properties, table("tl").properties());
+
+ // update old
+ tEnv.executeSql("ALTER TABLE tl SET('oldK'='oldV2')");
+ properties.put("oldK", "oldV2");
+ Assert.assertEquals(properties, table("tl").properties());
+
+ // remove property
+ CatalogTable catalogTable = catalogTable("tl");
+ properties.remove("oldK");
+ tEnv.getCatalog(tEnv.getCurrentCatalog()).get().alterTable(
+ new ObjectPath(DATABASE, "tl"), catalogTable.copy(properties), false);
+ Assert.assertEquals(properties, table("tl").properties());
+ }
+
+ @Test
+ public void testRelocateTable() {
+ Assume.assumeFalse("HadoopCatalog does not support relocate table",
isHadoopCatalog);
+
+ tEnv.executeSql("CREATE TABLE tl(id BIGINT)");
+ tEnv.executeSql("ALTER TABLE tl SET('location'='/tmp/location')");
Review comment:
The Hive dialect should only works for Hive tables. There are lots of
tricky things..
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
