Revert "Jonmv/remove storage policy"

2 files changed, 12 insertions, 12 deletions

diff --git a/vespaclient-java/src/test/java/com/yahoo/vespaget/DocumentRetrieverTest.java b/vespaclient-java/src/test/java/com/yahoo/vespaget/DocumentRetrieverTest.java
index d6bda4b8bc1..d8b5c267bf2 100644
--- a/vespaclient-java/src/test/java/com/yahoo/vespaget/DocumentRetrieverTest.java
+++ b/vespaclient-java/src/test/java/com/yahoo/vespaget/DocumentRetrieverTest.java
@@ -234,14 +234,14 @@ public class DocumentRetrieverTest {
 
     @Test
     public void testClusterLookup() throws DocumentRetrieverException {
-        final String cluster = "storage",
-                expectedRoute = "storage-direct";
+        final String cluster = "storage", configId = "content/cluster.foo/storage",
+                expectedRoute = "[Storage:cluster=storage;clusterconfigid=content/cluster.foo/storage]";
 
         ClientParameters params = createParameters()
                 .setCluster(cluster)
                 .build();
 
-        ClusterList clusterList = new ClusterList(Collections.singletonList(new ClusterDef(cluster)));
+        ClusterList clusterList = new ClusterList(Collections.singletonList(new ClusterDef(cluster, configId)));
 
         DocumentRetriever documentRetriever = createDocumentRetriever(params, clusterList);
         documentRetriever.retrieveDocuments();
@@ -258,7 +258,7 @@ public class DocumentRetrieverTest {
                 .setCluster("invalidclustername")
                 .build();
 
-        ClusterList clusterList = new ClusterList(Collections.singletonList(new ClusterDef("storage")));
+        ClusterList clusterList = new ClusterList(Collections.singletonList(new ClusterDef("storage", "content/cluster.foo/storage")));
 
         DocumentRetriever documentRetriever = createDocumentRetriever(params, clusterList);
         documentRetriever.retrieveDocuments();
diff --git a/vespaclient-java/src/test/java/com/yahoo/vespavisit/VdsVisitTestCase.java b/vespaclient-java/src/test/java/com/yahoo/vespavisit/VdsVisitTestCase.java
index 7828fbb249d..4c8fbb1beee 100644
--- a/vespaclient-java/src/test/java/com/yahoo/vespavisit/VdsVisitTestCase.java
+++ b/vespaclient-java/src/test/java/com/yahoo/vespavisit/VdsVisitTestCase.java
@@ -228,17 +228,17 @@ public class VdsVisitTestCase {
     @Test
     public void testAutoSelectClusterRoute() throws Exception {
         List<ClusterDef> clusterDefs = new ArrayList<>();
-        clusterDefs.add(new ClusterDef("storage"));
+        clusterDefs.add(new ClusterDef("storage", "content/cluster.foo/storage"));
         ClusterList clusterList = new ClusterList(clusterDefs);
 
         String route = VdsVisit.resolveClusterRoute(clusterList, null);
-        assertEquals("storage-direct", route);
+        assertEquals("[Storage:cluster=storage;clusterconfigid=content/cluster.foo/storage]", route);
     }
 
     @Test
     public void testBadClusterName() throws Exception {
         List<ClusterDef> clusterDefs = new ArrayList<>();
-        clusterDefs.add(new ClusterDef("storage"));
+        clusterDefs.add(new ClusterDef("storage", "content/cluster.foo/storage"));
         ClusterList clusterList = new ClusterList(clusterDefs);
         try {
             VdsVisit.resolveClusterRoute(clusterList, "borkbork");
@@ -252,8 +252,8 @@ public class VdsVisitTestCase {
     @Test
     public void testRequireClusterOptionIfMultipleClusters() {
         List<ClusterDef> clusterDefs = new ArrayList<>();
-        clusterDefs.add(new ClusterDef("storage"));
-        clusterDefs.add(new ClusterDef("storage2"));
+        clusterDefs.add(new ClusterDef("storage", "content/cluster.foo/storage"));
+        clusterDefs.add(new ClusterDef("storage2", "content/cluster.bar/storage"));
         ClusterList clusterList = new ClusterList(clusterDefs);
         try {
             VdsVisit.resolveClusterRoute(clusterList, null);
@@ -265,12 +265,12 @@ public class VdsVisitTestCase {
     @Test
     public void testExplicitClusterOptionWithMultipleClusters() {
         List<ClusterDef> clusterDefs = new ArrayList<>();
-        clusterDefs.add(new ClusterDef("storage"));
-        clusterDefs.add(new ClusterDef("storage2"));
+        clusterDefs.add(new ClusterDef("storage", "content/cluster.foo/storage"));
+        clusterDefs.add(new ClusterDef("storage2", "content/cluster.bar/storage"));
         ClusterList clusterList = new ClusterList(clusterDefs);
 
         String route = VdsVisit.resolveClusterRoute(clusterList, "storage2");
-        assertEquals("storage2-direct", route);
+        assertEquals("[Storage:cluster=storage2;clusterconfigid=content/cluster.bar/storage]", route);
     }
 
     @Test