Add new table format test for ParquetTypeGenerator

snowflakedb · Jul 27, 2024 · 89dd17c · 89dd17c
1 parent 3325c1d
commit 89dd17c
Show file tree

Hide file tree

Showing 2 changed files with 670 additions and 9 deletions.
diff --git a/src/test/java/net/snowflake/ingest/streaming/internal/ColumnMetadataBuilder.java b/src/test/java/net/snowflake/ingest/streaming/internal/ColumnMetadataBuilder.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2022 Snowflake Computing Inc. All rights reserved.
+ * Copyright (c) 2022-2024 Snowflake Computing Inc. All rights reserved.
  */
 
 package net.snowflake.ingest.streaming.internal;
@@ -16,6 +16,7 @@ public class ColumnMetadataBuilder {
   private Integer length;
   private boolean nullable;
   private String collation;
+  private String sourceIcebergDataType;
 
   private Integer ordinal;
 
@@ -144,6 +145,17 @@ public ColumnMetadataBuilder collation(String collation) {
     return this;
   }
 
+  /**
+   * Set column source Iceberg data type
+   *
+   * @param sourceIcebergDataType source Iceberg data type string
+   * @return columnMetadataBuilder object
+   */
+  public ColumnMetadataBuilder sourceIcebergDataType(String sourceIcebergDataType) {
+    this.sourceIcebergDataType = sourceIcebergDataType;
+    return this;
+  }
+
   /**
    * Set column ordinal
    *
@@ -172,6 +184,7 @@ public ColumnMetadata build() {
     colMetadata.setScale(scale);
     colMetadata.setPrecision(precision);
     colMetadata.setCollation(collation);
+    colMetadata.setSourceIcebergDataType(sourceIcebergDataType);
     colMetadata.setOrdinal(ordinal);
     return colMetadata;
   }