[jboss-cvs] JBossAS SVN: r58941 - trunk/testsuite/src/main/org/jboss/test/cluster/web

[jboss-cvs-commits at lists.jboss.org]

Author: bstansberry at jboss.com
Date: 2006-12-08 22:44:46 -0500 (Fri, 08 Dec 2006)
New Revision: 58941

Modified:
   trunk/testsuite/src/main/org/jboss/test/cluster/web/CacheHelper.java
   trunk/testsuite/src/main/org/jboss/test/cluster/web/CacheHelperMBean.java
Log:
Add getSSOIds for use in ClusteredSSO tests

Modified: trunk/testsuite/src/main/org/jboss/test/cluster/web/CacheHelper.java
===================================================================
--- trunk/testsuite/src/main/org/jboss/test/cluster/web/CacheHelper.java	2006-12-09 03:40:02 UTC (rev 58940)
+++ trunk/testsuite/src/main/org/jboss/test/cluster/web/CacheHelper.java	2006-12-09 03:44:46 UTC (rev 58941)
@@ -137,6 +137,33 @@
       return result;
    }
    
+   public Set getSSOIds() throws Exception
+   {
+      Set result = new HashSet();
+      
+      Fqn fqn = Fqn.fromString("/SSO");
+      Node main = getCache().getChild(fqn);
+      if (main != null)
+      {
+         result.addAll(main.getChildrenNames());
+      }
+      
+      // Check in the buddy backup tree
+      
+      Set buddies = getBuddyBackupRoots();
+      for (Iterator iter = buddies.iterator(); iter.hasNext();)
+      {
+         Node buddy = (Node) iter.next();
+         Node ssoRoot = buddy.getChild(fqn);
+         if (ssoRoot != null)
+         {
+            result.addAll(ssoRoot.getChildrenNames());
+         }
+      }
+      
+      return result;
+   }
+   
    public boolean getLeaveInstalledAfterShutdown()
    {
       return leaveInstalledAfterShutdown;

Modified: trunk/testsuite/src/main/org/jboss/test/cluster/web/CacheHelperMBean.java
===================================================================
--- trunk/testsuite/src/main/org/jboss/test/cluster/web/CacheHelperMBean.java	2006-12-09 03:40:02 UTC (rev 58940)
+++ trunk/testsuite/src/main/org/jboss/test/cluster/web/CacheHelperMBean.java	2006-12-09 03:44:46 UTC (rev 58941)
@@ -10,6 +10,8 @@
 
    Set getSessionIds(String warFqn) throws Exception;
    
+   Set getSSOIds() throws Exception;
+   
    void setLeaveInstalledAfterShutdown();
    
    void uninstall();