Fix term tests

Signed-off-by: Harsha Vamsi Kalluri <[email protected]>
opensearch-project · Jan 24, 2024 · ef9b400 · ef9b400
1 parent 3934e9c
commit ef9b400
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/server/src/test/java/org/opensearch/index/mapper/IpFieldTypeTests.java b/server/src/test/java/org/opensearch/index/mapper/IpFieldTypeTests.java
@@ -83,14 +83,20 @@ public void testTermQuery() {
         Query query = InetAddressPoint.newExactQuery("field", InetAddresses.forString(ip));
 
         assertEquals(
-            new IndexOrDocValuesQuery(query, SortedSetDocValuesField.newSlowExactQuery("field", new BytesRef(ip))),
+            new IndexOrDocValuesQuery(
+                query,
+                SortedSetDocValuesField.newSlowExactQuery("field", new BytesRef(((PointRangeQuery) query).getLowerPoint()))
+            ),
             ft.termQuery(ip, null)
         );
 
         ip = "192.168.1.7";
         query = InetAddressPoint.newExactQuery("field", InetAddresses.forString(ip));
         assertEquals(
-            new IndexOrDocValuesQuery(query, SortedSetDocValuesField.newSlowExactQuery("field", new BytesRef(ip))),
+            new IndexOrDocValuesQuery(
+                query,
+                SortedSetDocValuesField.newSlowExactQuery("field", new BytesRef(((PointRangeQuery) query).getLowerPoint()))
+            ),
             ft.termQuery(ip, null)
         );