move template query builder tests back to core (using mockscript)

This commit is contained in:
Ryan Ernst 2015-12-08 20:26:45 -08:00
parent 2e299860e5
commit 907d3cf878
2 changed files with 11 additions and 5 deletions

View File

@ -40,7 +40,6 @@ import java.util.HashMap;
import java.util.Map;
import java.util.Random;
@ESTestCase.AwaitsFix(bugUrl = "nopush")
public class TemplateQueryBuilderTests extends AbstractQueryTestCase<TemplateQueryBuilder> {
/**
@ -52,7 +51,7 @@ public class TemplateQueryBuilderTests extends AbstractQueryTestCase<TemplateQue
public static void setupClass() {
templateBase = createQuery(getRandom());
}
static QueryBuilder<?> createQuery(Random r) {
switch (RandomInts.randomIntBetween(r, 0, 2)) {
case 0:
@ -73,7 +72,7 @@ public class TemplateQueryBuilderTests extends AbstractQueryTestCase<TemplateQue
@Override
protected TemplateQueryBuilder doCreateTestQueryBuilder() {
return new TemplateQueryBuilder(new Template(templateBase.toString()));
return new TemplateQueryBuilder(new Template(templateBase.toString(), ScriptType.INLINE, "mockscript", null, null));
}
@Override

View File

@ -21,6 +21,8 @@ package org.elasticsearch.script;
import org.apache.lucene.index.LeafReaderContext;
import org.elasticsearch.common.Nullable;
import org.elasticsearch.common.bytes.BytesArray;
import org.elasticsearch.common.bytes.BytesReference;
import org.elasticsearch.plugins.Plugin;
import org.elasticsearch.search.lookup.SearchLookup;
@ -71,12 +73,17 @@ public class MockScriptEngine implements ScriptEngineService {
@Override
public Object compile(String script) {
return Integer.parseInt(script);
return script;
}
@Override
public ExecutableScript executable(CompiledScript compiledScript, @Nullable Map<String, Object> vars) {
return null;
return new AbstractExecutableScript() {
@Override
public Object run() {
return new BytesArray((String)compiledScript.compiled());
}
};
}
@Override