chore: add some unit test for TableSchela

GreptimeTeam · Feb 2, 2024 · ced4c5d · ced4c5d
1 parent d17a218
commit ced4c5d
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 5 deletions.
diff --git a/ingester-protocol/src/main/java/io/greptime/models/TableSchema.java b/ingester-protocol/src/main/java/io/greptime/models/TableSchema.java
@@ -94,7 +94,8 @@ public Builder addTag(String name, DataType dataType) {
          * @return this builder
          */
         public Builder addTimestamp(String name, DataType dataType) {
-            Ensures.ensure(dataType.isTimestamp(), "Invalid timestamp data type");
+            Ensures.ensure(dataType.isTimestamp(),
+                    "Invalid timestamp data type: %s, only support `DataType.TimestampXXX`", dataType);
             return addColumn(name, SemanticType.Timestamp, dataType);
         }
 
@@ -139,7 +140,8 @@ public Builder addColumn(String name, //
             Ensures.ensureNonNull(dataType, "Null data type");
 
             if (semanticType == SemanticType.Timestamp) {
-                Ensures.ensure(dataType.isTimestamp(), "Invalid timestamp data type");
+                Ensures.ensure(dataType.isTimestamp(),
+                        "Invalid timestamp data type: %s, only support `DataType.TimestampXXX`", dataType);
             }
 
             this.columnNames.add(name);

diff --git a/...st/java/io/greptime/models/TableTest.java → ...a/io/greptime/models/TableSchemaTest.java b/...st/java/io/greptime/models/TableTest.java → ...a/io/greptime/models/TableSchemaTest.java
@@ -24,10 +24,10 @@
 /**
  * @author jiachun.fjc
  */
-public class TableTest {
+public class TableSchemaTest {
 
     @Test
-    public void testTableNonNull() {
+    public void testNonNull() {
         TableSchema schema = TableSchema.newBuilder("test_table") //
                 .addTag("col1", DataType.String) //
                 .addTag("col2", DataType.String) //
@@ -54,7 +54,7 @@ public void testTableNonNull() {
     }
 
     @Test
-    public void testTableSomeNull() {
+    public void testSomeNull() {
         TableSchema schema = TableSchema.newBuilder("test_table") //
                 .addTag("col1", DataType.String) //
                 .addTag("col2", DataType.String) //
@@ -73,4 +73,35 @@ public void testTableSomeNull() {
         Assert.assertFalse(rawRows.getRows(2).getValues(2).hasI32Value());
         Assert.assertFalse(rawRows.getRows(1).getValues(1).hasStringValue());
     }
+
+
+    @Test
+    public void testNotSupportTimestamp() {
+        TableSchema.Builder builder = TableSchema.newBuilder("test_table") //
+                .addTag("col1", DataType.String) //
+                .addTag("col2", DataType.String) //
+                .addField("col3", DataType.Int32);
+
+        try {
+            builder.addTimestamp("col4", DataType.Int32);
+            Assert.fail();
+        } catch (Exception e) {
+            Assert.assertTrue(e.getMessage().contains("Invalid timestamp data type"));
+        }
+    }
+
+    @Test
+    public void testNotSupportDecimalExtension() {
+        TableSchema.Builder builder = TableSchema.newBuilder("test_table") //
+                .addTag("col1", DataType.String) //
+                .addTag("col2", DataType.String) //
+                .addField("col3", DataType.Int32);
+
+        try {
+            builder.addColumn("col4", SemanticType.Field, DataType.Float64, new DataType.DecimalTypeExtension(39, 9));
+            Assert.fail();
+        } catch (Exception e) {
+            Assert.assertTrue(e.getMessage().contains("Only decimal type can have decimal type extension"));
+        }
+    }
 }