Gitblit devel
view | history | commit | commitdiff
James Moger
2012-10-01 eb1405f736f2f98e14215774dd53eea9b9a77017
commit | author | age
93f472 1 /*
JM 2  * Copyright 2011 gitblit.com.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.gitblit.tests;
17
7e8873 18 import static org.junit.Assert.assertEquals;
JM 19 import static org.junit.Assert.assertFalse;
20 import static org.junit.Assert.assertTrue;
21
93f472 22 import java.io.File;
JM 23 import java.io.IOException;
24
7e8873 25 import org.junit.Test;
93f472 26
JM 27 import com.gitblit.ConfigUserService;
28 import com.gitblit.FileUserService;
29 import com.gitblit.IUserService;
7e8873 30 import com.gitblit.models.TeamModel;
93f472 31 import com.gitblit.models.UserModel;
JM 32
7e8873 33 public class UserServiceTest {
93f472 34
7e8873 35     @Test
93f472 36     public void testFileUserService() throws IOException {
JM 37         File file = new File("us-test.properties");
38         file.delete();
7e8873 39         IUserService service = new FileUserService(file);
JM 40         testUsers(service);
fe24a0 41         testTeams(service);
93f472 42         file.delete();
JM 43     }
44
7e8873 45     @Test
93f472 46     public void testConfigUserService() throws IOException {
JM 47         File file = new File("us-test.conf");
48         file.delete();
7e8873 49         IUserService service = new ConfigUserService(file);
JM 50         testUsers(service);
fe24a0 51         testTeams(service);
93f472 52         file.delete();
JM 53     }
54
7e8873 55     protected void testUsers(IUserService service) {
93f472 56
JM 57         UserModel admin = service.getUserModel("admin");
58         assertTrue(admin == null);
59
dcf575 60         // add admin and admins team
JM 61         TeamModel admins = new TeamModel("admins");
62         admins.mailingLists.add("admins@localhost.com");
63         
93f472 64         admin = new UserModel("admin");
JM 65         admin.password = "password";
66         admin.canAdmin = true;
67         admin.excludeFromFederation = true;
dcf575 68         admin.teams.add(admins);
JM 69         
93f472 70         service.updateUserModel(admin);
JM 71         admin = null;
dcf575 72         admins = null;
93f472 73
JM 74         // add new user
75         UserModel newUser = new UserModel("test");
76         newUser.password = "testPassword";
77         newUser.addRepository("repo1");
78         newUser.addRepository("repo2");
79         newUser.addRepository("sub/repo3");
80         service.updateUserModel(newUser);
81
82         // add one more new user and then test reload of first new user
ae0b13 83         newUser = new UserModel("GARBAGE");
93f472 84         newUser.password = "garbage";
JM 85         service.updateUserModel(newUser);
86
87         // confirm all added users
88         assertEquals(3, service.getAllUsernames().size());
ae0b13 89         assertTrue(service.getUserModel("garbage") != null);
JM 90         assertTrue(service.getUserModel("GaRbAgE") != null);
91         assertTrue(service.getUserModel("GARBAGE") != null);
93f472 92
JM 93         // confirm reloaded test user
94         newUser = service.getUserModel("test");
95         assertEquals("testPassword", newUser.password);
96         assertEquals(3, newUser.repositories.size());
97         assertTrue(newUser.hasRepository("repo1"));
98         assertTrue(newUser.hasRepository("repo2"));
99         assertTrue(newUser.hasRepository("sub/repo3"));
100
101         // confirm authentication of test user
102         UserModel testUser = service.authenticate("test", "testPassword".toCharArray());
103         assertEquals("test", testUser.username);
104         assertEquals("testPassword", testUser.password);
105
106         // delete a repository role and confirm role removal from test user
107         service.deleteRepositoryRole("repo2");
108         testUser = service.getUserModel("test");
109         assertEquals(2, testUser.repositories.size());
110
111         // delete garbage user and confirm user count
112         service.deleteUser("garbage");
113         assertEquals(2, service.getAllUsernames().size());
114
115         // rename repository and confirm role change for test user
116         service.renameRepositoryRole("repo1", "newrepo1");
117         testUser = service.getUserModel("test");
118         assertTrue(testUser.hasRepository("newrepo1"));
119     }
fe24a0 120
JM 121     protected void testTeams(IUserService service) {
122
dcf575 123         // confirm we have 1 team (admins)
JM 124         assertEquals(1, service.getAllTeamNames().size());
125         assertEquals("admins", service.getAllTeamNames().get(0));
fe24a0 126
JM 127         // remove newrepo1 from test user
128         // now test user has no repositories
129         UserModel user = service.getUserModel("test");
130         user.repositories.clear();
131         service.updateUserModel(user);
132         user = service.getUserModel("test");
133         assertEquals(0, user.repositories.size());
134         assertFalse(user.canAccessRepository("newrepo1"));
135         assertFalse(user.canAccessRepository("NEWREPO1"));
136
137         // create test team and add test user and newrepo1
138         TeamModel team = new TeamModel("testteam");
139         team.addUser("test");
140         team.addRepository("newrepo1");
141         service.updateTeamModel(team);
142
143         // confirm 1 user and 1 repo
144         team = service.getTeamModel("testteam");
145         assertEquals(1, team.repositories.size());
146         assertEquals(1, team.users.size());
147
148         // confirm team membership
149         user = service.getUserModel("test");
150         assertEquals(0, user.repositories.size());
151         assertEquals(1, user.teams.size());
152
153         // confirm team access
154         assertTrue(team.hasRepository("newrepo1"));
155         assertTrue(user.hasTeamAccess("newrepo1"));
156         assertTrue(team.hasRepository("NEWREPO1"));
157         assertTrue(user.hasTeamAccess("NEWREPO1"));
158
159         // rename the team and add new repository
160         team.addRepository("newrepo2");
161         team.name = "testteam2";
162         service.updateTeamModel("testteam", team);
163
164         team = service.getTeamModel("testteam2");
165         user = service.getUserModel("test");
166
167         // confirm user and team can access newrepo2
168         assertEquals(2, team.repositories.size());
169         assertTrue(team.hasRepository("newrepo2"));
170         assertTrue(user.hasTeamAccess("newrepo2"));
171         assertTrue(team.hasRepository("NEWREPO2"));
172         assertTrue(user.hasTeamAccess("NEWREPO2"));
173
174         // delete testteam2
175         service.deleteTeam("testteam2");
176         team = service.getTeamModel("testteam2");
177         user = service.getUserModel("test");
178
179         // confirm team does not exist and user can not access newrepo1 and 2
180         assertEquals(null, team);
181         assertFalse(user.canAccessRepository("newrepo1"));
182         assertFalse(user.canAccessRepository("newrepo2"));
183
184         // create new team and add it to user
185         // this tests the inverse team creation/team addition
186         team = new TeamModel("testteam");
187         team.addRepository("NEWREPO1");
188         team.addRepository("NEWREPO2");
189         user.teams.add(team);
190         service.updateUserModel(user);
191
192         // confirm the inverted team addition
193         user = service.getUserModel("test");
194         team = service.getTeamModel("testteam");
195         assertTrue(user.hasTeamAccess("newrepo1"));
196         assertTrue(user.hasTeamAccess("newrepo2"));
197         assertTrue(team.hasUser("test"));
198
199         // drop testteam from user and add nextteam to user
200         team = new TeamModel("nextteam");
201         team.addRepository("NEWREPO1");
202         team.addRepository("NEWREPO2");
203         user.teams.clear();
204         user.teams.add(team);
205         service.updateUserModel(user);
206
207         // confirm implicit drop
208         user = service.getUserModel("test");
209         team = service.getTeamModel("testteam");
210         assertTrue(user.hasTeamAccess("newrepo1"));
211         assertTrue(user.hasTeamAccess("newrepo2"));
212         assertFalse(team.hasUser("test"));
213         team = service.getTeamModel("nextteam");
214         assertTrue(team.hasUser("test"));
215
216         // delete the user and confirm team no longer has user
217         service.deleteUser("test");
218         team = service.getTeamModel("testteam");
219         assertFalse(team.hasUser("test"));
220
221         // delete both teams
222         service.deleteTeam("testteam");
223         service.deleteTeam("nextteam");
dcf575 224         
JM 225         // assert we still have the admins team
226         assertEquals(1, service.getAllTeamNames().size());
227         assertEquals("admins", service.getAllTeamNames().get(0));
228         
229         team = service.getTeamModel("admins");
230         assertEquals(1, team.mailingLists.size());
231         assertTrue(team.mailingLists.contains("admins@localhost.com"));
fe24a0 232     }
93f472 233 }