Register number of parties equal to possible concurrent arrivals

1 files changed, 2 insertions, 0 deletions

diff --git a/zkfacade/src/test/java/com/yahoo/vespa/curator/CuratorWrapperTest.java b/zkfacade/src/test/java/com/yahoo/vespa/curator/CuratorWrapperTest.java
index 131222e62f8..bf65045193e 100644
--- a/zkfacade/src/test/java/com/yahoo/vespa/curator/CuratorWrapperTest.java
+++ b/zkfacade/src/test/java/com/yahoo/vespa/curator/CuratorWrapperTest.java
@@ -87,7 +87,9 @@ public class CuratorWrapperTest {
             Singleton singleton = new Singleton(curator);
             assertTrue(singleton.isActive);
             assertTrue(wrapped.exists(lockPath));
+            stunning.register();
             assertTrue(curator.isActive(singleton.id()));
+            stunning.arriveAndDeregister();
             singleton.shutdown();
             assertFalse(singleton.isActive);
             // ... and deactivated as a result of unregistering again.