From 20714aee0d2d2a989d93d6065e081aed8ac85fbf Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Wed, 10 Oct 2012 00:05:34 -0400
Subject: [PATCH] Finer-grained repository access permissions (issue 36)

---
 tests/com/gitblit/tests/GitBlitTest.java |   23 ++++++++++++-----------
 1 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/tests/com/gitblit/tests/GitBlitTest.java b/tests/com/gitblit/tests/GitBlitTest.java
index 418f938..a188f18 100644
--- a/tests/com/gitblit/tests/GitBlitTest.java
+++ b/tests/com/gitblit/tests/GitBlitTest.java
@@ -52,20 +52,21 @@
 		List<String> users = GitBlit.self().getAllUsernames();
 		assertTrue("No users found!", users.size() > 0);
 		assertTrue("Admin not found", users.contains("admin"));
-		UserModel model = GitBlit.self().getUserModel("admin");
-		assertEquals("admin", model.toString());
-		assertTrue("Admin missing #admin role!", model.canAdmin);
-		model.canAdmin = false;
-		assertFalse("Admin should not have #admin!", model.canAdmin);
+		UserModel user = GitBlit.self().getUserModel("admin");
+		assertEquals("admin", user.toString());
+		assertTrue("Admin missing #admin role!", user.canAdmin);
+		user.canAdmin = false;
+		assertFalse("Admin should not have #admin!", user.canAdmin);
 		String repository = GitBlitSuite.getHelloworldRepository().getDirectory().getName();
 		RepositoryModel repositoryModel = GitBlit.self().getRepositoryModel(repository);
+		repositoryModel.accessRestriction = AccessRestrictionType.VIEW;
 		assertFalse("Admin can still access repository!",
-				model.canAccessRepository(repositoryModel));
-		model.addRepository(repository);
-		assertTrue("Admin can't access repository!", model.canAccessRepository(repositoryModel));
-		assertEquals(GitBlit.self().getRepositoryModel(model, "pretend"), null);
-		assertNotNull(GitBlit.self().getRepositoryModel(model, repository));
-		assertTrue(GitBlit.self().getRepositoryModels(model).size() > 0);
+				user.canView(repositoryModel));
+		user.addRepositoryPermission(repository);
+		assertTrue("Admin can't access repository!", user.canView(repositoryModel));
+		assertEquals(GitBlit.self().getRepositoryModel(user, "pretend"), null);
+		assertNotNull(GitBlit.self().getRepositoryModel(user, repository));
+		assertTrue(GitBlit.self().getRepositoryModels(user).size() > 0);
 	}
 
 	@Test

--
Gitblit v1.9.1