Fix failing release tests (Require capability for DECAY function and SET and INLINESTATS commands) (#134358)

Three PRs have just been merged that cause failures in `test-release`,
and a third PR changes ES|QL railroad diagrams for another recently
released change to `KNN` (also a snapshot-only feature).

* #132729 - `DecayTests` has over 300 failures
* #134029 - `ParsingTests` had six new failures
* #124725 - Failures in `VerifierTests` and `OptimizerVerificationTests`

This PR attempts to fix all in one, because if we do this in separate
PRs they will fail due to the failures cause by the other PRs.

Author: craigtaverner

docs/reference/query-languages/esql/images/functions/decay.svg

@@ -1329,6 +1329,11 @@ public enum Cap {
          */
         FIX_MV_EXPAND_INCONSISTENT_COLUMN_ORDER,
 
+        /**
+         * Support for the SET command.
+         */
+        SET_COMMAND(Build.current().isSnapshot()),
+
         /**
          * (Re)Added EXPLAIN command
          */

docs/reference/query-languages/esql/images/functions/knn.svg

@@ -213,6 +213,7 @@ public void testInvalidSample() {
     }
 
     public void testSet() {
+        assumeTrue("SET command available in snapshot only", EsqlCapabilities.Cap.SET_COMMAND.isEnabled());
         EsqlStatement query = parse("SET foo = \"bar\"; row a = 1", new QueryParams());
         assertThat(query.plan(), is(instanceOf(Row.class)));
         assertThat(query.settings().size(), is(1));
@@ -232,6 +233,7 @@ public void testSet() {
     }
 
     public void testSetWithTripleQuotes() {
+        assumeTrue("SET command available in snapshot only", EsqlCapabilities.Cap.SET_COMMAND.isEnabled());
         EsqlStatement query = parse("SET foo = \"\"\"bar\"baz\"\"\"; row a = 1", new QueryParams());
         assertThat(query.plan(), is(instanceOf(Row.class)));
         assertThat(query.settings().size(), is(1));
@@ -249,6 +251,7 @@ public void testSetWithTripleQuotes() {
     }
 
     public void testMultipleSet() {
+        assumeTrue("SET command available in snapshot only", EsqlCapabilities.Cap.SET_COMMAND.isEnabled());
         EsqlStatement query = parse(
             "SET foo = \"bar\"; SET bar = 2; SET foo = \"baz\"; SET x = 3.5; SET y = false; SET z = null; row a = 1",
             new QueryParams()
@@ -265,6 +268,7 @@ public void testMultipleSet() {
     }
 
     public void testSetArrays() {
+        assumeTrue("SET command available in snapshot only", EsqlCapabilities.Cap.SET_COMMAND.isEnabled());
         EsqlStatement query = parse("SET foo = [\"bar\", \"baz\"]; SET bar = [1, 2, 3]; row a = 1", new QueryParams());
         assertThat(query.plan(), is(instanceOf(Row.class)));
         assertThat(query.settings().size(), is(2));
@@ -274,6 +278,7 @@ public void testSetArrays() {
     }
 
     public void testSetWithNamedParams() {
+        assumeTrue("SET command available in snapshot only", EsqlCapabilities.Cap.SET_COMMAND.isEnabled());
         EsqlStatement query = parse(
             "SET foo = \"bar\"; SET bar = ?a; SET foo = \"baz\"; SET x = ?x; row a = 1",
             new QueryParams(
@@ -293,6 +298,7 @@ public void testSetWithNamedParams() {
     }
 
     public void testSetWithPositionalParams() {
+        assumeTrue("SET command available in snapshot only", EsqlCapabilities.Cap.SET_COMMAND.isEnabled());
         EsqlStatement query = parse(
             "SET foo = \"bar\"; SET bar = ?; SET foo = \"baz\"; SET x = ?; row a = ?",
             new QueryParams(

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/action/EsqlCapabilities.java

@@ -2069,6 +2069,8 @@ public void testCategorizeOptionSimilarityThreshold() {
     }
 
     public void testCategorizeWithInlineStats() {
+        assumeTrue("CATEGORIZE must be enabled", EsqlCapabilities.Cap.CATEGORIZE_V6.isEnabled());
+        assumeTrue("INLINESTATS must be enabled", EsqlCapabilities.Cap.INLINESTATS_V11.isEnabled());
         assertEquals(
             "1:37: CATEGORIZE [CATEGORIZE(last_name, { \"similarity_threshold\": 1 })] is not yet supported with "
                 + "INLINESTATS [INLINESTATS COUNT(*) BY CATEGORIZE(last_name, { \"similarity_threshold\": 1 })]",

x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/analysis/ParsingTests.java

@@ -13,13 +13,15 @@
 import org.elasticsearch.common.geo.GeoPoint;
 import org.elasticsearch.common.unit.DistanceUnit;
 import org.elasticsearch.script.ScoreScriptUtils;
+import org.elasticsearch.xpack.esql.action.EsqlCapabilities;
 import org.elasticsearch.xpack.esql.core.expression.Expression;
 import org.elasticsearch.xpack.esql.core.expression.Literal;
 import org.elasticsearch.xpack.esql.core.expression.MapExpression;
 import org.elasticsearch.xpack.esql.core.tree.Source;
 import org.elasticsearch.xpack.esql.core.type.DataType;
 import org.elasticsearch.xpack.esql.expression.function.AbstractScalarFunctionTestCase;
 import org.elasticsearch.xpack.esql.expression.function.TestCaseSupplier;
+import org.junit.BeforeClass;
 
 import java.time.Duration;
 import java.time.Instant;
@@ -44,6 +46,11 @@ public DecayTests(@Name("TestCase") Supplier<TestCaseSupplier.TestCase> testCase
         this.testCase = testCaseSupplier.get();
     }
 
+    @BeforeClass
+    public static void checkCapability() {
+        assumeTrue("Decay function tests require the DECAY_FUNCTION capability", EsqlCapabilities.Cap.DECAY_FUNCTION.isEnabled());
+    }
+
     @ParametersFactory
     public static Iterable<Object[]> parameters() {
         List<TestCaseSupplier> testCaseSuppliers = new ArrayList<>();

x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/analysis/VerifierTests.java

@@ -431,6 +431,7 @@ public void testRemoteEnrichAfterLookupJoinWithPipelineBreaker() {
     }
 
     public void testDanglingOrderByInInlineStats() {
+        assumeTrue("INLINESTATS must be enabled", EsqlCapabilities.Cap.INLINESTATS_V11.isEnabled());
         var analyzer = AnalyzerTestUtils.analyzer(loadMapping("mapping-default.json", "test"));
 
         var err = error("""

x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/expression/function/scalar/score/DecayTests.java

@@ -225,7 +225,7 @@ setup:
   - gt: {esql.functions.to_long: $functions_to_long}
   - match: {esql.functions.coalesce: $functions_coalesce}
   - gt: {esql.functions.categorize: $functions_categorize}
-  - length: {esql.functions: 142} # check the "sister" test above for a likely update to the same esql.functions length check
+  - length: {esql.functions: 144} # check the "sister" test above for a likely update to the same esql.functions length check
 
 ---
 took: