fix coverage

Signed-off-by: Lantao Jin <[email protected]>
opensearch-project · Nov 2, 2024 · 42a5f33 · 42a5f33
1 parent faa6291
commit 42a5f33
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/core/src/test/java/org/opensearch/sql/planner/physical/collector/RoundingTest.java b/core/src/test/java/org/opensearch/sql/planner/physical/collector/RoundingTest.java
@@ -57,6 +57,10 @@ void datetime_rounding_non_core_type_span() {
         DSL.span(DSL.ref("timestamp", new MockTimestampExprType()), DSL.literal(1), "h");
     rounding = Rounding.createRounding(timestampSpan);
     assertInstanceOf(Rounding.TimestampRounding.class, rounding);
+    SpanExpression datetimeSpan =
+        DSL.span(DSL.ref("datetime", new MockDateTimeExprType()), DSL.literal(1), "h");
+    rounding = Rounding.createRounding(datetimeSpan);
+    assertInstanceOf(Rounding.DatetimeRounding.class, rounding);
   }
 
   @Test
@@ -95,4 +99,11 @@ public String typeName() {
       return "TIMESTAMP";
     }
   }
+
+  static class MockDateTimeExprType implements ExprType {
+    @Override
+    public String typeName() {
+      return "DATETIME";
+    }
+  }
 }