From cb285cbfddfc0b633d6b8cdb4dc0d2bd2b8b51ef Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Thu, 05 Jan 2012 17:34:05 -0500 Subject: [PATCH] Fixed bug in receive hook for repositories in subfolders --- tests/com/gitblit/tests/GitServletTest.java | 141 ++++++++++++++++++++++++++++++++++------------ 1 files changed, 104 insertions(+), 37 deletions(-) diff --git a/tests/com/gitblit/tests/GitServletTest.java b/tests/com/gitblit/tests/GitServletTest.java index 0ede7ca..a4c5e88 100644 --- a/tests/com/gitblit/tests/GitServletTest.java +++ b/tests/com/gitblit/tests/GitServletTest.java @@ -1,73 +1,140 @@ package com.gitblit.tests; +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.OutputStreamWriter; import java.text.MessageFormat; import java.util.Date; -import java.util.concurrent.Executors; - -import junit.framework.TestCase; +import java.util.concurrent.atomic.AtomicBoolean; import org.eclipse.jgit.api.CloneCommand; import org.eclipse.jgit.api.Git; +import org.eclipse.jgit.transport.UsernamePasswordCredentialsProvider; import org.eclipse.jgit.util.FileUtils; +import org.junit.AfterClass; +import org.junit.BeforeClass; +import org.junit.Test; -import com.gitblit.GitBlitServer; +import com.gitblit.Constants.AccessRestrictionType; +import com.gitblit.GitBlit; +import com.gitblit.models.RepositoryModel; -public class GitServletTest extends TestCase { +public class GitServletTest { File folder = new File(GitBlitSuite.REPOSITORIES, "working/ticgit"); + + File jgitFolder = new File(GitBlitSuite.REPOSITORIES, "working/jgit"); - int port = 8180; + String url = GitBlitSuite.url; + String account = GitBlitSuite.account; + String password = GitBlitSuite.password; - int shutdownPort = 8181; + private static final AtomicBoolean started = new AtomicBoolean(false); - @Override - protected void setUp() throws Exception { - // Start a Gitblit instance - Executors.newSingleThreadExecutor().execute(new Runnable() { - public void run() { - GitBlitServer.main("--httpPort", "" + port, "--httpsPort", "0", "--shutdownPort", - "" + shutdownPort, "--repositoriesFolder", - "\"" + GitBlitSuite.REPOSITORIES.getAbsolutePath() + "\"", "--userService", - "distrib/users.properties"); - } - }); - - // Wait a few seconds for it to be running - Thread.sleep(2500); + @BeforeClass + public static void startGitblit() throws Exception { + started.set(GitBlitSuite.startGitblit()); } - @Override - protected void tearDown() throws Exception { - // Stop Gitblit - GitBlitServer.main("--stop", "--shutdownPort", "" + shutdownPort); - - // Wait a few seconds for it to be running - Thread.sleep(2500); + @AfterClass + public static void stopGitblit() throws Exception { + if (started.get()) { + GitBlitSuite.stopGitblit(); + } } + @Test public void testClone() throws Exception { - FileUtils.delete(folder, FileUtils.RECURSIVE); + if (folder.exists()) { + FileUtils.delete(folder, FileUtils.RECURSIVE); + } CloneCommand clone = Git.cloneRepository(); - clone.setURI(MessageFormat.format("http://localhost:{0,number,#}/git/ticgit.git", port)); + clone.setURI(MessageFormat.format("{0}/git/ticgit.git", url)); clone.setDirectory(folder); clone.setBare(false); clone.setCloneAllBranches(true); + clone.setCredentialsProvider(new UsernamePasswordCredentialsProvider(account, password)); clone.call(); + assertTrue(true); } - public void testAnonymousCommit() throws Exception { + @Test + public void testBogusLoginClone() throws Exception { + // restrict repository access + RepositoryModel model = GitBlit.self().getRepositoryModel("ticgit.git"); + 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); + } + boolean cloned = false; + try { + CloneCommand clone = Git.cloneRepository(); + clone.setURI(MessageFormat.format("{0}/git/ticgit.git", url)); + clone.setDirectory(folder); + clone.setBare(false); + clone.setCloneAllBranches(true); + clone.setCredentialsProvider(new UsernamePasswordCredentialsProvider("bogus", "bogus")); + clone.call(); + cloned = true; + } catch (Exception e) { + // swallow the exception which we expect + } + + // restore anonymous repository access + model.accessRestriction = AccessRestrictionType.NONE; + GitBlit.self().updateRepositoryModel(model.name, model, false); + + assertFalse("Bogus login cloned a repository?!", cloned); + } + + @Test + public void testAnonymousPush() throws Exception { Git git = Git.open(folder); File file = new File(folder, "TODO"); - 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").call(); + 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.push().setPushAll().call(); + git.getRepository().close(); + } + + @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(); + 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(); + } git.push().setPushAll().call(); git.getRepository().close(); } -- Gitblit v1.9.1