diff options
Diffstat (limited to 'container-search/src/test/java/com/yahoo/search/dispatch/searchcluster')
-rw-r--r-- | container-search/src/test/java/com/yahoo/search/dispatch/searchcluster/MockSearchCluster.java | 14 |
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); } |