Fix test broken by previous code review change

apache · Jan 3, 2025 · 67dd3ea · 67dd3ea
1 parent e7f5185
commit 67dd3ea
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/.../java/org/apache/accumulo/server/security/delegation/ZooAuthenticationKeyWatcherTest.java b/.../java/org/apache/accumulo/server/security/delegation/ZooAuthenticationKeyWatcherTest.java
@@ -41,6 +41,7 @@
 
 import org.apache.accumulo.core.Constants;
 import org.apache.accumulo.core.data.InstanceId;
+import org.apache.accumulo.core.fate.zookeeper.ZooReader;
 import org.apache.accumulo.core.fate.zookeeper.ZooUtil;
 import org.apache.accumulo.core.zookeeper.ZooSession;
 import org.apache.zookeeper.KeeperException.NoNodeException;
@@ -80,7 +81,11 @@ public void setupMocks() {
     instanceId = InstanceId.of(UUID.randomUUID());
     baseNode = ZooUtil.getRoot(instanceId) + Constants.ZDELEGATION_TOKEN_KEYS;
     secretManager = new AuthenticationTokenSecretManager(instanceId, tokenLifetime);
+
+    expect(zk.asReader()).andReturn(new ZooReader(zk)).once();
+    replay(zk);
     keyWatcher = new ZooAuthenticationKeyWatcher(secretManager, zk, baseNode);
+    reset(zk);
   }
 
   @AfterEach