From 31abc26dd0354bc2dafe27c011c2e54934a89486 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Sun, 02 Oct 2011 15:37:24 -0400
Subject: [PATCH] Fairly complete json rpc interface to view/control Gitblit data objects.

---
 tests/com/gitblit/tests/RpcTests.java |  178 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 169 insertions(+), 9 deletions(-)

diff --git a/tests/com/gitblit/tests/RpcTests.java b/tests/com/gitblit/tests/RpcTests.java
index e140fb4..cb98a56 100644
--- a/tests/com/gitblit/tests/RpcTests.java
+++ b/tests/com/gitblit/tests/RpcTests.java
@@ -16,25 +16,185 @@
 package com.gitblit.tests;
 
 import java.io.IOException;
+import java.util.List;
 import java.util.Map;
 
 import junit.framework.TestCase;
 
+import com.gitblit.Constants.AccessRestrictionType;
+import com.gitblit.GitBlitException.UnauthorizedException;
+import com.gitblit.models.FederationModel;
+import com.gitblit.models.FederationProposal;
+import com.gitblit.models.FederationSet;
 import com.gitblit.models.RepositoryModel;
+import com.gitblit.models.UserModel;
 import com.gitblit.utils.RpcUtils;
 
+/**
+ * Tests all the rpc client utility methods, the rpc filter and rpc servlet.
+ * 
+ * @author James Moger
+ * 
+ */
 public class RpcTests extends TestCase {
 
-	public void testListRepositories() throws Exception {
-		Map<String, RepositoryModel> map = null;
-		try {
-			map = RpcUtils.getRepositories("https://localhost:8443");
-		} catch (IOException e) {
-			if (!e.getMessage().contains("403")) {
-				throw e;
-			}
-		}
+	String url = "https://localhost:8443";
+	String account = "admin";
+	String password = "admin";
+
+	public void testListRepositories() throws IOException {
+		Map<String, RepositoryModel> map = RpcUtils.getRepositories(url, null, null);
 		assertTrue("Repository list is null!", map != null);
 		assertTrue("Repository list is empty!", map.size() > 0);
 	}
+
+	public void testListUsers() throws IOException {
+		List<UserModel> list = null;
+		try {
+			list = RpcUtils.getUsers(url, null, null);
+		} catch (UnauthorizedException e) {
+		}
+		assertTrue("Server allows anyone to admin!", list == null);
+
+		list = RpcUtils.getUsers(url, "admin", "admin".toCharArray());
+		assertTrue("User list is empty!", list.size() > 0);
+	}
+
+	public void testUserAdministration() throws IOException {
+		UserModel user = new UserModel("garbage");
+		user.canAdmin = true;
+		user.password = "whocares";
+
+		// create
+		assertTrue("Failed to create user!",
+				RpcUtils.createUser(user, url, account, password.toCharArray()));
+
+		UserModel retrievedUser = findUser(user.username);
+		assertTrue("Failed to find " + user.username, retrievedUser != null);
+		assertTrue("Retrieved user can not administer Gitblit", retrievedUser.canAdmin);
+
+		// rename and toggle admin permission
+		String originalName = user.username;
+		user.username = "garbage2";
+		user.canAdmin = false;
+		assertTrue("Failed to update user!",
+				RpcUtils.updateUser(originalName, user, url, account, password.toCharArray()));
+
+		retrievedUser = findUser(user.username);
+		assertTrue("Failed to find " + user.username, retrievedUser != null);
+		assertTrue("Retrieved user did not update", !retrievedUser.canAdmin);
+
+		// delete
+		assertTrue("Failed to delete " + user.username,
+				RpcUtils.deleteUser(retrievedUser, url, account, password.toCharArray()));
+
+		retrievedUser = findUser(user.username);
+		assertTrue("Failed to delete " + user.username, retrievedUser == null);
+	}
+
+	private UserModel findUser(String name) throws IOException {
+		List<UserModel> users = RpcUtils.getUsers(url, account, password.toCharArray());
+		UserModel retrievedUser = null;
+		for (UserModel model : users) {
+			if (model.username.equalsIgnoreCase(name)) {
+				retrievedUser = model;
+				break;
+			}
+		}
+		return retrievedUser;
+	}
+
+	public void testRepositoryAdministration() throws IOException {
+		RepositoryModel model = new RepositoryModel();
+		model.name = "garbagerepo.git";
+		model.description = "created by RpcUtils";
+		model.owner = "garbage";
+		model.accessRestriction = AccessRestrictionType.VIEW;
+
+		// create
+		assertTrue("Failed to create repository!",
+				RpcUtils.createRepository(model, url, account, password.toCharArray()));
+
+		RepositoryModel retrievedRepository = findRepository(model.name);
+		assertTrue("Failed to find " + model.name, retrievedRepository != null);
+		assertTrue("Access retriction type is wrong",
+				AccessRestrictionType.VIEW.equals(retrievedRepository.accessRestriction));
+
+		// rename and change access restriciton
+		String originalName = model.name;
+		model.name = "garbagerepo2.git";
+		model.accessRestriction = AccessRestrictionType.PUSH;
+		assertTrue("Failed to update repository!", RpcUtils.updateRepository(originalName, model,
+				url, account, password.toCharArray()));
+
+		retrievedRepository = findRepository(model.name);
+		assertTrue("Failed to find " + model.name, retrievedRepository != null);
+		assertTrue("Access retriction type is wrong",
+				AccessRestrictionType.PUSH.equals(retrievedRepository.accessRestriction));
+
+		// memberships
+		String testMember = "justadded";
+		List<String> members = RpcUtils.getRepositoryMembers(retrievedRepository, url, account,
+				password.toCharArray());
+		assertTrue("Membership roster is not empty!", members.size() == 0);
+		members.add(testMember);
+		assertTrue(
+				"Failed to set memberships!",
+				RpcUtils.setRepositoryMembers(retrievedRepository, members, url, account,
+						password.toCharArray()));
+		members = RpcUtils.getRepositoryMembers(retrievedRepository, url, account,
+				password.toCharArray());
+		boolean foundMember = false;
+		for (String member : members) {
+			if (member.equalsIgnoreCase(testMember)) {
+				foundMember = true;
+				break;
+			}
+		}
+		assertTrue("Failed to find member!", foundMember);
+
+		// delete
+		assertTrue("Failed to delete " + model.name, RpcUtils.deleteRepository(retrievedRepository,
+				url, account, password.toCharArray()));
+
+		retrievedRepository = findRepository(model.name);
+		assertTrue("Failed to delete " + model.name, retrievedRepository == null);
+	}
+
+	private RepositoryModel findRepository(String name) throws IOException {
+		Map<String, RepositoryModel> repositories = RpcUtils.getRepositories(url, account,
+				password.toCharArray());
+		RepositoryModel retrievedRepository = null;
+		for (RepositoryModel model : repositories.values()) {
+			if (model.name.equalsIgnoreCase(name)) {
+				retrievedRepository = model;
+				break;
+			}
+		}
+		return retrievedRepository;
+	}
+
+	public void testFederationRegistrations() throws Exception {
+		List<FederationModel> registrations = RpcUtils.getFederationRegistrations(url, account,
+				password.toCharArray());
+		assertTrue("No federation registrations wre retrieved!", registrations.size() > 0);
+	}
+
+	public void testFederationResultRegistrations() throws Exception {
+		List<FederationModel> registrations = RpcUtils.getFederationResultRegistrations(url,
+				account, password.toCharArray());
+		assertTrue("No federation result registrations were retrieved!", registrations.size() > 0);
+	}
+	
+	public void testFederationProposals() throws Exception {
+		List<FederationProposal> proposals = RpcUtils.getFederationProposals(url,
+				account, password.toCharArray());
+		assertTrue("No federation proposals were retrieved!", proposals.size() > 0);
+	}
+	
+	public void testFederationSets() throws Exception {
+		List<FederationSet> sets = RpcUtils.getFederationSets(url,
+				account, password.toCharArray());
+		assertTrue("No federation sets were retrieved!", sets.size() > 0);
+	}
 }

--
Gitblit v1.9.1