Added unit test

Signed-off-by: Guian Gumpac <[email protected]>
Bit-Quill · Mar 18, 2023 · 66ae9aa · 66ae9aa
1 parent 88b7d6b
commit 66ae9aa
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/core/src/test/java/org/opensearch/sql/planner/physical/PhysicalPlanTest.java b/core/src/test/java/org/opensearch/sql/planner/physical/PhysicalPlanTest.java
@@ -5,7 +5,9 @@
 
 package org.opensearch.sql.planner.physical;
 
+import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
 
 import java.util.List;
 import org.junit.jupiter.api.Test;
@@ -50,4 +52,12 @@ void addSplitToChildByDefault() {
     testPlan.add(split);
     verify(child).add(split);
   }
+
+  @Test
+  void testGetRawResponse() {
+    when(child.getRawResponse()).thenReturn("not empty", "", null);
+    assertEquals("not empty", testPlan.getRawResponse());
+    assertEquals("", testPlan.getRawResponse());
+    assertEquals("", testPlan.getRawResponse());
+  }
 }