Build proper connect string instead of just hostname

1 files changed, 17 insertions, 0 deletions

diff --git a/zkfacade/src/test/java/com/yahoo/vespa/curator/CuratorTest.java b/zkfacade/src/test/java/com/yahoo/vespa/curator/CuratorTest.java
index 2fc4c2a7fc4..c5227fcbaa5 100644
--- a/zkfacade/src/test/java/com/yahoo/vespa/curator/CuratorTest.java
+++ b/zkfacade/src/test/java/com/yahoo/vespa/curator/CuratorTest.java
@@ -2,6 +2,7 @@
 package com.yahoo.vespa.curator;
 
 import com.yahoo.cloud.config.ConfigserverConfig;
+import com.yahoo.net.HostName;
 import org.apache.curator.test.TestingServer;
 import org.junit.After;
 import org.junit.Before;
@@ -72,6 +73,22 @@ public class CuratorTest {
         }
     }
 
+    @Test
+     public void localhost_affinity() {
+         String localhostHostName = "myhost";
+         int localhostPort = 123;
+
+         ConfigserverConfig.Builder builder = new ConfigserverConfig.Builder();
+         builder.zookeeperserver(createZKBuilder(localhostHostName, localhostPort));
+         builder.zookeeperserver(createZKBuilder("otherhost", 345));
+         ConfigserverConfig config = new ConfigserverConfig(builder);
+
+         HostName.setHostNameForTestingOnly(localhostHostName);
+
+        String localhostSpec = localhostHostName + ":" + localhostPort;
+         assertThat(Curator.createConnectionSpecForLocalhost(config), is(localhostSpec));
+     }
+
     private ConfigserverConfig createTestConfig() {
         ConfigserverConfig.Builder builder = new ConfigserverConfig.Builder();
         builder.zookeeperserver(createZKBuilder("localhost", port1));