Add overrides to more places

Signed-off-by: Harsha Vamsi Kalluri <[email protected]>
harshavamsi · Aug 8, 2024 · ae85095 · ae85095
1 parent 6566923
commit ae85095
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 9 deletions.
diff --git a/...es/mapper-extras/src/test/java/org/opensearch/index/mapper/ScaledFloatFieldTypeTests.java b/...es/mapper-extras/src/test/java/org/opensearch/index/mapper/ScaledFloatFieldTypeTests.java
@@ -122,6 +122,7 @@ public void testRangeQuery() throws IOException {
                 u,
                 includeLower,
                 includeUpper,
+                RewriteOverride.DEFAULT,
                 false,
                 true,
                 MOCK_QSC

diff --git a/server/src/test/java/org/opensearch/index/mapper/NumberFieldTypeTests.java b/server/src/test/java/org/opensearch/index/mapper/NumberFieldTypeTests.java
@@ -550,8 +550,28 @@ public void testHalfFloatRange() throws IOException {
             float u = (randomFloat() * 2 - 1) * 65504;
             boolean includeLower = randomBoolean();
             boolean includeUpper = randomBoolean();
-            Query floatQ = NumberType.FLOAT.rangeQuery("float", l, u, includeLower, includeUpper, false, true, MOCK_QSC);
-            Query halfFloatQ = NumberType.HALF_FLOAT.rangeQuery("half_float", l, u, includeLower, includeUpper, false, true, MOCK_QSC);
+            Query floatQ = NumberType.FLOAT.rangeQuery(
+                "float",
+                l,
+                u,
+                includeLower,
+                includeUpper,
+                RewriteOverride.DEFAULT,
+                false,
+                true,
+                MOCK_QSC
+            );
+            Query halfFloatQ = NumberType.HALF_FLOAT.rangeQuery(
+                "half_float",
+                l,
+                u,
+                includeLower,
+                includeUpper,
+                RewriteOverride.DEFAULT,
+                false,
+                true,
+                MOCK_QSC
+            );
             assertEquals(searcher.count(floatQ), searcher.count(halfFloatQ));
         }
         IOUtils.close(reader, dir);
@@ -587,28 +607,39 @@ public void testUnsignedLongRange() throws IOException {
                 u,
                 includeLower,
                 includeUpper,
+                RewriteOverride.DEFAULT,
+                false,
+                true,
+                MOCK_QSC
+            );
+            Query doubleQ = NumberType.DOUBLE.rangeQuery(
+                "double",
+                l,
+                u,
+                includeLower,
+                includeUpper,
+                RewriteOverride.DEFAULT,
                 false,
                 true,
                 MOCK_QSC
             );
-            Query doubleQ = NumberType.DOUBLE.rangeQuery("double", l, u, includeLower, includeUpper, false, true, MOCK_QSC);
             assertEquals(searcher.count(doubleQ), searcher.count(unsignedLongQ));
         }
         IOUtils.close(reader, dir);
     }
 
     public void testNegativeZero() {
         assertEquals(
-            NumberType.DOUBLE.rangeQuery("field", null, -0d, true, true, false, true, MOCK_QSC),
-            NumberType.DOUBLE.rangeQuery("field", null, +0d, true, false, false, true, MOCK_QSC)
+            NumberType.DOUBLE.rangeQuery("field", null, -0d, true, true, RewriteOverride.DEFAULT, false, true, MOCK_QSC),
+            NumberType.DOUBLE.rangeQuery("field", null, +0d, true, false, RewriteOverride.DEFAULT, false, true, MOCK_QSC)
         );
         assertEquals(
-            NumberType.FLOAT.rangeQuery("field", null, -0f, true, true, false, true, MOCK_QSC),
-            NumberType.FLOAT.rangeQuery("field", null, +0f, true, false, false, true, MOCK_QSC)
+            NumberType.FLOAT.rangeQuery("field", null, -0f, true, true, RewriteOverride.DEFAULT, false, true, MOCK_QSC),
+            NumberType.FLOAT.rangeQuery("field", null, +0f, true, false, RewriteOverride.DEFAULT, false, true, MOCK_QSC)
         );
         assertEquals(
-            NumberType.HALF_FLOAT.rangeQuery("field", null, -0f, true, true, false, true, MOCK_QSC),
-            NumberType.HALF_FLOAT.rangeQuery("field", null, +0f, true, false, false, true, MOCK_QSC)
+            NumberType.HALF_FLOAT.rangeQuery("field", null, -0f, true, true, RewriteOverride.DEFAULT, false, true, MOCK_QSC),
+            NumberType.HALF_FLOAT.rangeQuery("field", null, +0f, true, false, RewriteOverride.DEFAULT, false, true, MOCK_QSC)
         );
 
         assertFalse(NumberType.DOUBLE.termQuery("field", -0d, true, true).equals(NumberType.DOUBLE.termQuery("field", +0d, true, true)));
@@ -670,6 +701,7 @@ public void doTestDocValueRangeQueries(NumberType type, Supplier<Number> valueSu
                 random().nextBoolean() ? null : valueSupplier.get(),
                 randomBoolean(),
                 randomBoolean(),
+                RewriteOverride.DEFAULT,
                 true,
                 true,
                 MOCK_QSC
@@ -753,6 +785,7 @@ public void doTestIndexSortRangeQueries(NumberType type, Supplier<Number> valueS
                 random().nextBoolean() ? null : valueSupplier.get(),
                 randomBoolean(),
                 randomBoolean(),
+                RewriteOverride.DEFAULT,
                 true,
                 true,
                 context