summaryrefslogtreecommitdiffstats
path: root/zkfacade
diff options
context:
space:
mode:
Diffstat (limited to 'zkfacade')
-rw-r--r--zkfacade/src/main/java/com/yahoo/vespa/curator/mock/MockCurator.java9
1 files changed, 9 insertions, 0 deletions
diff --git a/zkfacade/src/main/java/com/yahoo/vespa/curator/mock/MockCurator.java b/zkfacade/src/main/java/com/yahoo/vespa/curator/mock/MockCurator.java
index c7a24c14a4d..80f5a120364 100644
--- a/zkfacade/src/main/java/com/yahoo/vespa/curator/mock/MockCurator.java
+++ b/zkfacade/src/main/java/com/yahoo/vespa/curator/mock/MockCurator.java
@@ -73,6 +73,7 @@ public class MockCurator extends Curator {
private boolean shouldTimeoutOnEnter = false;
private int monotonicallyIncreasingNumber = 0;
private final boolean stableOrdering;
+ private String connectionSpec = "";
/** The file system used by this mock to store zookeeper files and directories */
private final MemoryFileSystem fileSystem = new MemoryFileSystem();
@@ -116,6 +117,14 @@ public class MockCurator extends Curator {
return Optional.ofNullable(atomicCounters.get(path));
}
+ /**
+ * Assigns the connection string, which must be on the form host1:port,host2:port ...
+ */
+ public void setConnectionSpec(String connectionSpec) { this.connectionSpec = connectionSpec; }
+
+ @Override
+ public String connectionSpec() { return connectionSpec; }
+
// ----- Start of adaptor methods from Curator to the mock file system -----
/** Creates a node below the given directory root */