Fix Iceberg DateTimeIT (#872)

fix
snowflakedb · Oct 24, 2024 · fba8ca7 · fba8ca7
1 parent 1fe05a5
commit fba8ca7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/test/java/net/snowflake/ingest/streaming/internal/datatypes/AbstractDataTypeTest.java b/src/test/java/net/snowflake/ingest/streaming/internal/datatypes/AbstractDataTypeTest.java
@@ -14,6 +14,7 @@
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
 import java.sql.SQLException;
+import java.sql.Timestamp;
 import java.time.ZoneId;
 import java.util.HashMap;
 import java.util.Map;
@@ -563,6 +564,8 @@ protected void testIcebergIngestAndQuery(
             .usingComparatorForType(BigDecimal::compareTo, BigDecimal.class)
             .usingRecursiveComparison()
             .isEqualTo(expectedValue);
+      } else if (expectedValue instanceof Timestamp) {
+        Assertions.assertThat(res.toString()).isEqualTo(expectedValue.toString());
       } else {
         Assertions.assertThat(res).isEqualTo(expectedValue);
       }