fix ut

Signed-off-by: bowenlan-amzn <[email protected]>
opensearch-project · Jan 7, 2025 · e152a01 · e152a01
1 parent 941a5c2
commit e152a01
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/server/src/test/java/org/opensearch/index/mapper/NumberFieldTypeTests.java b/server/src/test/java/org/opensearch/index/mapper/NumberFieldTypeTests.java
@@ -89,6 +89,7 @@
 import java.util.List;
 import java.util.function.Supplier;
 
+import org.opensearch.search.query.BitmapIndexQuery;
 import org.roaringbitmap.RoaringBitmap;
 
 import static org.hamcrest.Matchers.containsString;
@@ -962,7 +963,7 @@ public void testBitmapQuery() throws IOException {
 
         NumberFieldType ft = new NumberFieldMapper.NumberFieldType("field", NumberType.INTEGER);
         assertEquals(
-            new IndexOrDocValuesQuery(NumberType.bitmapIndexQuery("field", r), new BitmapDocValuesQuery("field", r)),
+            new IndexOrDocValuesQuery(new BitmapIndexQuery("field", r), new BitmapDocValuesQuery("field", r)),
             ft.bitmapQuery(bitmap)
         );