[teiid-commits] teiid SVN: r4141 - branches/7.7.x/runtime/src/test/java/org/teiid/services.

[teiid-commits at lists.jboss.org]

Author: rareddy
Date: 2012-05-24 15:23:47 -0400 (Thu, 24 May 2012)
New Revision: 4141

Modified:
   branches/7.7.x/runtime/src/test/java/org/teiid/services/TestMembershipServiceImpl.java
Log:
TEIID-2037, TEIID-2055: fixing the test

Modified: branches/7.7.x/runtime/src/test/java/org/teiid/services/TestMembershipServiceImpl.java
===================================================================
--- branches/7.7.x/runtime/src/test/java/org/teiid/services/TestMembershipServiceImpl.java	2012-05-24 18:00:40 UTC (rev 4140)
+++ branches/7.7.x/runtime/src/test/java/org/teiid/services/TestMembershipServiceImpl.java	2012-05-24 19:23:47 UTC (rev 4141)
@@ -59,17 +59,19 @@
     	HashSet<Principal> principals = new HashSet<Principal>();
     	principals.add(p);
     	
-    	Subject subject = new Subject(false, principals, new HashSet(), new HashSet());
+    	final Subject subject = new Subject(false, principals, new HashSet(), new HashSet());
     	SecurityHelper sh = Mockito.mock(SecurityHelper.class);
     	Mockito.stub(sh.getSubjectInContext("passthrough")).toReturn(subject); //$NON-NLS-1$
     	
         TeiidLoginContext membershipService = new TeiidLoginContext(sh) {
 			public LoginContext createLoginContext(String domain, CallbackHandler handler) throws LoginException {
         		LoginContext context =  Mockito.mock(LoginContext.class);
+        		Mockito.stub(context.getSubject()).toReturn(subject);
         		return context;
         	}
 			protected LoginContext createLoginContext(String domain, Subject subject) throws LoginException {
         		LoginContext context =  Mockito.mock(LoginContext.class);
+        		Mockito.stub(context.getSubject()).toReturn(subject);
         		return context;
 		    }			
         };