[jbosscache-commits] JBoss Cache SVN: r7522 - core/trunk/src/test/java/org/jboss/cache/buddyreplication.

[jbosscache-commits at lists.jboss.org]

Author: mircea.markus
Date: 2009-01-20 06:05:42 -0500 (Tue, 20 Jan 2009)
New Revision: 7522

Modified:
   core/trunk/src/test/java/org/jboss/cache/buddyreplication/RemoveRootBuddyTest.java
Log:
more strict replication control

Modified: core/trunk/src/test/java/org/jboss/cache/buddyreplication/RemoveRootBuddyTest.java
===================================================================
--- core/trunk/src/test/java/org/jboss/cache/buddyreplication/RemoveRootBuddyTest.java	2009-01-20 10:37:05 UTC (rev 7521)
+++ core/trunk/src/test/java/org/jboss/cache/buddyreplication/RemoveRootBuddyTest.java	2009-01-20 11:05:42 UTC (rev 7522)
@@ -2,6 +2,8 @@
 
 import org.jboss.cache.CacheSPI;
 import org.jboss.cache.Fqn;
+import org.jboss.cache.commands.write.PutKeyValueCommand;
+import org.jboss.cache.util.internals.replicationlisteners.ReplicationListener;
 import org.testng.annotations.Test;
 
 import java.util.ArrayList;
@@ -20,6 +22,7 @@
    {
       CacheSPI cache1 = createCache(false, 1, "myBuddyPoolReplicationGroup", false, true, true);
       CacheSPI cache2 = createCache(false, 1, "myBuddyPoolReplicationGroup", false, true, true);
+      ReplicationListener replicationListener2 = ReplicationListener.getReplicationListener(cache2);
       List<CacheSPI<Object, Object>> caches = new ArrayList<CacheSPI<Object, Object>>(2);
       cachesTL.set(caches);
       caches.add(cache1);
@@ -30,7 +33,9 @@
          String key = String.valueOf(opCount);
          String value = String.valueOf(opCount);
          Fqn f = Fqn.fromElements("test", key);
+         replicationListener2.expect(PutKeyValueCommand.class);
          cache1.put(f, key, value);
+         replicationListener2.waitForReplicationToOccur();
       }
       cache1.removeNode(Fqn.ROOT);
    }