1 9 package org.jboss.portal.test.core; 10 11 import java.util.Arrays ; 12 import java.util.HashSet ; 13 import java.util.Iterator ; 14 import java.util.Set ; 15 16 import org.jboss.portal.core.impl.role.RoleImpl; 17 import org.jboss.portal.core.impl.user.UserImpl; 18 import org.jboss.portal.core.model.Role; 19 import org.jboss.portal.core.model.User; 20 import org.jboss.portal.core.modules.ModuleException; 21 22 26 public class RoleModelTestCase 27 extends ModelTestCase 28 { 29 30 public RoleModelTestCase(String name) 31 { 32 super(name); 33 } 34 35 public void testFindUser1() throws Exception 36 { 37 User u1 = userModule.findUserByUserName("user1"); 38 assertNotNull(u1); 39 assertEquals("user1", u1.getUserName()); 40 assertNotNull(u1.getRoles()); 41 assertEquals(1, u1.getRoles().size()); 42 assertEquals("role1", ((Role) u1.getRoles().iterator().next()).getName()); 43 assertEquals(u1.getID(), userModule.findUserByID(u1.getID()).getID()); 44 } 45 46 public void testFindUser2() throws Exception 47 { 48 Set roleNames = new HashSet (Arrays.asList(new Object [] { "role1", "role2" })); 49 User u2 = userModule.findUserByUserName("user2"); 50 assertNotNull(u2); 51 assertEquals("user2", u2.getUserName()); 52 assertNotNull(u2.getRoles()); 53 assertEquals(2, u2.getRoles().size()); 54 Iterator iterator = u2.getRoles().iterator(); 55 assertTrue(roleNames.contains(((Role) iterator.next()).getName())); 56 assertTrue(roleNames.contains(((Role) iterator.next()).getName())); 57 assertEquals(u2.getID(), userModule.findUserByID(u2.getID()).getID()); 58 } 59 60 public void testFindUsers() throws Exception 61 { 62 Set c1 = userModule.findUsers(0, 1); 63 assertNotNull(c1); 64 assertEquals(1, c1.size()); 65 66 Set c2 = userModule.findUsers(1, 1); 67 assertNotNull(c2); 68 assertEquals(1, c2.size()); 69 70 Set c3 = userModule.findUsers(0, 2); 71 assertNotNull(c3); 72 assertEquals(2, c3.size()); 73 74 Set s1 = new HashSet (); 75 s1.add(((User) c1.iterator().next()).getUserName()); 76 s1.add(((User) c2.iterator().next()).getUserName()); 77 Set s2 = new HashSet (); 78 Iterator iterator = c3.iterator(); 79 s2.add(((User) iterator.next()).getUserName()); 80 s2.add(((User) iterator.next()).getUserName()); 81 assertEquals(s1, s2); 82 } 83 84 public void testCreateUser() throws Exception 85 { 86 User u3 = userModule.createUser("user3", "password", "realemail"); 87 Role g1 = roleModule.findRoleByName("role1"); 88 u3.getRoles().add(g1); 89 assertEquals("user3", u3.getUserName()); 90 assertNotNull(u3.getRoles()); 91 assertEquals(1, u3.getRoles().size()); 92 assertEquals("role1", ((Role) u3.getRoles().iterator().next()).getName()); 93 nextSession(); 94 u3 = userModule.findUserByUserName("user3"); 95 assertNotNull(u3); 96 assertEquals("user3", u3.getUserName()); 97 assertNotNull(u3.getRoles()); 98 assertEquals(1, u3.getRoles().size()); 99 assertEquals("role1", ((Role) u3.getRoles().iterator().next()).getName()); 100 } 101 102 114 public void testCreateRole() throws Exception 115 { 116 Role g3 = roleModule.createRole("role3", "roleName"); 117 assertEquals("role3", g3.getName()); 118 nextSession(); 119 g3 = roleModule.findRoleByName("role3"); 120 assertNotNull(g3); 121 assertEquals("role3", g3.getName()); 122 } 123 124 136 public void testCountUser() throws Exception 137 { 138 assertEquals(2, userModule.getUserCount()); 139 } 140 141 public void testRemoveNonExistingRole() throws Exception 142 { 143 try 144 { 145 roleModule.removeRole(new Integer (Integer.MAX_VALUE)); 146 fail("Expected exception"); 147 } 148 catch (ModuleException expected) 149 { 150 } 151 } 152 153 public void testRemoveRole() throws Exception 154 { 155 Role g1 = roleModule.findRoleByName("role1"); 156 Integer id = g1.getID(); 157 roleModule.removeRole(id); 158 try 159 { 160 roleModule.findRoleByID(id); 161 fail("Expected exception"); 162 } 163 catch (ModuleException expected) 164 { 165 } 166 nextSession(); 167 try 168 { 169 roleModule.findRoleByID(id); 170 fail("Expected exception"); 171 } 172 catch (ModuleException expected) 173 { 174 } 175 User u1 = userModule.findUserByUserName("user1"); 176 assertNotNull(u1); 177 assertEquals("user1", u1.getUserName()); 178 } 179 180 public void testRemoveUser() throws Exception 181 { 182 User u1 = userModule.findUserByUserName("user1"); 183 Integer id = u1.getID(); 184 userModule.removeUser(id); 185 try 186 { 187 userModule.findUserByID(id); 188 fail("Expected exception"); 189 } 190 catch (ModuleException expected) 191 { 192 } 193 nextSession(); 194 try 195 { 196 userModule.findUserByID(id); 197 fail("Expected exception"); 198 } 199 catch (ModuleException expected) 200 { 201 } 202 Role g1 = roleModule.findRoleByName("role1"); 203 assertNotNull(g1); 204 assertEquals("role1", g1.getName()); 205 Role g2 = roleModule.findRoleByName("role2"); 206 assertNotNull(g2); 207 assertEquals("role2", g2.getName()); 208 } 209 210 public void testFindRoles() throws Exception 211 { 212 Set roleNames = new HashSet (Arrays.asList(new Object [] { "role1", "role2" })); 213 Set roles = roleModule.findRoles(); 214 assertEquals(2, roles.size()); 215 Iterator iterator = roles.iterator(); 216 assertTrue(roleNames.contains(((Role) iterator.next()).getName())); 217 assertTrue(roleNames.contains(((Role) iterator.next()).getName())); 218 } 219 220 protected void populate() throws Exception 221 { 222 User u1 = new UserImpl("user1"); 223 User u2 = new UserImpl("user2"); 224 Role g1 = new RoleImpl("role1"); 225 Role g2 = new RoleImpl("role2"); 226 227 session.save(u1); 228 session.save(u2); 229 session.save(g1); 230 session.save(g2); 231 232 u1.getRoles().add(g1); 233 u2.getRoles().add(g1); 234 u2.getRoles().add(g2); 235 } 236 } 237 | Popular Tags |