1 package info.jtrac.domain; 2 3 4 import java.util.Arrays ; 5 import java.util.HashSet ; 6 import java.util.Set ; 7 8 import junit.framework.TestCase; 9 10 import org.acegisecurity.GrantedAuthority; 11 12 public class UserTest extends TestCase { 13 14 public void testGetAuthoritiesFromUserSpaceRoles() { 15 16 Space s1 = new Space(); 17 s1.setPrefixCode("SPACE-ONE"); 18 19 User u = new User(); 20 u.setLoginName("test"); 21 22 u.addSpaceWithRole(s1, "ROLE_ONE-ONE"); 23 u.addSpaceWithRole(s1, "ROLE_ONE-TWO"); 24 u.addSpaceWithRole(null, "ROLE_ADMIN"); 25 u.setId(1); 26 27 GrantedAuthority[] gas = u.getAuthorities(); 28 29 Set <String > set = new HashSet <String >(); 30 for(GrantedAuthority ga : gas) { 31 set.add(ga.getAuthority()); 32 } 33 34 assertEquals(4, gas.length); 35 36 assertTrue(set.contains("ROLE_USER")); 37 assertTrue(set.contains("ROLE_ONE-ONE_SPACE-ONE")); 38 assertTrue(set.contains("ROLE_ONE-TWO_SPACE-ONE")); 39 assertTrue(set.contains("ROLE_ADMIN")); 40 41 } 42 43 public void testCheckIfAdminForAllSpaces() { 44 User u = new User(); 45 u.setLoginName("test"); 46 u.addSpaceWithRole(null, "ROLE_ADMIN"); 47 assertTrue(u.isAdminForAllSpaces()); 48 } 49 50 } 51 | Popular Tags |