[CURATOR-331] Make unhandledErrorListenable in TreeCache
diff --git a/curator-recipes/src/main/java/org/apache/curator/framework/recipes/cache/TreeCache.java b/curator-recipes/src/main/java/org/apache/curator/framework/recipes/cache/TreeCache.java
index 914d336..95b8b99 100644
--- a/curator-recipes/src/main/java/org/apache/curator/framework/recipes/cache/TreeCache.java
+++ b/curator-recipes/src/main/java/org/apache/curator/framework/recipes/cache/TreeCache.java
@@ -633,7 +633,7 @@
* TODO: consider making public.
*/
@VisibleForTesting
- Listenable<UnhandledErrorListener> getUnhandledErrorListenable()
+ public Listenable<UnhandledErrorListener> getUnhandledErrorListenable()
{
return errorListeners;
}
diff --git a/curator-recipes/src/test/java/org/apache/curator/framework/recipes/cache/TestTreeCache.java b/curator-recipes/src/test/java/org/apache/curator/framework/recipes/cache/TestTreeCache.java
index 2cf9e70..2d0ae83 100644
--- a/curator-recipes/src/test/java/org/apache/curator/framework/recipes/cache/TestTreeCache.java
+++ b/curator-recipes/src/test/java/org/apache/curator/framework/recipes/cache/TestTreeCache.java
@@ -6,9 +6,9 @@
* 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
- *
+ * <p>
+ * http://www.apache.org/licenses/LICENSE-2.0
+ * <p>
* 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
@@ -21,12 +21,15 @@
import com.google.common.collect.ImmutableSet;
import org.apache.curator.framework.CuratorFramework;
+import org.apache.curator.framework.api.UnhandledErrorListener;
+import org.apache.curator.framework.recipes.cache.TreeCacheEvent.Type;
import org.apache.curator.test.KillSession;
import org.apache.curator.utils.CloseableUtils;
import org.apache.zookeeper.CreateMode;
import org.testng.Assert;
import org.testng.annotations.Test;
import java.util.concurrent.Semaphore;
+import java.util.concurrent.atomic.AtomicBoolean;
public class TestTreeCache extends BaseTestTreeCache
{
@@ -567,4 +570,45 @@
assertEvent(TreeCacheEvent.Type.NODE_ADDED, "/test");
assertNoMoreEvents();
}
+
+ @Test
+ public void testErrorListener() throws Exception
+ {
+ client.create().forPath("/test");
+
+ cache = buildWithListeners(TreeCache.newBuilder(client, "/test"));
+
+ // Register a listener that throws an exception for the event
+ cache.getListenable().addListener(new TreeCacheListener()
+ {
+ @Override
+ public void childEvent(CuratorFramework client, TreeCacheEvent event) throws Exception
+ {
+ if (event.getType() == Type.NODE_UPDATED) {
+ throw new RuntimeException("Test Exception");
+ }
+ }
+ });
+
+ cache.getUnhandledErrorListenable().removeListener(errorListener);
+ final AtomicBoolean isProcessed = new AtomicBoolean(false);
+ cache.getUnhandledErrorListenable().addListener(new UnhandledErrorListener()
+ {
+ @Override
+ public void unhandledError(String message, Throwable e)
+ {
+ Assert.assertFalse(isProcessed.compareAndSet(false, true));
+ }
+ });
+
+ cache.start();
+
+ assertEvent(TreeCacheEvent.Type.NODE_ADDED, "/test");
+ assertEvent(TreeCacheEvent.Type.INITIALIZED);
+
+ client.setData().forPath("/test", "hey there".getBytes());
+ assertEvent(TreeCacheEvent.Type.NODE_UPDATED, "/test");
+
+ assertNoMoreEvents();
+ }
}