From 48eae4822b34552ca22662051be67d02eaeb2c7b Mon Sep 17 00:00:00 2001
From: jdnvn <joe@donovan.to>
Date: Sun, 1 Sep 2024 11:58:51 -0400
Subject: [PATCH] add assertion that inner query builder was visited
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Konrad Gołuchowski <konrad.goluchowski@synerise.com>
---
 .../query/functionscore/FunctionScoreQueryBuilderTests.java      | 1 +
 1 file changed, 1 insertion(+)

diff --git a/server/src/test/java/org/opensearch/index/query/functionscore/FunctionScoreQueryBuilderTests.java b/server/src/test/java/org/opensearch/index/query/functionscore/FunctionScoreQueryBuilderTests.java
index 78f4163d96d33..8cf7941941bcb 100644
--- a/server/src/test/java/org/opensearch/index/query/functionscore/FunctionScoreQueryBuilderTests.java
+++ b/server/src/test/java/org/opensearch/index/query/functionscore/FunctionScoreQueryBuilderTests.java
@@ -948,5 +948,6 @@ public void testVisit() {
         builder.visit(createTestVisitor(visitedQueries));
 
         assertEquals(2, visitedQueries.size());
+        assertTrue(visitedQueries.contains(termQueryBuilder));
     }
 }