refine test cases

Signed-off-by: Heng Qian <[email protected]>
opensearch-project · Aug 9, 2024 · 460189e · 460189e
1 parent 48cf382
commit 460189e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/integ-test/src/test/java/org/opensearch/sql/legacy/ExplainIT.java b/integ-test/src/test/java/org/opensearch/sql/legacy/ExplainIT.java
@@ -283,7 +283,10 @@ public void explainAlias() throws IOException {
 
     String result =
         explainQuery(
-            String.format("SELECT (age + 1) AS agePlusOne FROM %s LIMIT 10", TEST_INDEX_ACCOUNT));
+            String.format(
+                "SELECT city, (age + 1) AS agePlusOne, (balance + 1000) as newBalance FROM %s LIMIT"
+                    + " 10",
+                TEST_INDEX_ACCOUNT));
     Assert.assertThat(
         result.replaceAll("\\s+", ""), equalTo(expectedOutput.replaceAll("\\s+", "")));
   }

diff --git a/integ-test/src/test/resources/expectedOutput/alias_explain.json b/integ-test/src/test/resources/expectedOutput/alias_explain.json
@@ -2,13 +2,13 @@
     "root": {
         "name": "ProjectOperator",
         "description": {
-            "fields": "[(age+1) AS agePlusOne]"
+            "fields": "[city, (age+1) AS agePlusOne, (balance+1000) AS newBalance]"
         },
         "children": [
             {
                 "name": "OpenSearchIndexScan",
                 "description": {
-                    "request": "OpenSearchQueryRequest(indexName=opensearch-sql_test_index_account,sourceBuilder={\"from\":0,\"size\":10,\"timeout\":\"1m\",\"_source\":{\"includes\":[\"age\"],\"excludes\":[]}},searchDone=false)"
+                    "request": "OpenSearchQueryRequest(indexName=opensearch-sql_test_index_account,sourceBuilder={\"from\":0,\"size\":10,\"timeout\":\"1m\",\"_source\":{\"includes\":[\"age\",\"city\",\"balance\"],\"excludes\":[]}},searchDone=false)"
                 },
                 "children": []
             }