Author: bstansberry(a)jboss.com Date: 2007-10-25 02:20:32 -0400 (Thu, 25 Oct 2007) New Revision: 4687 Modified: core/branches/1.4.X/tests/functional/org/jboss/cache/buddyreplication/BuddyReplicationFailoverTest.java Log: [JBCACHE-1194] Test data gravitation when nodes with the data are slow to respond Modified: core/branches/1.4.X/tests/functional/org/jboss/cache/buddyreplication/BuddyReplicationFailoverTest.java =================================================================== --- core/branches/1.4.X/tests/functional/org/jboss/cache/buddyreplication/BuddyReplicationFailoverTest.java 2007-10-25 06:00:40 UTC (rev 4686) +++ core/branches/1.4.X/tests/functional/org/jboss/cache/buddyreplication/BuddyReplicationFailoverTest.java 2007-10-25 06:20:32 UTC (rev 4687) @@ -7,13 +7,18 @@ package org.jboss.cache.buddyreplication; import java.util.HashSet; +import java.util.List; import java.util.Set; import org.jboss.cache.AbstractTreeCacheListener; import org.jboss.cache.TreeCache; import org.jboss.cache.Fqn; import org.jboss.cache.config.Option; +import org.jboss.cache.interceptors.Interceptor; +import org.jboss.cache.marshall.JBCMethodCall; +import org.jboss.cache.marshall.MethodDeclarations; import org.jboss.cache.misc.TestingUtil; +import org.jgroups.blocks.MethodCall; /** * Tests behaviour when data owners fail - essentially this tests data gravitation @@ -248,40 +253,77 @@ assertNull("Backup should not exist on node3", caches[3].get(backupFqn, key)); // Add a listener that will delay data gravitation from node0 - BlockingListener listener0 = new BlockingListener(); - caches[1].addTreeCacheListener(listener0); + BlockingInterceptor interceptor0 = insertBlockingInterceptor(caches[0]); // Add a listener that will delay data gravitation from node1 - BlockingListener listener1 = new BlockingListener(); - caches[1].addTreeCacheListener(listener1); + BlockingInterceptor interceptor1 = insertBlockingInterceptor(caches[1]); Option opt = new Option(); opt.setForceDataGravitation(true); - assertEquals("Gravitation to node4 successful", value, caches[3].get(fqn, key, opt)); + assertEquals("Gravitation to node4 successful", value, caches[3].get(fqn, key, opt)); // Just double-check the listeners worked as expected - assertNull("Listener0 saw no exception", listener0.exception); - assertNull("Listener1 saw no exception", listener1.exception); - assertTrue("Known issue JBCACHE-1192: listener0 blocked " + fqn, listener0.visited.contains(fqn)); - assertTrue("Known issue JBCACHE-1192: listener1 blocked " + backupFqn, listener1.visited.contains(backupFqn)); + synchronized (interceptor0.monitor) + { + interceptor0.monitor.notifyAll(); + } + synchronized (interceptor1.monitor) + { + interceptor1.monitor.notifyAll(); + } + + TestingUtil.sleepThread(5); + + assertNull("interceptor0 saw no exception", interceptor0.exception); + assertNull("interceptor1 saw no exception", interceptor1.exception); + assertTrue("interceptor0 blocked " + fqn, interceptor0.blocked.contains(fqn)); + assertTrue("interceptor1 blocked " + backupFqn, interceptor1.blocked.contains(backupFqn)); + + } - private class BlockingListener extends AbstractTreeCacheListener + private BlockingInterceptor insertBlockingInterceptor(TreeCache cache) { + BlockingInterceptor interceptor = new BlockingInterceptor(); + + List interceptors = cache.getInterceptors(); + Interceptor first = (Interceptor) interceptors.get(0); + Interceptor next = first.getNext(); + + interceptor.setNext(next); + first.setNext(interceptor); + + return interceptor; + } + + private class BlockingInterceptor extends Interceptor + { Exception exception = null; - Set visited = new HashSet(); + Set blocked = new HashSet(); + Object monitor = new Object(); - public void nodeVisited(Fqn fqn) - { - try + public Object invoke(MethodCall m) throws Throwable + { + JBCMethodCall call = (JBCMethodCall) m; + if (call.getMethodId() == MethodDeclarations.getNodeMethodLocal_id) { - Thread.sleep(2000); - visited.add(fqn); + try + { + Object[] args = call.getArgs(); + Fqn fqn = (Fqn) args[0]; + synchronized (monitor) + { + monitor.wait(2000); + } + blocked.add(fqn); + } + catch (Exception e) + { + exception = e; + } + } - catch (Exception e) - { - exception = e; - } + return super.invoke(m); } }