isWorking is tri-state

2 files changed, 1 insertions, 2 deletions

diff --git a/container-search/src/main/java/com/yahoo/search/dispatch/searchcluster/Group.java b/container-search/src/main/java/com/yahoo/search/dispatch/searchcluster/Group.java
index ad2581696a7..ec616a18e09 100644
--- a/container-search/src/main/java/com/yahoo/search/dispatch/searchcluster/Group.java
+++ b/container-search/src/main/java/com/yahoo/search/dispatch/searchcluster/Group.java
@@ -63,7 +63,7 @@ public class Group {
     }
 
     void aggregateNodeValues() {
-        activeDocuments.set(nodes.stream().filter(Node::isWorking).mapToLong(Node::getActiveDocuments).sum());
+        activeDocuments.set(nodes.stream().filter(node -> node.isWorking() == Boolean.TRUE).mapToLong(Node::getActiveDocuments).sum());
         isBlockingWrites.set(nodes.stream().anyMatch(node -> node.isBlockingWrites()));
     }
 
diff --git a/container-search/src/test/java/com/yahoo/search/dispatch/MockSearchCluster.java b/container-search/src/test/java/com/yahoo/search/dispatch/MockSearchCluster.java
index cebe5c5e2ba..cafba58662e 100644
--- a/container-search/src/test/java/com/yahoo/search/dispatch/MockSearchCluster.java
+++ b/container-search/src/test/java/com/yahoo/search/dispatch/MockSearchCluster.java
@@ -40,7 +40,6 @@ public class MockSearchCluster extends SearchCluster {
             List<Node> nodes = new ArrayList<>();
             for (int node = 0; node < nodesPerGroup; node++) {
                 Node n = new Node(distributionKey, "host" + distributionKey, group);
-                n.setWorking(true);
                 nodes.add(n);
                 hostBuilder.put(n.hostname(), n);
                 distributionKey++;