Fix tests after feature flag

Signed-off-by: mgodwan <mgodwan@amazon.com>
opensearch-project · Feb 1, 2024 · b898785 · b898785
1 parent 278dfea
commit b898785
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/server/src/main/java/org/opensearch/index/codec/fuzzy/BloomFilter.java b/server/src/main/java/org/opensearch/index/codec/fuzzy/BloomFilter.java
@@ -61,7 +61,7 @@ public class BloomFilter extends AbstractFuzzySet {
         int setSize = (int) Math.ceil((maxDocs * Math.log(maxFpp)) / Math.log(1 / Math.pow(2, Math.log(2))));
         setSize = getNearestSetSize(setSize < Integer.MAX_VALUE / 2 ? 2 * setSize : Integer.MAX_VALUE);
         int optimalK = (int) Math.round(((double) setSize / maxDocs) * Math.log(2));
-        this.bitset = new LongArrayBackedBitSet(setSize + 1);
+        this.bitset = new LongArrayBackedBitSet(setSize);
         this.setSize = setSize;
         this.hashCount = optimalK;
         addAll(fieldIteratorProvider);

diff --git a/server/src/main/java/org/opensearch/index/codec/fuzzy/LongArrayBackedBitSet.java b/server/src/main/java/org/opensearch/index/codec/fuzzy/LongArrayBackedBitSet.java
@@ -32,7 +32,7 @@ class LongArrayBackedBitSet implements Accountable, Closeable {
      */
     LongArrayBackedBitSet(long capacity) {
         // Since the bitset is backed by a long array, we only need 1 element for every 64 bits in the underlying array.
-        underlyingArrayLength = ((capacity - 1L) >> 6) + 1;
+        underlyingArrayLength = (capacity >> 6) + 1L;
         this.longArray = BigArrays.NON_RECYCLING_INSTANCE.withCircuitBreaking().newLongArray(underlyingArrayLength);
     }
 

diff --git a/test/framework/src/main/java/org/opensearch/test/OpenSearchIntegTestCase.java b/test/framework/src/main/java/org/opensearch/test/OpenSearchIntegTestCase.java
@@ -653,6 +653,9 @@ protected Settings featureFlagSettings() {
         }
         // Enabling Telemetry setting by default
         featureSettings.put(FeatureFlags.TELEMETRY_SETTING.getKey(), true);
+
+        // Enabling fuzzy set for tests by default
+        featureSettings.put(FeatureFlags.DOC_ID_FUZZY_SET_SETTING.getKey(), true);
         return featureSettings.build();
     }