Let RpcResourcePool handle live changes.

1 files changed, 14 insertions, 0 deletions

diff --git a/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/MockSearchCluster.java b/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/MockSearchCluster.java
index cbf6273d3ae..5fb5b465c69 100644
--- a/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/MockSearchCluster.java
+++ b/container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/MockSearchCluster.java
@@ -2,6 +2,7 @@
 package com.yahoo.search.dispatch.searchcluster;
 
 import com.yahoo.vespa.config.search.DispatchConfig;
+import com.yahoo.vespa.config.search.DispatchNodesConfig;
 
 import java.util.ArrayList;
 import java.util.HashMap;
@@ -52,6 +53,19 @@ public class MockSearchCluster extends SearchCluster {
         return builder;
     }
 
+    public static DispatchNodesConfig createNodesConfig(int numGroups, int nodesPerGroup) {
+        var builder = new DispatchNodesConfig.Builder();
+        int key = 0;
+        for (int g = 0; g < numGroups; g++) {
+            for (int i = 0; i < nodesPerGroup; i++) {
+                var nodeBuilder = new DispatchNodesConfig.Node.Builder();
+                nodeBuilder.key(key++).port(0).group(g).host("host" + g + "." + i);
+                builder.node.add(nodeBuilder);
+            }
+        }
+        return builder.build();
+    }
+
     public static SearchGroupsImpl buildGroupListForTest(int numGroups, int nodesPerGroup, double minActivedocsPercentage) {
         return new SearchGroupsImpl(buildGroupMapForTest(numGroups, nodesPerGroup), minActivedocsPercentage);
     }