KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > jboss > portal > test > core > RoleModelTestCase


1 /*****************************************
2  * *
3  * JBoss Portal: The OpenSource Portal *
4  * *
5  * Distributable under LGPL license. *
6  * See terms of license at gnu.org. *
7  * *
8  *****************************************/

9 package org.jboss.portal.test.core;
10
11 import java.util.Arrays JavaDoc;
12 import java.util.HashSet JavaDoc;
13 import java.util.Iterator JavaDoc;
14 import java.util.Set JavaDoc;
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 /**
23  * @author <a HREF="mailto:julien@jboss.org">Julien Viet</a>
24  * @version $Revision: 1.1 $
25  */

26 public class RoleModelTestCase
27    extends ModelTestCase
28 {
29
30    public RoleModelTestCase(String JavaDoc name)
31    {
32       super(name);
33    }
34
35    public void testFindUser1() throws Exception JavaDoc
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 JavaDoc
47    {
48       Set JavaDoc roleNames = new HashSet JavaDoc(Arrays.asList(new Object JavaDoc[] { "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 JavaDoc 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 JavaDoc
61    {
62       Set JavaDoc c1 = userModule.findUsers(0, 1);
63       assertNotNull(c1);
64       assertEquals(1, c1.size());
65
66       Set JavaDoc c2 = userModule.findUsers(1, 1);
67       assertNotNull(c2);
68       assertEquals(1, c2.size());
69
70       Set JavaDoc c3 = userModule.findUsers(0, 2);
71       assertNotNull(c3);
72       assertEquals(2, c3.size());
73
74       Set JavaDoc s1 = new HashSet JavaDoc();
75       s1.add(((User) c1.iterator().next()).getUserName());
76       s1.add(((User) c2.iterator().next()).getUserName());
77       Set JavaDoc s2 = new HashSet JavaDoc();
78       Iterator JavaDoc 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 JavaDoc
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    // public void testCreateDuplicateUser() throws Exception
103
// {
104
// try
105
// {
106
// userModule.createUser("user1");
107
// fail("Expected exception on duplicate user");
108
// }
109
// catch (ModuleException expected)
110
// {
111
// }
112
// }
113

114    public void testCreateRole() throws Exception JavaDoc
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    // public void testCreateDuplicateRole() throws Exception
125
// {
126
// try
127
// {
128
// roleModule.createRole("user1");
129
// fail("Expected exception on duplicate role");
130
// }
131
// catch (ModuleException expected)
132
// {
133
// }
134
// }
135

136    public void testCountUser() throws Exception JavaDoc
137    {
138       assertEquals(2, userModule.getUserCount());
139    }
140
141    public void testRemoveNonExistingRole() throws Exception JavaDoc
142    {
143       try
144       {
145          roleModule.removeRole(new Integer JavaDoc(Integer.MAX_VALUE));
146          fail("Expected exception");
147       }
148       catch (ModuleException expected)
149       {
150       }
151    }
152
153    public void testRemoveRole() throws Exception JavaDoc
154    {
155       Role g1 = roleModule.findRoleByName("role1");
156       Integer JavaDoc 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 JavaDoc
181    {
182       User u1 = userModule.findUserByUserName("user1");
183       Integer JavaDoc 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 JavaDoc
211    {
212       Set JavaDoc roleNames = new HashSet JavaDoc(Arrays.asList(new Object JavaDoc[] { "role1", "role2" }));
213       Set JavaDoc roles = roleModule.findRoles();
214       assertEquals(2, roles.size());
215       Iterator JavaDoc 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 JavaDoc
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