Draft project pages, project metadata, and RSS feeds

James Moger

2012-09-07 13a3f5bc3e2d25fc76850f86070dc34efe60d77a

 tests/com/gitblit/tests/GitServletTest.java

@@ -13,6 +13,7 @@


import org.eclipse.jgit.api.CloneCommand;

import org.eclipse.jgit.api.Git;

import org.eclipse.jgit.lib.Constants;

import org.eclipse.jgit.transport.UsernamePasswordCredentialsProvider;

import org.eclipse.jgit.util.FileUtils;

import org.junit.AfterClass;

@@ -20,14 +21,20 @@
import org.junit.Test;



import com.gitblit.Constants.AccessRestrictionType;

import com.gitblit.Constants.AuthorizationControl;

import com.gitblit.GitBlit;

import com.gitblit.models.RepositoryModel;

import com.gitblit.models.UserModel;



public class GitServletTest {



   File folder = new File(GitBlitSuite.REPOSITORIES, "working/ticgit");

   static File ticgitFolder = new File(GitBlitSuite.REPOSITORIES, "working/ticgit");

   

   File jgitFolder = new File(GitBlitSuite.REPOSITORIES, "working/jgit");

   static File ticgit2Folder = new File(GitBlitSuite.REPOSITORIES, "working/ticgit2");



   static File jgitFolder = new File(GitBlitSuite.REPOSITORIES, "working/jgit");

	
   static File jgit2Folder = new File(GitBlitSuite.REPOSITORIES, "working/jgit2");



   String url = GitBlitSuite.url;

   String account = GitBlitSuite.account;

@@ -44,21 +51,34 @@
   public static void stopGitblit() throws Exception {

      if (started.get()) {

         GitBlitSuite.stopGitblit();

         deleteWorkingFolders();

      }

   }

	
   public static void deleteWorkingFolders() throws Exception {

      if (ticgitFolder.exists()) {

         FileUtils.delete(ticgitFolder, FileUtils.RECURSIVE);

      }

      if (ticgit2Folder.exists()) {

         FileUtils.delete(ticgit2Folder, FileUtils.RECURSIVE);

      }

      if (jgitFolder.exists()) {

         FileUtils.delete(jgitFolder, FileUtils.RECURSIVE);

      }

      if (jgit2Folder.exists()) {

         FileUtils.delete(jgit2Folder, FileUtils.RECURSIVE);

      }

   }



   @Test

   public void testClone() throws Exception {

      if (folder.exists()) {

         FileUtils.delete(folder, FileUtils.RECURSIVE);

      }

      CloneCommand clone = Git.cloneRepository();

      clone.setURI(MessageFormat.format("{0}/git/ticgit.git", url));

      clone.setDirectory(folder);

      clone.setDirectory(ticgitFolder);

      clone.setBare(false);

      clone.setCloneAllBranches(true);

      clone.setCredentialsProvider(new UsernamePasswordCredentialsProvider(account, password));

      clone.call();

      close(clone.call());		
      assertTrue(true);

   }



@@ -69,20 +89,16 @@
      model.accessRestriction = AccessRestrictionType.CLONE;

      GitBlit.self().updateRepositoryModel(model.name, model, false);



      // delete any existing working folder

      File folder = new File(GitBlitSuite.REPOSITORIES, "working/gitblit");

      if (folder.exists()) {

         FileUtils.delete(folder, FileUtils.RECURSIVE);

      }

      // delete any existing working folder		
      boolean cloned = false;

      try {

         CloneCommand clone = Git.cloneRepository();

         clone.setURI(MessageFormat.format("{0}/git/ticgit.git", url));

         clone.setDirectory(folder);

         clone.setDirectory(ticgit2Folder);

         clone.setBare(false);

         clone.setCloneAllBranches(true);

         clone.setCredentialsProvider(new UsernamePasswordCredentialsProvider("bogus", "bogus"));

         clone.call();

         close(clone.call());

         cloned = true;

      } catch (Exception e) {

         // swallow the exception which we expect

@@ -94,48 +110,135 @@


      assertFalse("Bogus login cloned a repository?!", cloned);

   }

	
   @Test

   public void testUnauthorizedLoginClone() throws Exception {

      // restrict repository access

      RepositoryModel model = GitBlit.self().getRepositoryModel("ticgit.git");

      model.accessRestriction = AccessRestrictionType.CLONE;

      model.authorizationControl = AuthorizationControl.NAMED;

      UserModel user = new UserModel("james");

      user.password = "james";

      GitBlit.self().updateUserModel(user.username, user, true);

      GitBlit.self().updateRepositoryModel(model.name, model, false);



      FileUtils.delete(ticgit2Folder, FileUtils.RECURSIVE);

		
      // delete any existing working folder		
      boolean cloned = false;

      try {

         CloneCommand clone = Git.cloneRepository();

         clone.setURI(MessageFormat.format("{0}/git/ticgit.git", url));

         clone.setDirectory(ticgit2Folder);

         clone.setBare(false);

         clone.setCloneAllBranches(true);

         clone.setCredentialsProvider(new UsernamePasswordCredentialsProvider(user.username, user.password));

         close(clone.call());

         cloned = true;

      } catch (Exception e) {

         // swallow the exception which we expect

      }



      assertFalse("Unauthorized login cloned a repository?!", cloned);



      FileUtils.delete(ticgit2Folder, FileUtils.RECURSIVE);

		
      // switch to authenticated

      model.authorizationControl = AuthorizationControl.AUTHENTICATED;

      GitBlit.self().updateRepositoryModel(model.name, model, false);

		
      // try clone again

      cloned = false;

      CloneCommand clone = Git.cloneRepository();

      clone.setURI(MessageFormat.format("{0}/git/ticgit.git", url));

      clone.setDirectory(ticgit2Folder);

      clone.setBare(false);

      clone.setCloneAllBranches(true);

      clone.setCredentialsProvider(new UsernamePasswordCredentialsProvider(user.username, user.password));

      close(clone.call());

      cloned = true;



      assertTrue("Authenticated login could not clone!", cloned);

		
      FileUtils.delete(ticgit2Folder, FileUtils.RECURSIVE);

		
      // restore anonymous repository access

      model.accessRestriction = AccessRestrictionType.NONE;

      model.authorizationControl = AuthorizationControl.NAMED;

      GitBlit.self().updateRepositoryModel(model.name, model, false);

      GitBlit.self().deleteUser(user.username);

   }



   @Test

   public void testAnonymousPush() throws Exception {

      Git git = Git.open(folder);

      File file = new File(folder, "TODO");

      for (int i = 0; i < 3; i++) {

         OutputStreamWriter os = new OutputStreamWriter(new FileOutputStream(file, true));

         BufferedWriter w = new BufferedWriter(os);

         w.write("// " + new Date().toString() + "\n");

         w.close();

         git.add().addFilepattern(file.getName()).call();

         git.commit().setMessage("test commit #" + (i + 1)).call();

      }

      Git git = Git.open(ticgitFolder);

      File file = new File(ticgitFolder, "TODO");

      OutputStreamWriter os = new OutputStreamWriter(new FileOutputStream(file, true), Constants.CHARSET);

      BufferedWriter w = new BufferedWriter(os);

      w.write("// hellol中文 " + new Date().toString() + "\n");

      w.close();

      git.add().addFilepattern(file.getName()).call();

      git.commit().setMessage("test commit").call();

      git.push().setPushAll().call();

      git.getRepository().close();

      close(git);

   }

	


   @Test

   public void testSubfolderPush() throws Exception {

      if (jgitFolder.exists()) {

         FileUtils.delete(jgitFolder, FileUtils.RECURSIVE);

      }

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

      clone.call();

      close(clone.call());

      assertTrue(true);

		


      Git git = Git.open(jgitFolder);

      File file = new File(jgitFolder, "TODO");

      for (int i = 0; i < 3; i++) {

         OutputStreamWriter os = new OutputStreamWriter(new FileOutputStream(file, true));

         BufferedWriter w = new BufferedWriter(os);

         w.write("// " + new Date().toString() + "\n");

         w.close();

         git.add().addFilepattern(file.getName()).call();

         git.commit().setMessage("test commit #" + (i + 1)).call();

      }

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

      git.push().setPushAll().call();

      git.getRepository().close();

      close(git);

   }

	
   @Test

   public void testPushToNonBareRepository() throws Exception {

      CloneCommand clone = Git.cloneRepository();

      clone.setURI(MessageFormat.format("{0}/git/working/jgit", url));

      clone.setDirectory(jgit2Folder);

      clone.setBare(false);

      clone.setCloneAllBranches(true);

      clone.setCredentialsProvider(new UsernamePasswordCredentialsProvider(account, password));

      close(clone.call());

      assertTrue(true);



      Git git = Git.open(jgit2Folder);

      File file = new File(jgit2Folder, "NONBARE");

      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 followed by push to non-bare repository").call();

      try {

         git.push().setPushAll().call();

         assertTrue(false);

      } catch (Exception e) {

         assertTrue(e.getCause().getMessage().contains("git-receive-pack not permitted"));

      }

      close(git);

   }

	
   private void close(Git git) {

      // really close the repository

      // decrement the use counter to 0

      for (int i = 0; i < 2; i++) {

         git.getRepository().close();

      }

   }

}