Add missed update to AstBuilderTest for sort option

Signed-off-by: James Duong <[email protected]>
Bit-Quill · Oct 31, 2024 · 49fc894 · 49fc894
1 parent b8dff3a
commit 49fc894
Showing 1 changed file with 48 additions and 1 deletion.
diff --git a/ppl/src/test/java/org/opensearch/sql/ppl/parser/AstBuilderTest.java b/ppl/src/test/java/org/opensearch/sql/ppl/parser/AstBuilderTest.java
@@ -47,6 +47,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import java.util.Arrays;
+import java.util.Optional;
 import org.junit.Ignore;
 import org.junit.Rule;
 import org.junit.Test;
@@ -670,15 +671,61 @@ public void testTrendline() {
             + " test_field_alias_2",
         trendline(
             relation("t"),
+            Optional.empty(),
             computation(5, field("test_field"), "test_field_alias", SMA),
             computation(1, field("test_field_2"), "test_field_alias_2", SMA)));
   }
 
+  @Test
+  public void testTrendlineSort() {
+    assertEqual(
+        "source=t | trendline sort test_field sma(5, test_field)",
+        trendline(
+            relation("t"),
+            Optional.of(
+                field(
+                    "test_field",
+                    argument("asc", booleanLiteral(true)),
+                    argument("type", nullLiteral()))),
+            computation(5, field("test_field"), "test_field_trendline", SMA)));
+  }
+
+  @Test
+  public void testTrendlineSortDesc() {
+    assertEqual(
+        "source=t | trendline sort - test_field sma(5, test_field)",
+        trendline(
+            relation("t"),
+            Optional.of(
+                field(
+                    "test_field",
+                    argument("asc", booleanLiteral(false)),
+                    argument("type", nullLiteral()))),
+            computation(5, field("test_field"), "test_field_trendline", SMA)));
+  }
+
+  @Test
+  public void testTrendlineSortAsc() {
+    assertEqual(
+        "source=t | trendline sort + test_field sma(5, test_field)",
+        trendline(
+            relation("t"),
+            Optional.of(
+                field(
+                    "test_field",
+                    argument("asc", booleanLiteral(true)),
+                    argument("type", nullLiteral()))),
+            computation(5, field("test_field"), "test_field_trendline", SMA)));
+  }
+
   @Test
   public void testTrendlineNoAlias() {
     assertEqual(
         "source=t | trendline sma(5, test_field)",
-        trendline(relation("t"), computation(5, field("test_field"), "test_field_trendline", SMA)));
+        trendline(
+            relation("t"),
+            Optional.empty(),
+            computation(5, field("test_field"), "test_field_trendline", SMA)));
   }
 
   @Test