githubFork/gitblit.git - Solinfo Gitblit

Fairly functional Lucene backend for commits, blobs, and issues

James Moger

2012-01-24 06ff61c04db32abc7ba4bbaeee604482c343152e

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
	83	newUser = new UserModel("garbage");
	84	newUser.password = "garbage";
	85	service.updateUserModel(newUser);
	86
	87	// confirm all added users
	88	assertEquals(3, service.getAllUsernames().size());
	89
	90	// confirm reloaded test user
	91	newUser = service.getUserModel("test");
	92	assertEquals("testPassword", newUser.password);
	93	assertEquals(3, newUser.repositories.size());
	94	assertTrue(newUser.hasRepository("repo1"));
	95	assertTrue(newUser.hasRepository("repo2"));
	96	assertTrue(newUser.hasRepository("sub/repo3"));
	97
	98	// confirm authentication of test user
	99	UserModel testUser = service.authenticate("test", "testPassword".toCharArray());
	100	assertEquals("test", testUser.username);
	101	assertEquals("testPassword", testUser.password);
	102
	103	// delete a repository role and confirm role removal from test user
	104	service.deleteRepositoryRole("repo2");
	105	testUser = service.getUserModel("test");
	106	assertEquals(2, testUser.repositories.size());
	107
	108	// delete garbage user and confirm user count
	109	service.deleteUser("garbage");
	110	assertEquals(2, service.getAllUsernames().size());
	111
	112	// rename repository and confirm role change for test user
	113	service.renameRepositoryRole("repo1", "newrepo1");
	114	testUser = service.getUserModel("test");
	115	assertTrue(testUser.hasRepository("newrepo1"));
	116	}
fe24a0	117
JM	118	protected void testTeams(IUserService service) {
	119
dcf575	120	// confirm we have 1 team (admins)
JM	121	assertEquals(1, service.getAllTeamNames().size());
	122	assertEquals("admins", service.getAllTeamNames().get(0));
fe24a0	123
JM	124	// remove newrepo1 from test user
	125	// now test user has no repositories
	126	UserModel user = service.getUserModel("test");
	127	user.repositories.clear();
	128	service.updateUserModel(user);
	129	user = service.getUserModel("test");
	130	assertEquals(0, user.repositories.size());
	131	assertFalse(user.canAccessRepository("newrepo1"));
	132	assertFalse(user.canAccessRepository("NEWREPO1"));
	133
	134	// create test team and add test user and newrepo1
	135	TeamModel team = new TeamModel("testteam");
	136	team.addUser("test");
	137	team.addRepository("newrepo1");
	138	service.updateTeamModel(team);
	139
	140	// confirm 1 user and 1 repo
	141	team = service.getTeamModel("testteam");
	142	assertEquals(1, team.repositories.size());
	143	assertEquals(1, team.users.size());
	144
	145	// confirm team membership
	146	user = service.getUserModel("test");
	147	assertEquals(0, user.repositories.size());
	148	assertEquals(1, user.teams.size());
	149
	150	// confirm team access
	151	assertTrue(team.hasRepository("newrepo1"));
	152	assertTrue(user.hasTeamAccess("newrepo1"));
	153	assertTrue(team.hasRepository("NEWREPO1"));
	154	assertTrue(user.hasTeamAccess("NEWREPO1"));
	155
	156	// rename the team and add new repository
	157	team.addRepository("newrepo2");
	158	team.name = "testteam2";
	159	service.updateTeamModel("testteam", team);
	160
	161	team = service.getTeamModel("testteam2");
	162	user = service.getUserModel("test");
	163
	164	// confirm user and team can access newrepo2
	165	assertEquals(2, team.repositories.size());
	166	assertTrue(team.hasRepository("newrepo2"));
	167	assertTrue(user.hasTeamAccess("newrepo2"));
	168	assertTrue(team.hasRepository("NEWREPO2"));
	169	assertTrue(user.hasTeamAccess("NEWREPO2"));
	170
	171	// delete testteam2
	172	service.deleteTeam("testteam2");
	173	team = service.getTeamModel("testteam2");
	174	user = service.getUserModel("test");
	175
	176	// confirm team does not exist and user can not access newrepo1 and 2
	177	assertEquals(null, team);
	178	assertFalse(user.canAccessRepository("newrepo1"));
	179	assertFalse(user.canAccessRepository("newrepo2"));
	180
	181	// create new team and add it to user
	182	// this tests the inverse team creation/team addition
	183	team = new TeamModel("testteam");
	184	team.addRepository("NEWREPO1");
	185	team.addRepository("NEWREPO2");
	186	user.teams.add(team);
	187	service.updateUserModel(user);
	188
	189	// confirm the inverted team addition
	190	user = service.getUserModel("test");
	191	team = service.getTeamModel("testteam");
	192	assertTrue(user.hasTeamAccess("newrepo1"));
	193	assertTrue(user.hasTeamAccess("newrepo2"));
	194	assertTrue(team.hasUser("test"));
	195
	196	// drop testteam from user and add nextteam to user
	197	team = new TeamModel("nextteam");
	198	team.addRepository("NEWREPO1");
	199	team.addRepository("NEWREPO2");
	200	user.teams.clear();
	201	user.teams.add(team);
	202	service.updateUserModel(user);
	203
	204	// confirm implicit drop
	205	user = service.getUserModel("test");
	206	team = service.getTeamModel("testteam");
	207	assertTrue(user.hasTeamAccess("newrepo1"));
	208	assertTrue(user.hasTeamAccess("newrepo2"));
	209	assertFalse(team.hasUser("test"));
	210	team = service.getTeamModel("nextteam");
	211	assertTrue(team.hasUser("test"));
	212
	213	// delete the user and confirm team no longer has user
	214	service.deleteUser("test");
	215	team = service.getTeamModel("testteam");
	216	assertFalse(team.hasUser("test"));
	217
	218	// delete both teams
	219	service.deleteTeam("testteam");
	220	service.deleteTeam("nextteam");
dcf575	221
JM	222	// assert we still have the admins team
	223	assertEquals(1, service.getAllTeamNames().size());
	224	assertEquals("admins", service.getAllTeamNames().get(0));
	225
	226	team = service.getTeamModel("admins");
	227	assertEquals(1, team.mailingLists.size());
	228	assertTrue(team.mailingLists.contains("admins@localhost.com"));
fe24a0	229	}
93f472	230	}