From 27ae9095639bb228a1b7ff86a3ebe4264abf05be Mon Sep 17 00:00:00 2001
From: mschaefers <mschaefers@scoop-gmbh.de>
Date: Thu, 29 Nov 2012 12:33:09 -0500
Subject: [PATCH] feature: when using LdapUserService one can configure Gitblit to fetch all users from ldap that can possibly login. This allows to see newly generated LDAP users instantly in Gitblit. By now an LDAP user had to log in once to appear in GitBlit.

---
 tests/com/gitblit/tests/UserServiceTest.java |  177 +++++++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 161 insertions(+), 16 deletions(-)

diff --git a/tests/com/gitblit/tests/UserServiceTest.java b/tests/com/gitblit/tests/UserServiceTest.java
index 3dfdf7a..710d1f3 100644
--- a/tests/com/gitblit/tests/UserServiceTest.java
+++ b/tests/com/gitblit/tests/UserServiceTest.java
@@ -15,68 +15,90 @@
  */
 package com.gitblit.tests;
 
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
 import java.io.File;
 import java.io.IOException;
 
-import junit.framework.TestCase;
+import org.junit.Test;
 
 import com.gitblit.ConfigUserService;
+import com.gitblit.Constants.AccessRestrictionType;
 import com.gitblit.FileUserService;
 import com.gitblit.IUserService;
+import com.gitblit.models.RepositoryModel;
+import com.gitblit.models.TeamModel;
 import com.gitblit.models.UserModel;
 
-public class UserServiceTest extends TestCase {
+public class UserServiceTest {
 
+	@Test
 	public void testFileUserService() throws IOException {
 		File file = new File("us-test.properties");
 		file.delete();
-		test(new FileUserService(file));
+		IUserService service = new FileUserService(file);
+		testUsers(service);
+		testTeams(service);
 		file.delete();
 	}
 
+	@Test
 	public void testConfigUserService() throws IOException {
 		File file = new File("us-test.conf");
 		file.delete();
-		test(new ConfigUserService(file));
+		IUserService service = new ConfigUserService(file);
+		testUsers(service);
+		testTeams(service);
 		file.delete();
 	}
 
-	protected void test(IUserService service) {
+	protected void testUsers(IUserService service) {
 
 		UserModel admin = service.getUserModel("admin");
 		assertTrue(admin == null);
 
-		// add admin
+		// add admin and admins team
+		TeamModel admins = new TeamModel("admins");
+		admins.mailingLists.add("admins@localhost.com");
+		
 		admin = new UserModel("admin");
 		admin.password = "password";
 		admin.canAdmin = true;
 		admin.excludeFromFederation = true;
+		admin.teams.add(admins);
+		
 		service.updateUserModel(admin);
 		admin = null;
+		admins = null;
 
 		// add new user
 		UserModel newUser = new UserModel("test");
 		newUser.password = "testPassword";
-		newUser.addRepository("repo1");
-		newUser.addRepository("repo2");
-		newUser.addRepository("sub/repo3");
+		newUser.addRepositoryPermission("repo1");
+		newUser.addRepositoryPermission("repo2");
+		newUser.addRepositoryPermission("sub/repo3");
 		service.updateUserModel(newUser);
 
 		// add one more new user and then test reload of first new user
-		newUser = new UserModel("garbage");
+		newUser = new UserModel("GARBAGE");
 		newUser.password = "garbage";
 		service.updateUserModel(newUser);
 
 		// confirm all added users
 		assertEquals(3, service.getAllUsernames().size());
+		assertTrue(service.getUserModel("garbage") != null);
+		assertTrue(service.getUserModel("GaRbAgE") != null);
+		assertTrue(service.getUserModel("GARBAGE") != null);
 
 		// confirm reloaded test user
 		newUser = service.getUserModel("test");
 		assertEquals("testPassword", newUser.password);
-		assertEquals(3, newUser.repositories.size());
-		assertTrue(newUser.hasRepository("repo1"));
-		assertTrue(newUser.hasRepository("repo2"));
-		assertTrue(newUser.hasRepository("sub/repo3"));
+		assertEquals(3, newUser.permissions.size());
+		assertTrue(newUser.hasRepositoryPermission("repo1"));
+		assertTrue(newUser.hasRepositoryPermission("repo2"));
+		assertTrue(newUser.hasRepositoryPermission("sub/repo3"));
 
 		// confirm authentication of test user
 		UserModel testUser = service.authenticate("test", "testPassword".toCharArray());
@@ -86,7 +108,7 @@
 		// delete a repository role and confirm role removal from test user
 		service.deleteRepositoryRole("repo2");
 		testUser = service.getUserModel("test");
-		assertEquals(2, testUser.repositories.size());
+		assertEquals(2, testUser.permissions.size());
 
 		// delete garbage user and confirm user count
 		service.deleteUser("garbage");
@@ -95,6 +117,129 @@
 		// rename repository and confirm role change for test user
 		service.renameRepositoryRole("repo1", "newrepo1");
 		testUser = service.getUserModel("test");
-		assertTrue(testUser.hasRepository("newrepo1"));
+		assertTrue(testUser.hasRepositoryPermission("newrepo1"));
+	}
+
+	protected void testTeams(IUserService service) {
+
+		// confirm we have 1 team (admins)
+		assertEquals(1, service.getAllTeamNames().size());
+		assertEquals("admins", service.getAllTeamNames().get(0));
+		
+		RepositoryModel newrepo1 = new RepositoryModel("newrepo1", null, null, null);
+		newrepo1.accessRestriction = AccessRestrictionType.VIEW;
+		RepositoryModel NEWREPO1 = new RepositoryModel("NEWREPO1", null, null, null);
+		NEWREPO1.accessRestriction = AccessRestrictionType.VIEW;
+
+		// remove newrepo1 from test user
+		// now test user has no repositories
+		UserModel user = service.getUserModel("test");
+		user.permissions.clear();
+		service.updateUserModel(user);
+		user = service.getUserModel("test");
+		assertEquals(0, user.permissions.size());
+		assertFalse(user.canView(newrepo1));
+		assertFalse(user.canView(NEWREPO1));
+
+		// create test team and add test user and newrepo1
+		TeamModel team = new TeamModel("testteam");
+		team.addUser("test");
+		team.addRepositoryPermission(newrepo1.name);
+		service.updateTeamModel(team);
+
+		// confirm 1 user and 1 repo
+		team = service.getTeamModel("testteam");
+		assertEquals(1, team.permissions.size());
+		assertEquals(1, team.users.size());
+
+		// confirm team membership
+		user = service.getUserModel("test");
+		assertEquals(0, user.permissions.size());
+		assertEquals(1, user.teams.size());
+
+		// confirm team access
+		assertTrue(team.hasRepositoryPermission(newrepo1.name));
+		assertTrue(user.canView(newrepo1));
+		assertTrue(team.hasRepositoryPermission(NEWREPO1.name));
+		assertTrue(user.canView(NEWREPO1));
+
+		// rename the team and add new repository
+		RepositoryModel newrepo2 = new RepositoryModel("newrepo2", null, null, null);
+		newrepo2.accessRestriction = AccessRestrictionType.VIEW;
+		RepositoryModel NEWREPO2 = new RepositoryModel("NEWREPO2", null, null, null);
+		NEWREPO2.accessRestriction = AccessRestrictionType.VIEW;
+		
+		team.addRepositoryPermission(newrepo2.name);
+		team.name = "testteam2";
+		service.updateTeamModel("testteam", team);
+
+		team = service.getTeamModel("testteam2");
+		user = service.getUserModel("test");
+
+		// confirm user and team can access newrepo2
+		assertEquals(2, team.permissions.size());
+		assertTrue(team.hasRepositoryPermission(newrepo2.name));
+		assertTrue(user.canView(newrepo2));
+		assertTrue(team.hasRepositoryPermission(NEWREPO2.name));
+		assertTrue(user.canView(NEWREPO2));
+
+		// delete testteam2
+		service.deleteTeam("testteam2");
+		team = service.getTeamModel("testteam2");
+		user = service.getUserModel("test");
+
+		// confirm team does not exist and user can not access newrepo1 and 2
+		assertEquals(null, team);
+		assertFalse(user.canView(newrepo1));
+		assertFalse(user.canView(newrepo2));
+
+		// create new team and add it to user
+		// this tests the inverse team creation/team addition
+		team = new TeamModel("testteam");
+		team.addRepositoryPermission(NEWREPO1.name);
+		team.addRepositoryPermission(NEWREPO2.name);
+		user.teams.add(team);
+		service.updateUserModel(user);
+
+		// confirm the inverted team addition
+		user = service.getUserModel("test");
+		team = service.getTeamModel("testteam");
+		assertTrue(user.canView(newrepo1));
+		assertTrue(user.canView(newrepo2));
+		assertTrue(team.hasUser("test"));
+
+		// drop testteam from user and add nextteam to user
+		team = new TeamModel("nextteam");
+		team.addRepositoryPermission(NEWREPO1.name);
+		team.addRepositoryPermission(NEWREPO2.name);
+		user.teams.clear();
+		user.teams.add(team);
+		service.updateUserModel(user);
+
+		// confirm implicit drop
+		user = service.getUserModel("test");
+		team = service.getTeamModel("testteam");
+		assertTrue(user.canView(newrepo1));
+		assertTrue(user.canView(newrepo2));
+		assertFalse(team.hasUser("test"));
+		team = service.getTeamModel("nextteam");
+		assertTrue(team.hasUser("test"));
+
+		// delete the user and confirm team no longer has user
+		service.deleteUser("test");
+		team = service.getTeamModel("testteam");
+		assertFalse(team.hasUser("test"));
+
+		// delete both teams
+		service.deleteTeam("testteam");
+		service.deleteTeam("nextteam");
+		
+		// assert we still have the admins team
+		assertEquals(1, service.getAllTeamNames().size());
+		assertEquals("admins", service.getAllTeamNames().get(0));
+		
+		team = service.getTeamModel("admins");
+		assertEquals(1, team.mailingLists.size());
+		assertTrue(team.mailingLists.contains("admins@localhost.com"));
 	}
 }
\ No newline at end of file

--
Gitblit v1.9.1