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

[jbosscache-commits at lists.jboss.org]

Author: manik.surtani at jboss.com
Date: 2007-10-23 13:47:12 -0400 (Tue, 23 Oct 2007)
New Revision: 4677

Modified:
   core/trunk/src/test/java/org/jboss/cache/buddyreplication/BuddyReplicationTestsBase.java
Log:
modified helper methods

Modified: core/trunk/src/test/java/org/jboss/cache/buddyreplication/BuddyReplicationTestsBase.java
===================================================================
--- core/trunk/src/test/java/org/jboss/cache/buddyreplication/BuddyReplicationTestsBase.java	2007-10-23 17:31:08 UTC (rev 4676)
+++ core/trunk/src/test/java/org/jboss/cache/buddyreplication/BuddyReplicationTestsBase.java	2007-10-23 17:47:12 UTC (rev 4677)
@@ -6,16 +6,6 @@
  */
 package org.jboss.cache.buddyreplication;
 
-import static org.testng.AssertJUnit.assertEquals;
-import static org.testng.AssertJUnit.assertFalse;
-import static org.testng.AssertJUnit.assertTrue;
-
-import java.io.File;
-import java.util.ArrayList;
-import java.util.List;
-
-import javax.transaction.TransactionManager;
-
 import org.jboss.cache.Cache;
 import org.jboss.cache.CacheImpl;
 import org.jboss.cache.CacheSPI;
@@ -33,10 +23,15 @@
 import org.jboss.cache.misc.TestingUtil;
 import org.jboss.cache.xml.XmlHelper;
 import org.jgroups.Address;
+import static org.testng.AssertJUnit.*;
 import org.testng.annotations.AfterMethod;
 import org.testng.annotations.Test;
 import org.w3c.dom.Element;
 
+import javax.transaction.TransactionManager;
+import java.util.ArrayList;
+import java.util.List;
+
 /**
  * Base class for BR tests
  *
@@ -275,11 +270,12 @@
       return caches;
    }
 
-   protected void printBuddyGroup(CacheSPI cache)
+   protected void printBuddyGroup(Cache cache)
    {
-      BuddyGroup bg = cache.getBuddyManager().buddyGroup;
+      BuddyManager bm = ((CacheSPI) cache).getBuddyManager();
+      BuddyGroup bg = bm.buddyGroup;
       System.out.println("*** " + bg);
-      System.out.println("    Groups I participate in: " + cache.getBuddyManager().buddyGroupsIParticipateIn.keySet());
+      System.out.println("    Groups I participate in: " + bm.buddyGroupsIParticipateIn.keySet());
    }
 
    /**
@@ -290,7 +286,7 @@
       return 1000;
    }
 
-   protected void assertIsBuddy(CacheSPI dataOwner, CacheSPI buddy, boolean onlyBuddy)
+   protected void assertIsBuddy(Cache dataOwner, Cache buddy, boolean onlyBuddy)
    {
       Address dataOwnerLocalAddress = dataOwner.getLocalAddress();
       Address buddyLocalAddress = buddy.getLocalAddress();
@@ -298,8 +294,8 @@
       System.out.println("*** assert with groups.  Testing that " + buddyLocalAddress + " is a buddy for owner " + dataOwnerLocalAddress + " only buddy? " + onlyBuddy);
       printBuddyGroup(dataOwner);
 
-      BuddyManager dataOwnerBuddyManager = dataOwner.getBuddyManager();
-      BuddyManager buddyBuddyManager = buddy.getBuddyManager();
+      BuddyManager dataOwnerBuddyManager = ((CacheSPI) dataOwner).getBuddyManager();
+      BuddyManager buddyBuddyManager = ((CacheSPI) buddy).getBuddyManager();
 
       // lets test things on the data owner's side of things
       if (onlyBuddy) assertEquals("Should only have one buddy", 1, dataOwnerBuddyManager.getBuddyAddresses().size());