From 104fa7daa215126227636e2e978b1ce312bea4ed Mon Sep 17 00:00:00 2001
From: Lin Liu <141371752+linliu-code@users.noreply.github.com>
Date: Thu, 1 Feb 2024 18:18:41 -0800
Subject: [PATCH] [HUDI-6902] Fix a test about timestamp format (#10606)

---
 .../apache/hudi/hadoop/TestHoodieParquetInputFormat.java  | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/hudi-hadoop-mr/src/test/java/org/apache/hudi/hadoop/TestHoodieParquetInputFormat.java b/hudi-hadoop-mr/src/test/java/org/apache/hudi/hadoop/TestHoodieParquetInputFormat.java
index 37ec5cef24f5..f824753b6bbb 100644
--- a/hudi-hadoop-mr/src/test/java/org/apache/hudi/hadoop/TestHoodieParquetInputFormat.java
+++ b/hudi-hadoop-mr/src/test/java/org/apache/hudi/hadoop/TestHoodieParquetInputFormat.java
@@ -66,12 +66,14 @@
 import java.io.IOException;
 import java.nio.file.Paths;
 import java.sql.Timestamp;
+import java.text.SimpleDateFormat;
 import java.time.Instant;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.ZoneOffset;
 import java.util.ArrayList;
 import java.util.Collections;
+import java.util.Date;
 import java.util.List;
 
 import static org.apache.hudi.common.testutils.SchemaTestUtil.getSchemaFromResource;
@@ -813,7 +815,11 @@ public void testHoodieParquetInputFormatReadTimeType() throws IOException {
               Instant.ofEpochMilli(testTimestampLong), ZoneOffset.UTC);
           assertEquals(Timestamp.valueOf(localDateTime).toString(), String.valueOf(writable.get()[0]));
         } else {
-          assertEquals(new Timestamp(testTimestampLong).toString(), String.valueOf(writable.get()[0]));
+          Date date = new Date();
+          date.setTime(testTimestampLong);
+          assertEquals(
+              new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS").format(date),
+              String.valueOf(writable.get()[0]));
         }
         // test long
         assertEquals(testTimestampLong * 1000, ((LongWritable) writable.get()[1]).get());