Fix flaky test, closing gap in previous backport (#707)

Signed-off-by: Martin Gaievski <[email protected]> Signed-off-by: Martin Gaievski <[email protected]>
opensearch-project · Jan 4, 2023 · efa2dd4 · efa2dd4
1 parent 1e8b8c9
commit efa2dd4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/test/java/org/opensearch/knn/index/codec/KNNCodecTestCase.java b/src/test/java/org/opensearch/knn/index/codec/KNNCodecTestCase.java
@@ -320,7 +320,7 @@ public void testKnnVectorIndex(
         IndexReader reader = writer.getReader();
         writer.close();
 
-        verify(perFieldKnnVectorsFormatSpy).getKnnVectorsFormatForField(eq(FIELD_NAME_ONE));
+        verify(perFieldKnnVectorsFormatSpy, atLeastOnce()).getKnnVectorsFormatForField(eq(FIELD_NAME_ONE));
 
         IndexSearcher searcher = new IndexSearcher(reader);
         Query query = KNNQueryFactory.create(KNNEngine.LUCENE, "dummy", FIELD_NAME_ONE, new float[] { 1.0f, 0.0f, 0.0f }, 1);