Fix IT

opendistro-for-elasticsearch · Dec 15, 2020 · 72d2a94 · 72d2a94
1 parent 9be9000
commit 72d2a94
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/integ-test/src/test/java/com/amazon/opendistroforelasticsearch/sql/legacy/AggregationIT.java b/integ-test/src/test/java/com/amazon/opendistroforelasticsearch/sql/legacy/AggregationIT.java
@@ -508,13 +508,15 @@ public void orderByAliasDescTest() throws IOException {
 
   @Test
   public void orderByGroupFieldWithAlias() throws IOException {
+    Assume.assumeTrue(isNewQueryEngineEabled());
+
     // ORDER BY field name
     JSONObject response = executeJdbcRequest(String.format("SELECT gender as g, COUNT(*) as count "
         + "FROM %s GROUP BY gender ORDER BY gender", TEST_INDEX_ACCOUNT));
 
     verifySchema(response,
-        schema("g", "g", "text"),
-        schema("count", "count", "integer"));
+        schema("gender", "g", "text"),
+        schema("COUNT(*)", "count", "integer"));
     verifyDataRowsInOrder(response,
         rows("f", 493),
         rows("m", 507));
@@ -524,8 +526,8 @@ public void orderByGroupFieldWithAlias() throws IOException {
         + "FROM %s GROUP BY gender ORDER BY g", TEST_INDEX_ACCOUNT));
 
     verifySchema(response,
-        schema("g", "g", "text"),
-        schema("count", "count", "integer"));
+        schema("gender", "g", "text"),
+        schema("COUNT(*)", "count", "integer"));
     verifyDataRowsInOrder(response,
         rows("f", 493),
         rows("m", 507));