1 files changed, 33 insertions, 0 deletions
diff --git a/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/SearchClusterTester.java b/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/SearchClusterTester.java
new file mode 100644
index 00000000000..5e7ecb854ff
--- /dev/null
+++ b/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/SearchClusterTester.java
@@ -0,0 +1,33 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.search.dispatch.searchcluster;
+
+import com.yahoo.search.dispatch.MockSearchCluster;
+
+public class SearchClusterTester {
+
+    private final SearchCluster cluster;
+
+    public SearchClusterTester(int groups, int nodesPerGroup) {
+        cluster = new MockSearchCluster("1", groups, nodesPerGroup);
+    }
+
+    public void pingIterationCompleted() {
+        cluster.pingIterationCompleted();
+    }
+
+    public Group group(int id) {
+        return cluster.group(id).get();
+    }
+
+    public void setWorking(int group, int node, boolean working) {
+        cluster.group(group).get().nodes().get(node).setWorking(working);
+    }
+
+    public void setDocsPerNode(int docs, int groupId) {
+        for (Node node : cluster.groups().get(groupId).nodes()) {
+            node.setWorking(true);
+            node.setActiveDocuments(docs);
+        }
+    }
+
+}