[jboss-cvs] JBossAS SVN: r107245 - projects/cluster/ha-server-cache-ispn/trunk/src/test/java/org/jboss/ha/web/tomcat/service/session/sso.

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

Author: smarlow at redhat.com
Date: 2010-07-30 14:00:50 -0400 (Fri, 30 Jul 2010)
New Revision: 107245

Modified:
   projects/cluster/ha-server-cache-ispn/trunk/src/test/java/org/jboss/ha/web/tomcat/service/session/sso/SSOClusterManagerTest.java
Log:
JBCLUSTER-286 unit test

Modified: projects/cluster/ha-server-cache-ispn/trunk/src/test/java/org/jboss/ha/web/tomcat/service/session/sso/SSOClusterManagerTest.java
===================================================================
--- projects/cluster/ha-server-cache-ispn/trunk/src/test/java/org/jboss/ha/web/tomcat/service/session/sso/SSOClusterManagerTest.java	2010-07-30 17:41:52 UTC (rev 107244)
+++ projects/cluster/ha-server-cache-ispn/trunk/src/test/java/org/jboss/ha/web/tomcat/service/session/sso/SSOClusterManagerTest.java	2010-07-30 18:00:50 UTC (rev 107245)
@@ -4,25 +4,26 @@
 import org.infinispan.AdvancedCache;
 import org.infinispan.Cache;
 import org.infinispan.atomic.AtomicMap;
-import org.infinispan.atomic.AtomicMapLookup;
 import org.jboss.ha.framework.server.InfinispanHAPartitionCacheHandler;
 import org.jboss.web.tomcat.service.sso.spi.FullyQualifiedSessionId;
 import org.junit.Assert;
 import org.junit.Test;
-
+import org.junit.Before;
 import javax.transaction.Transaction;
 import javax.transaction.TransactionManager;
 
 public class SSOClusterManagerTest
 {
 
-   @Test
-   public void addSession() throws Exception
-   {
-      SSOClusterManagerImpl ssoManager = new SSOClusterManagerImpl();
-      String ssoId = "sso session 1";
+   SSOClusterManagerImpl ssoManager;
+   Cache<Object, Object> cache;
+   String ssoId;
 
-      Cache<Object, Object> cache = EasyMock.createMock(Cache.class);
+   @Before
+   public void setup() throws Exception
+   {
+      ssoManager = new SSOClusterManagerImpl();
+      cache = EasyMock.createMock(Cache.class);
       InfinispanHAPartitionCacheHandler cacheHandler = EasyMock.createMock(InfinispanHAPartitionCacheHandler.class);
       EasyMock.expect(cacheHandler.getCache()).andReturn(cache);
       EasyMock.expectLastCall().anyTimes();
@@ -44,31 +45,26 @@
       EasyMock.expectLastCall().anyTimes();
       tm.commit();
       EasyMock.expectLastCall().anyTimes();
-      
+
       EasyMock.expect(advancedCache.getTransactionManager()).andReturn(tm);
       EasyMock.expectLastCall().anyTimes();
 
+      ssoManager.setCacheHandler(cacheHandler);
+      ssoId = "sso session 1";
       AtomicMap atomicMap = EasyMock.createMock(AtomicMap.class);
       EasyMock.expect(cache.get(EasyMock.anyObject())).andReturn(atomicMap);
       EasyMock.expectLastCall().anyTimes();
-
       EasyMock.replay(cache, cacheHandler, advancedCache, tm, transaction, atomicMap);
-      
-      ssoManager.setCacheHandler(cacheHandler);
+   }
 
-      Assert.assertEquals(ssoManager.getCacheHandler(), cacheHandler);
-      Assert.assertEquals(cache, cacheHandler.getCache());
-      Assert.assertEquals(cache, ssoManager.getCacheHandler().getCache());
-
-      Assert.assertNull(ssoManager.getSSOLocalManager());
-
+   @Test
+   public void addSession() throws Exception
+   {
       ssoManager.start();
 
-
       FullyQualifiedSessionId fqs = new FullyQualifiedSessionId(ssoId, "context", "Boston");
       ssoManager.addSession(ssoId, fqs);
-
-      
+     
    }