From 52e861932b4c8e91f0a64672fcf4906aedf4f12f Mon Sep 17 00:00:00 2001
From: carlosdelest <carlos.delgado@elastic.co>
Date: Tue, 6 Aug 2024 15:26:11 +0200
Subject: [PATCH] Add test for multiple MATCH commands

---
 .../LocalPhysicalPlanOptimizerTests.java      | 33 +++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/optimizer/LocalPhysicalPlanOptimizerTests.java b/x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/optimizer/LocalPhysicalPlanOptimizerTests.java
index 315e4972a2280..a0385fc6b4001 100644
--- a/x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/optimizer/LocalPhysicalPlanOptimizerTests.java
+++ b/x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/optimizer/LocalPhysicalPlanOptimizerTests.java
@@ -432,6 +432,39 @@ public void testMatchCommandWithWhereClause() {
         assertThat(query.query().toString(), is(expected.toString()));
     }
 
+    /**
+     * TopNExec[[Order[emp_no{f}#3,ASC,LAST]],1000[INTEGER],0]
+     * \_ExchangeExec[[],false]
+     *   \_ProjectExec[[_meta_field{f}#9, emp_no{f}#3, first_name{f}#4, gender{f}#5, job{f}#10, job.raw{f}#11, languages{f}#6, last_n
+     * ame{f}#7, long_noidx{f}#12, salary{f}#8]]
+     *     \_FieldExtractExec[_meta_field{f}#9, emp_no{f}#3, first_name{f}#4, gen]
+     *       \_EsQueryExec[test],
+     *          query[{"bool":{"must":[{"query_string":{"query":"last_name: Smith","fields":[]}},
+     *          {"query_string":{"query":"John","fields":[]}}],"boost":1.0}}]
+     *          sort[[FieldSort[field=emp_no{f}#3, direction=ASC, nulls=LAST]]]
+     */
+    public void testMatchCommandWithMultipleMatches() {
+        var plan = plannerOptimizer.plan("""
+            from test
+            | match "last_name: Smith"
+            | sort emp_no
+            | MATCH "John"
+            """, IS_SV_STATS);
+
+        var limit = as(plan, TopNExec.class);
+        var exchange = as(limit.child(), ExchangeExec.class);
+        var project = as(exchange.child(), ProjectExec.class);
+        var field = as(project.child(), FieldExtractExec.class);
+        var query = as(field.child(), EsQueryExec.class);
+        assertThat(query.limit().fold(), is(1000));
+
+        Source source = new Source(2, 8, "emp_no > 10010");
+        var queryString1 = QueryBuilders.queryStringQuery("last_name: Smith");
+        var queryString2 = QueryBuilders.queryStringQuery("John");
+        var expected = QueryBuilders.boolQuery().must(queryString1).must(queryString2);
+        assertThat(query.query().toString(), is(expected.toString()));
+    }
+
     // optimizer doesn't know yet how to break down different multi count
     public void testCountFieldsAndAllWithFilter() {
         var plan = plannerOptimizer.plan("""