Jeroen Baten
2013-01-04 9af47c10c6a268877c1d232c8d71ee6df4f8a7ab
tests/com/gitblit/tests/GitServletTest.java
@@ -1,5 +1,6 @@
package com.gitblit.tests;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
@@ -32,6 +33,7 @@
import com.gitblit.Constants.AccessRestrictionType;
import com.gitblit.Constants.AuthorizationControl;
import com.gitblit.GitBlit;
import com.gitblit.Keys;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.JGitUtils;
@@ -86,6 +88,11 @@
   @Test
   public void testClone() throws Exception {
      GitBlitSuite.close(ticgitFolder);
      if (ticgitFolder.exists()) {
         FileUtils.delete(ticgitFolder, FileUtils.RECURSIVE | FileUtils.RETRY);
      }
      CloneCommand clone = Git.cloneRepository();
      clone.setURI(MessageFormat.format("{0}/git/ticgit.git", url));
      clone.setDirectory(ticgitFolder);
@@ -185,6 +192,20 @@
   @Test
   public void testAnonymousPush() throws Exception {
      GitBlitSuite.close(ticgitFolder);
      if (ticgitFolder.exists()) {
         FileUtils.delete(ticgitFolder, FileUtils.RECURSIVE | FileUtils.RETRY);
      }
      CloneCommand clone = Git.cloneRepository();
      clone.setURI(MessageFormat.format("{0}/git/ticgit.git", url));
      clone.setDirectory(ticgitFolder);
      clone.setBare(false);
      clone.setCloneAllBranches(true);
      clone.setCredentialsProvider(new UsernamePasswordCredentialsProvider(account, password));
      GitBlitSuite.close(clone.call());
      assertTrue(true);
      Git git = Git.open(ticgitFolder);
      File file = new File(ticgitFolder, "TODO");
      OutputStreamWriter os = new OutputStreamWriter(new FileOutputStream(file, true), Constants.CHARSET);
@@ -199,6 +220,11 @@
   @Test
   public void testSubfolderPush() throws Exception {
      GitBlitSuite.close(jgitFolder);
      if (jgitFolder.exists()) {
         FileUtils.delete(jgitFolder, FileUtils.RECURSIVE | FileUtils.RETRY);
      }
      CloneCommand clone = Git.cloneRepository();
      clone.setURI(MessageFormat.format("{0}/git/test/jgit.git", url));
      clone.setDirectory(jgitFolder);
@@ -216,6 +242,51 @@
      w.close();
      git.add().addFilepattern(file.getName()).call();
      git.commit().setMessage("test commit").call();
      git.push().setPushAll().call();
      GitBlitSuite.close(git);
   }
   @Test
   public void testPushToFrozenRepo() throws Exception {
      GitBlitSuite.close(jgitFolder);
      if (jgitFolder.exists()) {
         FileUtils.delete(jgitFolder, FileUtils.RECURSIVE | FileUtils.RETRY);
      }
      CloneCommand clone = Git.cloneRepository();
      clone.setURI(MessageFormat.format("{0}/git/test/jgit.git", url));
      clone.setDirectory(jgitFolder);
      clone.setBare(false);
      clone.setCloneAllBranches(true);
      clone.setCredentialsProvider(new UsernamePasswordCredentialsProvider(account, password));
      GitBlitSuite.close(clone.call());
      assertTrue(true);
      // freeze repo
      RepositoryModel model = GitBlit.self().getRepositoryModel("test/jgit.git");
      model.isFrozen = true;
      GitBlit.self().updateRepositoryModel(model.name, model, false);
      Git git = Git.open(jgitFolder);
      File file = new File(jgitFolder, "TODO");
      OutputStreamWriter os = new OutputStreamWriter(new FileOutputStream(file, true), Constants.CHARSET);
      BufferedWriter w = new BufferedWriter(os);
      w.write("// " + new Date().toString() + "\n");
      w.close();
      git.add().addFilepattern(file.getName()).call();
      git.commit().setMessage("test commit").call();
      try {
         git.push().setPushAll().call();
         assertTrue(false);
      } catch (Exception e) {
         assertTrue(e.getCause().getMessage().contains("access forbidden"));
      }
      // unfreeze repo
      model.isFrozen = false;
      GitBlit.self().updateRepositoryModel(model.name, model, false);
      git.push().setPushAll().call();
      GitBlitSuite.close(git);
   }
@@ -577,4 +648,112 @@
      GitBlit.self().deleteUser(user.username);
   }
   @Test
   public void testCreateOnPush() throws Exception {
      testCreateOnPush(false, false);
      testCreateOnPush(true, false);
      testCreateOnPush(false, true);
   }
   private void testCreateOnPush(boolean canCreate, boolean canAdmin) throws Exception {
      UserModel user = new UserModel("sampleuser");
      user.password = user.username;
      if (GitBlit.self().getUserModel(user.username) != null) {
         GitBlit.self().deleteUser(user.username);
      }
      user.canCreate = canCreate;
      user.canAdmin = canAdmin;
      GitBlit.self().updateUserModel(user.username, user, true);
      CredentialsProvider cp = new UsernamePasswordCredentialsProvider(user.username, user.password);
      // fork from original to a temporary bare repo
      File tmpFolder = File.createTempFile("gitblit", "").getParentFile();
      File createCheck = new File(tmpFolder, "ticgit.git");
      if (createCheck.exists()) {
         FileUtils.delete(createCheck, FileUtils.RECURSIVE);
      }
      File personalRepo = new File(GitBlitSuite.REPOSITORIES, MessageFormat.format("~{0}/ticgit.git", user.username));
      GitBlitSuite.close(personalRepo);
      if (personalRepo.exists()) {
         FileUtils.delete(personalRepo, FileUtils.RECURSIVE);
      }
      File projectRepo = new File(GitBlitSuite.REPOSITORIES, "project/ticgit.git");
      GitBlitSuite.close(projectRepo);
      if (projectRepo.exists()) {
         FileUtils.delete(projectRepo, FileUtils.RECURSIVE);
      }
      CloneCommand clone = Git.cloneRepository();
      clone.setURI(MessageFormat.format("{0}/git/ticgit.git", url));
      clone.setDirectory(createCheck);
      clone.setBare(true);
      clone.setCloneAllBranches(true);
      clone.setCredentialsProvider(cp);
      Git git = clone.call();
      GitBlitSuite.close(personalRepo);
      // add a personal repository remote and a project remote
      git.getRepository().getConfig().setString("remote", "user", "url", MessageFormat.format("{0}/git/~{1}/ticgit.git", url, user.username));
      git.getRepository().getConfig().setString("remote", "project", "url", MessageFormat.format("{0}/git/project/ticgit.git", url));
      git.getRepository().getConfig().save();
      // push to non-existent user repository
      try {
         Iterable<PushResult> results = git.push().setRemote("user").setPushAll().setCredentialsProvider(cp).call();
         for (PushResult result : results) {
            RemoteRefUpdate ref = result.getRemoteUpdate("refs/heads/master");
            Status status = ref.getStatus();
            assertTrue("User failed to create repository?! " + status.name(), Status.OK.equals(status));
         }
         assertTrue("User canAdmin:" + user.canAdmin + " canCreate:" + user.canCreate, user.canAdmin || user.canCreate);
         // confirm default personal repository permissions
         RepositoryModel model = GitBlit.self().getRepositoryModel(MessageFormat.format("~{0}/ticgit.git", user.username));
         assertEquals("Unexpected owner", user.username, model.owner);
         assertEquals("Unexpected authorization control", AuthorizationControl.NAMED, model.authorizationControl);
         assertEquals("Unexpected access restriction", AccessRestrictionType.VIEW, model.accessRestriction);
      } catch (GitAPIException e) {
         assertTrue(e.getMessage(), e.getMessage().contains("git-receive-pack not found"));
         assertFalse("User canAdmin:" + user.canAdmin + " canCreate:" + user.canCreate, user.canAdmin || user.canCreate);
      }
      // push to non-existent project repository
      try {
         Iterable<PushResult> results = git.push().setRemote("project").setPushAll().setCredentialsProvider(cp).call();
         GitBlitSuite.close(git);
         for (PushResult result : results) {
            RemoteRefUpdate ref = result.getRemoteUpdate("refs/heads/master");
            Status status = ref.getStatus();
            assertTrue("User failed to create repository?! " + status.name(), Status.OK.equals(status));
         }
         assertTrue("User canAdmin:" + user.canAdmin, user.canAdmin);
         // confirm default project repository permissions
         RepositoryModel model = GitBlit.self().getRepositoryModel("project/ticgit.git");
         assertEquals("Unexpected owner", user.username, model.owner);
         assertEquals("Unexpected authorization control", AuthorizationControl.fromName(GitBlit.getString(Keys.git.defaultAuthorizationControl, "NAMED")), model.authorizationControl);
         assertEquals("Unexpected access restriction", AccessRestrictionType.fromName(GitBlit.getString(Keys.git.defaultAccessRestriction, "NONE")), model.accessRestriction);
      } catch (GitAPIException e) {
         assertTrue(e.getMessage(), e.getMessage().contains("git-receive-pack not found"));
         assertFalse("User canAdmin:" + user.canAdmin, user.canAdmin);
      }
      GitBlitSuite.close(git);
      GitBlit.self().deleteUser(user.username);
   }
}