| | |
|---|
| | | */
|
|---|
| | | package com.gitblit.tests;
|
|---|
| | |
|
|---|
| | | import static org.junit.Assert.assertEquals;
|
|---|
| | | import static org.junit.Assert.assertNotNull;
|
|---|
| | | import static org.junit.Assert.assertNull;
|
|---|
| | | import static org.junit.Assert.assertTrue;
|
|---|
| | |
|
|---|
| | | import java.io.IOException;
|
|---|
| | | import java.util.ArrayList;
|
|---|
| | | import java.util.Collection;
|
|---|
| | | import java.util.HashMap;
|
|---|
| | | import java.util.List;
|
|---|
| | | import java.util.Map;
|
|---|
| | | import java.util.concurrent.atomic.AtomicBoolean;
|
|---|
| | |
|
|---|
| | | import junit.framework.TestCase;
|
|---|
| | | import org.junit.AfterClass;
|
|---|
| | | import org.junit.BeforeClass;
|
|---|
| | | import org.junit.Test;
|
|---|
| | |
|
|---|
| | | import com.gitblit.Constants.AccessRestrictionType;
|
|---|
| | | import com.gitblit.GitBlitException.UnauthorizedException;
|
|---|
| | | import com.gitblit.Keys;
|
|---|
| | | import com.gitblit.RpcServlet;
|
|---|
| | | import com.gitblit.models.FederationModel;
|
|---|
| | | import com.gitblit.models.FederationProposal;
|
|---|
| | | import com.gitblit.models.FederationSet;
|
|---|
| | | import com.gitblit.models.RepositoryModel;
|
|---|
| | | import com.gitblit.models.ServerSettings;
|
|---|
| | | import com.gitblit.models.ServerStatus;
|
|---|
| | | import com.gitblit.models.TeamModel;
|
|---|
| | | import com.gitblit.models.UserModel;
|
|---|
| | | import com.gitblit.utils.RpcUtils;
|
|---|
| | |
|
|---|
| | | public class RpcTests extends TestCase {
|
|---|
| | | /**
|
|---|
| | | * Tests all the rpc client utility methods, the rpc filter and rpc servlet.
|
|---|
| | | * |
|---|
| | | * @author James Moger
|
|---|
| | | * |
|---|
| | | */
|
|---|
| | | public class RpcTests {
|
|---|
| | |
|
|---|
| | | public void testListRepositories() throws Exception {
|
|---|
| | | Map<String, RepositoryModel> map = null;
|
|---|
| | | String url = GitBlitSuite.url;
|
|---|
| | | String account = GitBlitSuite.account;
|
|---|
| | | String password = GitBlitSuite.password;
|
|---|
| | |
|
|---|
| | | private static final AtomicBoolean started = new AtomicBoolean(false);
|
|---|
| | |
|
|---|
| | | @BeforeClass
|
|---|
| | | public static void startGitblit() throws Exception {
|
|---|
| | | started.set(GitBlitSuite.startGitblit());
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @AfterClass
|
|---|
| | | public static void stopGitblit() throws Exception {
|
|---|
| | | if (started.get()) {
|
|---|
| | | GitBlitSuite.stopGitblit();
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testGetProtocolVersion() throws IOException {
|
|---|
| | | int protocol = RpcUtils.getProtocolVersion(url, null, null);
|
|---|
| | | assertEquals(RpcServlet.PROTOCOL_VERSION, protocol);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testListRepositories() throws IOException {
|
|---|
| | | Map<String, RepositoryModel> map = RpcUtils.getRepositories(url, null, null);
|
|---|
| | | assertNotNull("Repository list is null!", map);
|
|---|
| | | assertTrue("Repository list is empty!", map.size() > 0);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testListUsers() throws IOException {
|
|---|
| | | List<UserModel> list = null;
|
|---|
| | | try {
|
|---|
| | | map = RpcUtils.getRepositories("https://localhost:8443");
|
|---|
| | | } catch (IOException e) {
|
|---|
| | | if (!e.getMessage().contains("403")) {
|
|---|
| | | throw e;
|
|---|
| | | list = RpcUtils.getUsers(url, null, null);
|
|---|
| | | } catch (UnauthorizedException e) {
|
|---|
| | | }
|
|---|
| | | assertNull("Server allows anyone to admin!", list);
|
|---|
| | |
|
|---|
| | | list = RpcUtils.getUsers(url, "admin", "admin".toCharArray());
|
|---|
| | | assertTrue("User list is empty!", list.size() > 0);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testListTeams() throws IOException {
|
|---|
| | | List<TeamModel> list = null;
|
|---|
| | | try {
|
|---|
| | | list = RpcUtils.getTeams(url, null, null);
|
|---|
| | | } catch (UnauthorizedException e) {
|
|---|
| | | }
|
|---|
| | | assertNull("Server allows anyone to admin!", list);
|
|---|
| | |
|
|---|
| | | list = RpcUtils.getTeams(url, "admin", "admin".toCharArray());
|
|---|
| | | assertTrue("Team list is empty!", list.size() > 0);
|
|---|
| | | assertEquals("admins", list.get(0).name);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | 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);
|
|---|
| | | assertNotNull("Failed to find " + user.username, retrievedUser);
|
|---|
| | | 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);
|
|---|
| | | assertNotNull("Failed to find " + user.username, retrievedUser);
|
|---|
| | | 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);
|
|---|
| | | assertNull("Failed to delete " + user.username, retrievedUser);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | 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;
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | | assertTrue("Repository list is null!", map != null);
|
|---|
| | | assertTrue("Repository list is empty!", map.size() > 0);
|
|---|
| | | return retrievedUser;
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | 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);
|
|---|
| | | assertNotNull("Failed to find " + model.name, retrievedRepository);
|
|---|
| | | assertEquals(AccessRestrictionType.VIEW, 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);
|
|---|
| | | assertNotNull("Failed to find " + model.name, retrievedRepository);
|
|---|
| | | assertTrue("Access retriction type is wrong",
|
|---|
| | | AccessRestrictionType.PUSH.equals(retrievedRepository.accessRestriction));
|
|---|
| | |
|
|---|
| | | // memberships
|
|---|
| | | UserModel testMember = new UserModel("justadded");
|
|---|
| | | assertTrue(RpcUtils.createUser(testMember, url, account, password.toCharArray()));
|
|---|
| | |
|
|---|
| | | List<String> members = RpcUtils.getRepositoryMembers(retrievedRepository, url, account,
|
|---|
| | | password.toCharArray());
|
|---|
| | | assertEquals("Membership roster is not empty!", 0, members.size());
|
|---|
| | | members.add(testMember.username);
|
|---|
| | | 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.username)) {
|
|---|
| | | 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);
|
|---|
| | | assertNull("Failed to delete " + model.name, retrievedRepository);
|
|---|
| | |
|
|---|
| | | for (UserModel u : RpcUtils.getUsers(url, account, password.toCharArray())) {
|
|---|
| | | if (u.username.equals(testMember.username)) {
|
|---|
| | | assertTrue(RpcUtils.deleteUser(u, url, account, password.toCharArray()));
|
|---|
| | | break;
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | 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;
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testTeamAdministration() throws IOException {
|
|---|
| | | List<TeamModel> teams = RpcUtils.getTeams(url, account, password.toCharArray());
|
|---|
| | | assertEquals(1, teams.size());
|
|---|
| | | |
|---|
| | | // Create the A-Team
|
|---|
| | | TeamModel aTeam = new TeamModel("A-Team");
|
|---|
| | | aTeam.users.add("admin");
|
|---|
| | | aTeam.repositories.add("helloworld.git");
|
|---|
| | | assertTrue(RpcUtils.createTeam(aTeam, url, account, password.toCharArray()));
|
|---|
| | |
|
|---|
| | | aTeam = null;
|
|---|
| | | teams = RpcUtils.getTeams(url, account, password.toCharArray());
|
|---|
| | | assertEquals(2, teams.size());
|
|---|
| | | for (TeamModel team : teams) {
|
|---|
| | | if (team.name.equals("A-Team")) {
|
|---|
| | | aTeam = team;
|
|---|
| | | break;
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | | assertNotNull(aTeam);
|
|---|
| | | assertTrue(aTeam.hasUser("admin"));
|
|---|
| | | assertTrue(aTeam.hasRepository("helloworld.git"));
|
|---|
| | |
|
|---|
| | | RepositoryModel helloworld = null;
|
|---|
| | | Map<String, RepositoryModel> repositories = RpcUtils.getRepositories(url, account,
|
|---|
| | | password.toCharArray());
|
|---|
| | | for (RepositoryModel repository : repositories.values()) {
|
|---|
| | | if (repository.name.equals("helloworld.git")) {
|
|---|
| | | helloworld = repository;
|
|---|
| | | break;
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | | assertNotNull(helloworld);
|
|---|
| | | |
|---|
| | | // Confirm that we have added the team
|
|---|
| | | List<String> helloworldTeams = RpcUtils.getRepositoryTeams(helloworld, url, account,
|
|---|
| | | password.toCharArray());
|
|---|
| | | assertEquals(1, helloworldTeams.size());
|
|---|
| | | assertTrue(helloworldTeams.contains(aTeam.name));
|
|---|
| | |
|
|---|
| | | // set no teams
|
|---|
| | | assertTrue(RpcUtils.setRepositoryTeams(helloworld, new ArrayList<String>(), url, account,
|
|---|
| | | password.toCharArray()));
|
|---|
| | | helloworldTeams = RpcUtils.getRepositoryTeams(helloworld, url, account,
|
|---|
| | | password.toCharArray());
|
|---|
| | | assertEquals(0, helloworldTeams.size());
|
|---|
| | | |
|---|
| | | // delete the A-Team
|
|---|
| | | assertTrue(RpcUtils.deleteTeam(aTeam, url, account, password.toCharArray()));
|
|---|
| | |
|
|---|
| | | teams = RpcUtils.getTeams(url, account, password.toCharArray());
|
|---|
| | | assertEquals(1, teams.size());
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testFederationRegistrations() throws Exception {
|
|---|
| | | List<FederationModel> registrations = RpcUtils.getFederationRegistrations(url, account,
|
|---|
| | | password.toCharArray());
|
|---|
| | | assertTrue("No federation registrations were retrieved!", registrations.size() >= 0);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testFederationResultRegistrations() throws Exception {
|
|---|
| | | List<FederationModel> registrations = RpcUtils.getFederationResultRegistrations(url,
|
|---|
| | | account, password.toCharArray());
|
|---|
| | | assertTrue("No federation result registrations were retrieved!", registrations.size() >= 0);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testFederationProposals() throws Exception {
|
|---|
| | | List<FederationProposal> proposals = RpcUtils.getFederationProposals(url, account,
|
|---|
| | | password.toCharArray());
|
|---|
| | | assertTrue("No federation proposals were retrieved!", proposals.size() >= 0);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testFederationSets() throws Exception {
|
|---|
| | | List<FederationSet> sets = RpcUtils.getFederationSets(url, account, password.toCharArray());
|
|---|
| | | assertTrue("No federation sets were retrieved!", sets.size() >= 0);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testSettings() throws Exception {
|
|---|
| | | ServerSettings settings = RpcUtils.getSettings(url, account, password.toCharArray());
|
|---|
| | | assertNotNull("No settings were retrieved!", settings);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testServerStatus() throws Exception {
|
|---|
| | | ServerStatus status = RpcUtils.getStatus(url, account, password.toCharArray());
|
|---|
| | | assertNotNull("No status was retrieved!", status);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testUpdateSettings() throws Exception {
|
|---|
| | | Map<String, String> updated = new HashMap<String, String>();
|
|---|
| | |
|
|---|
| | | // grab current setting
|
|---|
| | | ServerSettings settings = RpcUtils.getSettings(url, account, password.toCharArray());
|
|---|
| | | boolean showSizes = settings.get(Keys.web.showRepositorySizes).getBoolean(true);
|
|---|
| | | showSizes = !showSizes;
|
|---|
| | |
|
|---|
| | | // update setting
|
|---|
| | | updated.put(Keys.web.showRepositorySizes, String.valueOf(showSizes));
|
|---|
| | | boolean success = RpcUtils.updateSettings(updated, url, account, password.toCharArray());
|
|---|
| | | assertTrue("Failed to update server settings", success);
|
|---|
| | |
|
|---|
| | | // confirm setting change
|
|---|
| | | settings = RpcUtils.getSettings(url, account, password.toCharArray());
|
|---|
| | | boolean newValue = settings.get(Keys.web.showRepositorySizes).getBoolean(false);
|
|---|
| | | assertEquals(newValue, showSizes);
|
|---|
| | |
|
|---|
| | | // restore setting
|
|---|
| | | newValue = !newValue;
|
|---|
| | | updated.put(Keys.web.showRepositorySizes, String.valueOf(newValue));
|
|---|
| | | success = RpcUtils.updateSettings(updated, url, account, password.toCharArray());
|
|---|
| | | assertTrue("Failed to update server settings", success);
|
|---|
| | | settings = RpcUtils.getSettings(url, account, password.toCharArray());
|
|---|
| | | showSizes = settings.get(Keys.web.showRepositorySizes).getBoolean(true);
|
|---|
| | | assertEquals(newValue, showSizes);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | @Test
|
|---|
| | | public void testBranches() throws Exception {
|
|---|
| | | Map<String, Collection<String>> branches = RpcUtils.getBranches(url, account,
|
|---|
| | | password.toCharArray());
|
|---|
| | | assertNotNull(branches);
|
|---|
| | | assertTrue(branches.size() > 0);
|
|---|
| | | }
|
|---|
| | | }
|
|---|
