add tests

databendlabs · Dec 2, 2024 · f0caf8c · f0caf8c
1 parent 68ed12b
commit f0caf8c
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/databend-client/src/test/java/com/databend/client/data/TestColumnTypeHandlerFactory.java b/databend-client/src/test/java/com/databend/client/data/TestColumnTypeHandlerFactory.java
@@ -31,14 +31,14 @@ public void testGetTypeFunction() {
         assertTypeHandler("Nullable(Float64)", Float64Handler.class, true);
         assertTypeHandler("String", StringHandler.class, false);
         assertTypeHandler("Nullable(String)", StringHandler.class, true);
-        assertTypeHandler("Date", StringHandler.class, false);
-        assertTypeHandler("Nullable(Date)", StringHandler.class, true);
-        assertTypeHandler("DateTime", StringHandler.class, false);
-        assertTypeHandler("Nullable(DateTime)", StringHandler.class, true);
-        assertTypeHandler("DateTime64", StringHandler.class, false);
-        assertTypeHandler("Nullable(DateTime64)", StringHandler.class, true);
-        assertTypeHandler("Timestamp", StringHandler.class, false);
-        assertTypeHandler("Nullable(Timestamp)", StringHandler.class, true);
+        assertTypeHandler("Date", DateHandler.class, false);
+        assertTypeHandler("Nullable(Date)", DateHandler.class, true);
+        assertTypeHandler("DateTime", TimestampHandler.class, false);
+        assertTypeHandler("Nullable(DateTime)", TimestampHandler.class, true);
+        assertTypeHandler("DateTime64", TimestampHandler.class, false);
+        assertTypeHandler("Nullable(DateTime64)", TimestampHandler.class, true);
+        assertTypeHandler("Timestamp", TimestampHandler.class, false);
+        assertTypeHandler("Nullable(Timestamp)", TimestampHandler.class, true);
         assertTypeHandler("Array(String)", StringHandler.class, false);
         assertTypeHandler("Nullable(Array(Int32))", StringHandler.class, true);
         assertTypeHandler("Struct", StringHandler.class, false);