Skip to content

Commit 46255ca

Browse files
authored
Fix ScriptedSimilarityTests by allowing for BulkScoreBulkScorer calling TermScorer::nextDocsAndScores (#128561)
1 parent be627e9 commit 46255ca

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

server/src/test/java/org/elasticsearch/index/similarity/ScriptedSimilarityTests.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,6 @@ public double execute(
152152
dir.close();
153153
}
154154

155-
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/128502")
156155
public void testInitScript() throws IOException {
157156
final AtomicBoolean initCalled = new AtomicBoolean();
158157
SimilarityWeightScript.Factory weightScriptFactory = () -> {
@@ -189,7 +188,8 @@ public double execute(
189188

190189
StackTraceElement[] stackTraceElements = Thread.currentThread().getStackTrace();
191190
if (Arrays.stream(stackTraceElements).anyMatch(ste -> {
192-
return ste.getClassName().endsWith(".TermScorer") && ste.getMethodName().equals("score");
191+
return ste.getClassName().endsWith(".TermScorer")
192+
&& (ste.getMethodName().equals("score") || ste.getMethodName().equals("nextDocsAndScores"));
193193
}) == false) {
194194
// this might happen when computing max scores
195195
return Float.MAX_VALUE;

0 commit comments

Comments
 (0)