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 index 054b3f5458..8f59b25f70 100644 --- 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 index 3bcbb2eaaf..5665bae667 100644 --- 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": [] }