Allow deviation of at least 1 document

Let content be well-balanced when there are few docs in a cluster

1 files changed, 18 insertions, 0 deletions

diff --git a/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/SearchClusterCoverageTest.java b/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/SearchClusterCoverageTest.java
index 65e7173c4ee..c9f7469acbb 100644
--- a/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/SearchClusterCoverageTest.java
+++ b/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/SearchClusterCoverageTest.java
@@ -100,4 +100,22 @@ public class SearchClusterCoverageTest {
         assertTrue(tester.group(2).hasSufficientCoverage());
     }
 
+    @Test
+    public void one_group_few_docs_has_well_balanced_content() {
+        var tester = new SearchClusterTester(1, 2);
+
+        Node node0 = tester.group(0).nodes().get(0);
+        Node node1 = tester.group(0).nodes().get(1);
+
+        // 1 document
+        node0.setWorking(true);
+        node1.setWorking(true);
+
+        node0.setActiveDocuments(1);
+        node1.setActiveDocuments(0);
+
+        tester.pingIterationCompleted();
+        assertTrue(tester.group(0).isContentWellBalanced());
+    }
+
 }