Skip to content

Commit 7664cd5

Browse files
committed
add UT
Signed-off-by: xinyual <xinyual@amazon.com>
1 parent d71e145 commit 7664cd5

File tree

2 files changed

+21
-0
lines changed

2 files changed

+21
-0
lines changed

core/src/main/java/org/opensearch/sql/ast/dsl/AstDSL.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@
4949
import org.opensearch.sql.ast.expression.WindowFunction;
5050
import org.opensearch.sql.ast.expression.Xor;
5151
import org.opensearch.sql.ast.tree.Aggregation;
52+
import org.opensearch.sql.ast.tree.AppendPipe;
5253
import org.opensearch.sql.ast.tree.Bin;
5354
import org.opensearch.sql.ast.tree.CountBin;
5455
import org.opensearch.sql.ast.tree.Dedupe;
@@ -550,6 +551,11 @@ public static Trendline trendline(
550551
return new Trendline(sortField, Arrays.asList(computations)).attach(input);
551552
}
552553

554+
public static AppendPipe appendPipe(UnresolvedPlan input, UnresolvedPlan subquery) {
555+
556+
return new AppendPipe(subquery).attach(input);
557+
}
558+
553559
public static Trendline.TrendlineComputation computation(
554560
Integer numDataPoints, Field dataField, String alias, Trendline.TrendlineType type) {
555561
return new Trendline.TrendlineComputation(numDataPoints, dataField, alias, type);

ppl/src/test/java/org/opensearch/sql/ppl/parser/AstBuilderTest.java

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
import static org.opensearch.sql.ast.dsl.AstDSL.agg;
1313
import static org.opensearch.sql.ast.dsl.AstDSL.aggregate;
1414
import static org.opensearch.sql.ast.dsl.AstDSL.alias;
15+
import static org.opensearch.sql.ast.dsl.AstDSL.appendPipe;
1516
import static org.opensearch.sql.ast.dsl.AstDSL.argument;
1617
import static org.opensearch.sql.ast.dsl.AstDSL.booleanLiteral;
1718
import static org.opensearch.sql.ast.dsl.AstDSL.compare;
@@ -870,6 +871,20 @@ public void testFillNullValueWithFields() {
870871
fillNull(relation("t"), intLiteral(0), true, field("a"), field("b"), field("c")));
871872
}
872873

874+
@Test
875+
public void testAppendPipe() {
876+
assertEqual(
877+
"source=t | appendpipe [ stats COUNT() ]",
878+
appendPipe(
879+
relation("t"),
880+
agg(
881+
null,
882+
exprList(alias("COUNT()", aggregate("count", AstDSL.allFields()))),
883+
emptyList(),
884+
emptyList(),
885+
defaultStatsArgs())));
886+
}
887+
873888
public void testTrendline() {
874889
assertEqual(
875890
"source=t | trendline sma(5, test_field) as test_field_alias sma(1, test_field_2) as"

0 commit comments

Comments
 (0)