Gitblit devel
blame | history | patch | commit | commitdiff | ignore whitespace
James Moger
2013-01-11 5316d20e861640867d10405b25cfe75aeca0a34c 
 tests/com/gitblit/tests/GitServletTest.java


@@ -1,14 +1,17 @@


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.BufferedWriter;



import java.io.File;



import java.io.FileOutputStream;



import java.io.IOException;



import java.io.OutputStreamWriter;



import java.text.MessageFormat;



import java.util.Date;



import java.util.List;



import java.util.concurrent.atomic.AtomicBoolean;







import org.eclipse.jgit.api.CloneCommand;



@@ -17,6 +20,7 @@


import org.eclipse.jgit.api.errors.GitAPIException;



import org.eclipse.jgit.lib.Constants;



import org.eclipse.jgit.revwalk.RevCommit;



import org.eclipse.jgit.storage.file.FileRepository;



import org.eclipse.jgit.transport.CredentialsProvider;



import org.eclipse.jgit.transport.PushResult;



import org.eclipse.jgit.transport.RefSpec;



@@ -32,9 +36,12 @@


import com.gitblit.Constants.AccessRestrictionType;



import com.gitblit.Constants.AuthorizationControl;



import com.gitblit.GitBlit;



import com.gitblit.Keys;



import com.gitblit.models.PushLogEntry;



import com.gitblit.models.RepositoryModel;



import com.gitblit.models.UserModel;



import com.gitblit.utils.JGitUtils;



import com.gitblit.utils.PushLogUtils;







public class GitServletTest {







@@ -86,6 +93,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 +197,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 +225,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 +247,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 +653,122 @@






      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);



   }



	


   @Test



   public void testPushLog() throws IOException {



      String name = "refchecks/ticgit.git";



      File refChecks = new File(GitBlitSuite.REPOSITORIES, name);



      FileRepository repository = new FileRepository(refChecks);



      List<PushLogEntry> pushes = PushLogUtils.getPushLog(name, repository);



      GitBlitSuite.close(repository);



      assertTrue("Repository has an empty push log!", pushes.size() > 0);



   }



}