Generate Painless Factory for Creating Script Instances (#25120)

1 files changed, 7 insertions, 2 deletions

diff --git a/test/framework/src/main/java/org/elasticsearch/script/MockScriptEngine.java b/test/framework/src/main/java/org/elasticsearch/script/MockScriptEngine.java
index 1a8a9c8d7f..2ca6f2aa0c 100644
--- a/test/framework/src/main/java/org/elasticsearch/script/MockScriptEngine.java
+++ b/test/framework/src/main/java/org/elasticsearch/script/MockScriptEngine.java
@@ -86,8 +86,13 @@ public class MockScriptEngine implements ScriptEngine {
                 // TODO: need a better way to implement all these new contexts
                 // this is just a shim to act as an executable script just as before
                 ExecutableScript execScript = mockCompiled.createExecutableScript(vars);
-                return () -> (String) execScript.run();
-            };
+                    return new TemplateScript(vars) {
+                        @Override
+                        public String execute() {
+                            return (String) execScript.run();
+                        }
+                    };
+                };
             return context.factoryClazz.cast(factory);
         }
         throw new IllegalArgumentException("mock script engine does not know how to handle context [" + context.name + "]");