ACCUMULO-4312 Fix jar sealing problem Move ZooLockTest.java to another package so it doesn't conflict with the package in the fate module (needed because the jars are sealed).
Project: http://git-wip-us.apache.org/repos/asf/accumulo/repo Commit: http://git-wip-us.apache.org/repos/asf/accumulo/commit/5b971d67 Tree: http://git-wip-us.apache.org/repos/asf/accumulo/tree/5b971d67 Diff: http://git-wip-us.apache.org/repos/asf/accumulo/diff/5b971d67 Branch: refs/heads/1.6 Commit: 5b971d675c8a395f6bc2924285b094d973e39ebe Parents: fd2c88b Author: Christopher Tubbs <ctubb...@apache.org> Authored: Fri May 27 15:06:35 2016 -0400 Committer: Christopher Tubbs <ctubb...@apache.org> Committed: Fri May 27 15:06:35 2016 -0400 ---------------------------------------------------------------------- .../accumulo/fate/zookeeper/ZooLockTest.java | 383 ------------------ .../test/fate/zookeeper/ZooLockTest.java | 385 +++++++++++++++++++ 2 files changed, 385 insertions(+), 383 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/accumulo/blob/5b971d67/test/src/test/java/org/apache/accumulo/fate/zookeeper/ZooLockTest.java ---------------------------------------------------------------------- diff --git a/test/src/test/java/org/apache/accumulo/fate/zookeeper/ZooLockTest.java b/test/src/test/java/org/apache/accumulo/fate/zookeeper/ZooLockTest.java deleted file mode 100644 index 2ad8161..0000000 --- a/test/src/test/java/org/apache/accumulo/fate/zookeeper/ZooLockTest.java +++ /dev/null @@ -1,383 +0,0 @@ -/* - * Licensed to the Apache Software Foundation (ASF) under one or more - * contributor license agreements. See the NOTICE file distributed with - * this work for additional information regarding copyright ownership. - * The ASF licenses this file to You under the Apache License, Version 2.0 - * (the "License"); you may not use this file except in compliance with - * the License. You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ -package org.apache.accumulo.fate.zookeeper; - -import java.io.File; -import java.lang.reflect.Field; -import java.util.Collections; -import java.util.List; -import java.util.concurrent.atomic.AtomicInteger; - -import org.apache.accumulo.fate.zookeeper.ZooLock.AsyncLockWatcher; -import org.apache.accumulo.fate.zookeeper.ZooLock.LockLossReason; -import org.apache.accumulo.minicluster.MiniAccumuloCluster; -import org.apache.log4j.Level; -import org.apache.log4j.Logger; -import org.apache.zookeeper.CreateMode; -import org.apache.zookeeper.WatchedEvent; -import org.apache.zookeeper.Watcher; -import org.apache.zookeeper.Watcher.Event.KeeperState; -import org.apache.zookeeper.ZooDefs; -import org.apache.zookeeper.ZooKeeper; -import org.junit.AfterClass; -import org.junit.Assert; -import org.junit.BeforeClass; -import org.junit.Test; -import org.junit.rules.TemporaryFolder; - -/** - * - */ -public class ZooLockTest { - - public static TemporaryFolder folder = new TemporaryFolder(new File(System.getProperty("user.dir") + "/target")); - - private static MiniAccumuloCluster accumulo; - - static class ConnectedWatcher implements Watcher { - volatile boolean connected = false; - - @Override - public synchronized void process(WatchedEvent event) { - if (event.getState() == KeeperState.SyncConnected) { // For ZK >3.4.... || event.getState() == KeeperState.ConnectedReadOnly) { - connected = true; - } else { - connected = false; - } - } - - public synchronized boolean isConnected() { - return connected; - } - } - - static class TestALW implements AsyncLockWatcher { - - LockLossReason reason = null; - boolean locked = false; - Exception exception = null; - int changes = 0; - - @Override - public synchronized void lostLock(LockLossReason reason) { - this.reason = reason; - changes++; - this.notifyAll(); - } - - @Override - public synchronized void acquiredLock() { - this.locked = true; - changes++; - this.notifyAll(); - } - - @Override - public synchronized void failedToAcquireLock(Exception e) { - this.exception = e; - changes++; - this.notifyAll(); - } - - public synchronized void waitForChanges(int numExpected) throws InterruptedException { - while (changes < numExpected) { - this.wait(); - } - } - - @Override - public synchronized void unableToMonitorLockNode(Throwable e) { - changes++; - this.notifyAll(); - } - } - - @BeforeClass - public static void setupMiniCluster() throws Exception { - - folder.create(); - - Logger.getLogger("org.apache.zookeeper").setLevel(Level.ERROR); - - accumulo = new MiniAccumuloCluster(folder.getRoot(), "superSecret"); - - accumulo.start(); - - } - - private static final AtomicInteger pdCount = new AtomicInteger(0); - - @Test(timeout = 10000) - public void testDeleteParent() throws Exception { - String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); - - ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); - - Assert.assertFalse(zl.isLocked()); - - ZooReaderWriter zk = ZooReaderWriter.getInstance(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes()); - - // intentionally created parent after lock - zk.mkdirs(parent); - - zk.delete(parent, -1); - - zk.mkdirs(parent); - - TestALW lw = new TestALW(); - - zl.lockAsync(lw, "test1".getBytes()); - - lw.waitForChanges(1); - - Assert.assertTrue(lw.locked); - Assert.assertTrue(zl.isLocked()); - Assert.assertNull(lw.exception); - Assert.assertNull(lw.reason); - - zl.unlock(); - } - - @Test(timeout = 10000) - public void testNoParent() throws Exception { - String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); - - ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); - - Assert.assertFalse(zl.isLocked()); - - TestALW lw = new TestALW(); - - zl.lockAsync(lw, "test1".getBytes()); - - lw.waitForChanges(1); - - Assert.assertFalse(lw.locked); - Assert.assertFalse(zl.isLocked()); - Assert.assertNotNull(lw.exception); - Assert.assertNull(lw.reason); - } - - @Test(timeout = 10000) - public void testDeleteLock() throws Exception { - String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); - - ZooReaderWriter zk = ZooReaderWriter.getInstance(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes()); - zk.mkdirs(parent); - - ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); - - Assert.assertFalse(zl.isLocked()); - - TestALW lw = new TestALW(); - - zl.lockAsync(lw, "test1".getBytes()); - - lw.waitForChanges(1); - - Assert.assertTrue(lw.locked); - Assert.assertTrue(zl.isLocked()); - Assert.assertNull(lw.exception); - Assert.assertNull(lw.reason); - - zk.delete(zl.getLockPath(), -1); - - lw.waitForChanges(2); - - Assert.assertEquals(LockLossReason.LOCK_DELETED, lw.reason); - Assert.assertNull(lw.exception); - - } - - @Test(timeout = 10000) - public void testDeleteWaiting() throws Exception { - String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); - - ZooReaderWriter zk = ZooReaderWriter.getInstance(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes()); - zk.mkdirs(parent); - - ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); - - Assert.assertFalse(zl.isLocked()); - - TestALW lw = new TestALW(); - - zl.lockAsync(lw, "test1".getBytes()); - - lw.waitForChanges(1); - - Assert.assertTrue(lw.locked); - Assert.assertTrue(zl.isLocked()); - Assert.assertNull(lw.exception); - Assert.assertNull(lw.reason); - - ZooLock zl2 = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); - - TestALW lw2 = new TestALW(); - - zl2.lockAsync(lw2, "test2".getBytes()); - - Assert.assertFalse(lw2.locked); - Assert.assertFalse(zl2.isLocked()); - - ZooLock zl3 = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); - - TestALW lw3 = new TestALW(); - - zl3.lockAsync(lw3, "test3".getBytes()); - - List<String> children = zk.getChildren(parent); - Collections.sort(children); - - zk.delete(parent + "/" + children.get(1), -1); - - lw2.waitForChanges(1); - - Assert.assertFalse(lw2.locked); - Assert.assertNotNull(lw2.exception); - Assert.assertNull(lw2.reason); - - zk.delete(parent + "/" + children.get(0), -1); - - lw.waitForChanges(2); - - Assert.assertEquals(LockLossReason.LOCK_DELETED, lw.reason); - Assert.assertNull(lw.exception); - - lw3.waitForChanges(1); - - Assert.assertTrue(lw3.locked); - Assert.assertTrue(zl3.isLocked()); - Assert.assertNull(lw3.exception); - Assert.assertNull(lw3.reason); - - zl3.unlock(); - - } - - @Test(timeout = 10000) - public void testUnexpectedEvent() throws Exception { - String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); - - ConnectedWatcher watcher = new ConnectedWatcher(); - ZooKeeper zk = new ZooKeeper(accumulo.getZooKeepers(), 30000, watcher); - zk.addAuthInfo("digest", "secret".getBytes()); - - while (!watcher.isConnected()) { - Thread.sleep(200); - } - - zk.create(parent, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); - - ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); - - Assert.assertFalse(zl.isLocked()); - - // would not expect data to be set on this node, but it should not cause problems..... - zk.setData(parent, "foo".getBytes(), -1); - - TestALW lw = new TestALW(); - - zl.lockAsync(lw, "test1".getBytes()); - - lw.waitForChanges(1); - - Assert.assertTrue(lw.locked); - Assert.assertTrue(zl.isLocked()); - Assert.assertNull(lw.exception); - Assert.assertNull(lw.reason); - - // would not expect data to be set on this node either - zk.setData(zl.getLockPath(), "bar".getBytes(), -1); - - zk.delete(zl.getLockPath(), -1); - - lw.waitForChanges(2); - - Assert.assertEquals(LockLossReason.LOCK_DELETED, lw.reason); - Assert.assertNull(lw.exception); - - } - - @Test(timeout = 10000) - public void testTryLock() throws Exception { - String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); - - ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 1000, "digest", "secret".getBytes(), parent); - - ConnectedWatcher watcher = new ConnectedWatcher(); - ZooKeeper zk = new ZooKeeper(accumulo.getZooKeepers(), 1000, watcher); - zk.addAuthInfo("digest", "secret".getBytes()); - - while (!watcher.isConnected()) { - Thread.sleep(200); - } - - for (int i = 0; i < 10; i++) { - zk.create(parent, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); - zk.delete(parent, -1); - } - - zk.create(parent, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); - - TestALW lw = new TestALW(); - - boolean ret = zl.tryLock(lw, "test1".getBytes()); - - Assert.assertTrue(ret); - - // make sure still watching parent even though a lot of events occurred for the parent - synchronized (zl) { - Field field = zl.getClass().getDeclaredField("watchingParent"); - field.setAccessible(true); - Assert.assertTrue((Boolean) field.get(zl)); - } - - zl.unlock(); - } - - @Test(timeout = 10000) - public void testChangeData() throws Exception { - String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); - ConnectedWatcher watcher = new ConnectedWatcher(); - ZooKeeper zk = new ZooKeeper(accumulo.getZooKeepers(), 1000, watcher); - zk.addAuthInfo("digest", "secret".getBytes()); - - while (!watcher.isConnected()) { - Thread.sleep(200); - } - - zk.create(parent, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); - - ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 1000, "digest", "secret".getBytes(), parent); - - TestALW lw = new TestALW(); - - zl.lockAsync(lw, "test1".getBytes()); - Assert.assertEquals("test1", new String(zk.getData(zl.getLockPath(), null, null))); - - zl.replaceLockData("test2".getBytes()); - Assert.assertEquals("test2", new String(zk.getData(zl.getLockPath(), null, null))); - } - - @AfterClass - public static void tearDownMiniCluster() throws Exception { - accumulo.stop(); - folder.delete(); - } - -} http://git-wip-us.apache.org/repos/asf/accumulo/blob/5b971d67/test/src/test/java/org/apache/accumulo/test/fate/zookeeper/ZooLockTest.java ---------------------------------------------------------------------- diff --git a/test/src/test/java/org/apache/accumulo/test/fate/zookeeper/ZooLockTest.java b/test/src/test/java/org/apache/accumulo/test/fate/zookeeper/ZooLockTest.java new file mode 100644 index 0000000..c0be29e --- /dev/null +++ b/test/src/test/java/org/apache/accumulo/test/fate/zookeeper/ZooLockTest.java @@ -0,0 +1,385 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.accumulo.test.fate.zookeeper; + +import java.io.File; +import java.lang.reflect.Field; +import java.util.Collections; +import java.util.List; +import java.util.concurrent.atomic.AtomicInteger; + +import org.apache.accumulo.fate.zookeeper.ZooLock; +import org.apache.accumulo.fate.zookeeper.ZooLock.AsyncLockWatcher; +import org.apache.accumulo.fate.zookeeper.ZooLock.LockLossReason; +import org.apache.accumulo.fate.zookeeper.ZooReaderWriter; +import org.apache.accumulo.minicluster.MiniAccumuloCluster; +import org.apache.log4j.Level; +import org.apache.log4j.Logger; +import org.apache.zookeeper.CreateMode; +import org.apache.zookeeper.WatchedEvent; +import org.apache.zookeeper.Watcher; +import org.apache.zookeeper.Watcher.Event.KeeperState; +import org.apache.zookeeper.ZooDefs; +import org.apache.zookeeper.ZooKeeper; +import org.junit.AfterClass; +import org.junit.Assert; +import org.junit.BeforeClass; +import org.junit.Test; +import org.junit.rules.TemporaryFolder; + +/** + * + */ +public class ZooLockTest { + + public static TemporaryFolder folder = new TemporaryFolder(new File(System.getProperty("user.dir") + "/target")); + + private static MiniAccumuloCluster accumulo; + + static class ConnectedWatcher implements Watcher { + volatile boolean connected = false; + + @Override + public synchronized void process(WatchedEvent event) { + if (event.getState() == KeeperState.SyncConnected) { // For ZK >3.4.... || event.getState() == KeeperState.ConnectedReadOnly) { + connected = true; + } else { + connected = false; + } + } + + public synchronized boolean isConnected() { + return connected; + } + } + + static class TestALW implements AsyncLockWatcher { + + LockLossReason reason = null; + boolean locked = false; + Exception exception = null; + int changes = 0; + + @Override + public synchronized void lostLock(LockLossReason reason) { + this.reason = reason; + changes++; + this.notifyAll(); + } + + @Override + public synchronized void acquiredLock() { + this.locked = true; + changes++; + this.notifyAll(); + } + + @Override + public synchronized void failedToAcquireLock(Exception e) { + this.exception = e; + changes++; + this.notifyAll(); + } + + public synchronized void waitForChanges(int numExpected) throws InterruptedException { + while (changes < numExpected) { + this.wait(); + } + } + + @Override + public synchronized void unableToMonitorLockNode(Throwable e) { + changes++; + this.notifyAll(); + } + } + + @BeforeClass + public static void setupMiniCluster() throws Exception { + + folder.create(); + + Logger.getLogger("org.apache.zookeeper").setLevel(Level.ERROR); + + accumulo = new MiniAccumuloCluster(folder.getRoot(), "superSecret"); + + accumulo.start(); + + } + + private static final AtomicInteger pdCount = new AtomicInteger(0); + + @Test(timeout = 10000) + public void testDeleteParent() throws Exception { + String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); + + ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); + + Assert.assertFalse(zl.isLocked()); + + ZooReaderWriter zk = ZooReaderWriter.getInstance(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes()); + + // intentionally created parent after lock + zk.mkdirs(parent); + + zk.delete(parent, -1); + + zk.mkdirs(parent); + + TestALW lw = new TestALW(); + + zl.lockAsync(lw, "test1".getBytes()); + + lw.waitForChanges(1); + + Assert.assertTrue(lw.locked); + Assert.assertTrue(zl.isLocked()); + Assert.assertNull(lw.exception); + Assert.assertNull(lw.reason); + + zl.unlock(); + } + + @Test(timeout = 10000) + public void testNoParent() throws Exception { + String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); + + ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); + + Assert.assertFalse(zl.isLocked()); + + TestALW lw = new TestALW(); + + zl.lockAsync(lw, "test1".getBytes()); + + lw.waitForChanges(1); + + Assert.assertFalse(lw.locked); + Assert.assertFalse(zl.isLocked()); + Assert.assertNotNull(lw.exception); + Assert.assertNull(lw.reason); + } + + @Test(timeout = 10000) + public void testDeleteLock() throws Exception { + String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); + + ZooReaderWriter zk = ZooReaderWriter.getInstance(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes()); + zk.mkdirs(parent); + + ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); + + Assert.assertFalse(zl.isLocked()); + + TestALW lw = new TestALW(); + + zl.lockAsync(lw, "test1".getBytes()); + + lw.waitForChanges(1); + + Assert.assertTrue(lw.locked); + Assert.assertTrue(zl.isLocked()); + Assert.assertNull(lw.exception); + Assert.assertNull(lw.reason); + + zk.delete(zl.getLockPath(), -1); + + lw.waitForChanges(2); + + Assert.assertEquals(LockLossReason.LOCK_DELETED, lw.reason); + Assert.assertNull(lw.exception); + + } + + @Test(timeout = 10000) + public void testDeleteWaiting() throws Exception { + String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); + + ZooReaderWriter zk = ZooReaderWriter.getInstance(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes()); + zk.mkdirs(parent); + + ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); + + Assert.assertFalse(zl.isLocked()); + + TestALW lw = new TestALW(); + + zl.lockAsync(lw, "test1".getBytes()); + + lw.waitForChanges(1); + + Assert.assertTrue(lw.locked); + Assert.assertTrue(zl.isLocked()); + Assert.assertNull(lw.exception); + Assert.assertNull(lw.reason); + + ZooLock zl2 = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); + + TestALW lw2 = new TestALW(); + + zl2.lockAsync(lw2, "test2".getBytes()); + + Assert.assertFalse(lw2.locked); + Assert.assertFalse(zl2.isLocked()); + + ZooLock zl3 = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); + + TestALW lw3 = new TestALW(); + + zl3.lockAsync(lw3, "test3".getBytes()); + + List<String> children = zk.getChildren(parent); + Collections.sort(children); + + zk.delete(parent + "/" + children.get(1), -1); + + lw2.waitForChanges(1); + + Assert.assertFalse(lw2.locked); + Assert.assertNotNull(lw2.exception); + Assert.assertNull(lw2.reason); + + zk.delete(parent + "/" + children.get(0), -1); + + lw.waitForChanges(2); + + Assert.assertEquals(LockLossReason.LOCK_DELETED, lw.reason); + Assert.assertNull(lw.exception); + + lw3.waitForChanges(1); + + Assert.assertTrue(lw3.locked); + Assert.assertTrue(zl3.isLocked()); + Assert.assertNull(lw3.exception); + Assert.assertNull(lw3.reason); + + zl3.unlock(); + + } + + @Test(timeout = 10000) + public void testUnexpectedEvent() throws Exception { + String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); + + ConnectedWatcher watcher = new ConnectedWatcher(); + ZooKeeper zk = new ZooKeeper(accumulo.getZooKeepers(), 30000, watcher); + zk.addAuthInfo("digest", "secret".getBytes()); + + while (!watcher.isConnected()) { + Thread.sleep(200); + } + + zk.create(parent, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); + + ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 30000, "digest", "secret".getBytes(), parent); + + Assert.assertFalse(zl.isLocked()); + + // would not expect data to be set on this node, but it should not cause problems..... + zk.setData(parent, "foo".getBytes(), -1); + + TestALW lw = new TestALW(); + + zl.lockAsync(lw, "test1".getBytes()); + + lw.waitForChanges(1); + + Assert.assertTrue(lw.locked); + Assert.assertTrue(zl.isLocked()); + Assert.assertNull(lw.exception); + Assert.assertNull(lw.reason); + + // would not expect data to be set on this node either + zk.setData(zl.getLockPath(), "bar".getBytes(), -1); + + zk.delete(zl.getLockPath(), -1); + + lw.waitForChanges(2); + + Assert.assertEquals(LockLossReason.LOCK_DELETED, lw.reason); + Assert.assertNull(lw.exception); + + } + + @Test(timeout = 10000) + public void testTryLock() throws Exception { + String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); + + ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 1000, "digest", "secret".getBytes(), parent); + + ConnectedWatcher watcher = new ConnectedWatcher(); + ZooKeeper zk = new ZooKeeper(accumulo.getZooKeepers(), 1000, watcher); + zk.addAuthInfo("digest", "secret".getBytes()); + + while (!watcher.isConnected()) { + Thread.sleep(200); + } + + for (int i = 0; i < 10; i++) { + zk.create(parent, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); + zk.delete(parent, -1); + } + + zk.create(parent, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); + + TestALW lw = new TestALW(); + + boolean ret = zl.tryLock(lw, "test1".getBytes()); + + Assert.assertTrue(ret); + + // make sure still watching parent even though a lot of events occurred for the parent + synchronized (zl) { + Field field = zl.getClass().getDeclaredField("watchingParent"); + field.setAccessible(true); + Assert.assertTrue((Boolean) field.get(zl)); + } + + zl.unlock(); + } + + @Test(timeout = 10000) + public void testChangeData() throws Exception { + String parent = "/zltest-" + this.hashCode() + "-l" + pdCount.incrementAndGet(); + ConnectedWatcher watcher = new ConnectedWatcher(); + ZooKeeper zk = new ZooKeeper(accumulo.getZooKeepers(), 1000, watcher); + zk.addAuthInfo("digest", "secret".getBytes()); + + while (!watcher.isConnected()) { + Thread.sleep(200); + } + + zk.create(parent, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); + + ZooLock zl = new ZooLock(accumulo.getZooKeepers(), 1000, "digest", "secret".getBytes(), parent); + + TestALW lw = new TestALW(); + + zl.lockAsync(lw, "test1".getBytes()); + Assert.assertEquals("test1", new String(zk.getData(zl.getLockPath(), null, null))); + + zl.replaceLockData("test2".getBytes()); + Assert.assertEquals("test2", new String(zk.getData(zl.getLockPath(), null, null))); + } + + @AfterClass + public static void tearDownMiniCluster() throws Exception { + accumulo.stop(); + folder.delete(); + } + +}
