Update for test coverage

Signed-off-by: currantw <[email protected]>
Bit-Quill · Feb 3, 2025 · bf49974 · bf49974
1 parent af9f294
commit bf49974
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/core/src/test/java/org/opensearch/sql/planner/physical/FlattenOperatorTest.java b/core/src/test/java/org/opensearch/sql/planner/physical/FlattenOperatorTest.java
@@ -85,6 +85,12 @@ void testStructMissing() {
     assertEquals(rowValue, actual);
   }
 
+  @Test
+  void testStructUnknown() {
+    ExprValue actual = execute(flatten(inputPlan, DSL.ref("struct_unknown", STRUCT))).getFirst();
+    assertEquals(rowValue, actual);
+  }
+
   @Test
   void testStruct() {
     ExprValue actual = execute(flatten(inputPlan, DSL.ref("struct", STRUCT))).getFirst();
@@ -140,14 +146,9 @@ void testAncestorStructMissing() {
   }
 
   @Test
-  void testPathMissing() {
-    ExprValue actual = execute(flatten(inputPlan, DSL.ref("struct.unknown", STRUCT))).getFirst();
-    assertEquals(rowValue, actual);
-  }
-
-  @Test
-  void testAncestorPathMissing() {
-    ExprValue actual = execute(flatten(inputPlan, DSL.ref("unknown", STRUCT))).getFirst();
+  void testAncestorStructUnknown() {
+    ExprValue actual =
+        execute(flatten(inputPlan, DSL.ref("struct_unknown.path", STRUCT))).getFirst();
     assertEquals(rowValue, actual);
   }