James Moger
2012-10-13 c658df9e87d65b08d5482cf04489cb0532ff83dd
tests/com/gitblit/tests/GitBlitSuite.java
@@ -16,11 +16,17 @@
package com.gitblit.tests;
import java.io.File;
import java.lang.reflect.Field;
import java.util.concurrent.Executors;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;
import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.lib.RepositoryCache;
import org.eclipse.jgit.lib.RepositoryCache.FileKey;
import org.eclipse.jgit.storage.file.FileRepository;
import org.eclipse.jgit.util.FS;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.runner.RunWith;
@@ -49,10 +55,11 @@
@RunWith(Suite.class)
@SuiteClasses({ ArrayUtilsTest.class, FileUtilsTest.class, TimeUtilsTest.class,
      StringUtilsTest.class, Base64Test.class, JsonUtilsTest.class, ByteFormatTest.class,
      ObjectCacheTest.class, UserServiceTest.class, MarkdownUtilsTest.class, JGitUtilsTest.class,
      SyndicationUtilsTest.class, DiffUtilsTest.class, MetricUtilsTest.class,
      TicgitUtilsTest.class, GitBlitTest.class, FederationTests.class, RpcTests.class,
      GitServletTest.class, GroovyScriptTest.class, LuceneUtilsTest.class, IssuesTest.class })
      ObjectCacheTest.class, PermissionsTest.class, UserServiceTest.class, LdapUserServiceTest.class,
      MarkdownUtilsTest.class, JGitUtilsTest.class, SyndicationUtilsTest.class,
      DiffUtilsTest.class, MetricUtilsTest.class, TicgitUtilsTest.class,
      GitBlitTest.class, FederationTests.class, RpcTests.class, GitServletTest.class,
      GroovyScriptTest.class, LuceneExecutorTest.class, IssuesTest.class, RepositoryModelTest.class })
public class GitBlitSuite {
   public static final File REPOSITORIES = new File("git");
@@ -87,11 +94,12 @@
   }
   public static Repository getIssuesTestRepository() throws Exception {
      JGitUtils.createRepository(REPOSITORIES, "gb-issues.git").close();
      return new FileRepository(new File(REPOSITORIES, "gb-issues.git"));
   }
   
   public static Repository getGitectiveRepository() throws Exception {
      return new FileRepository(new File("test/gitective.git"));
      return new FileRepository(new File(REPOSITORIES, "test/gitective.git"));
   }
   public static boolean startGitblit() throws Exception {
@@ -99,6 +107,9 @@
         // already started
         return false;
      }
      GitServletTest.deleteWorkingFolders();
      // Start a Gitblit instance
      Executors.newSingleThreadExecutor().execute(new Runnable() {
         public void run() {
@@ -121,7 +132,7 @@
      GitBlitServer.main("--stop", "--shutdownPort", "" + shutdownPort);
      // Wait a few seconds for it to be running
      Thread.sleep(2500);
      Thread.sleep(5000);
   }
   @BeforeClass
@@ -130,17 +141,13 @@
      if (REPOSITORIES.exists() || REPOSITORIES.mkdirs()) {
         cloneOrFetch("helloworld.git", "https://github.com/git/hello-world.git");
         cloneOrFetch("ticgit.git", "https://github.com/jeffWelling/ticgit.git");
         cloneOrFetch("test/bluez-gnome.git",
               "https://git.kernel.org/pub/scm/bluetooth/bluez-gnome.git");
         cloneOrFetch("ticgit.git", "https://github.com/schacon/ticgit.git");
         cloneOrFetch("test/jgit.git", "https://github.com/eclipse/jgit.git");
         cloneOrFetch("test/helloworld.git", "https://github.com/git/hello-world.git");
         cloneOrFetch("test/ambition.git", "https://github.com/defunkt/ambition.git");
         cloneOrFetch("test/theoretical-physics.git", "https://github.com/certik/theoretical-physics.git");
         cloneOrFetch("test/gitective.git", "https://github.com/kevinsawicki/gitective.git");
         
         JGitUtils.createRepository(REPOSITORIES, "gb-issues.git").close();
         enableTickets("ticgit.git");
         enableDocs("ticgit.git");
         showRemoteBranches("ticgit.git");
@@ -188,4 +195,36 @@
         g.printStackTrace();
      }
   }
   public static void close(File repository) {
      try {
         File gitDir = FileKey.resolve(repository, FS.detect());
         if (gitDir != null && gitDir.exists()) {
            close(RepositoryCache.open(FileKey.exact(gitDir, FS.detect())));
         }
      } catch (Exception e) {
         e.printStackTrace();
      }
   }
   public static void close(Git git) {
      close(git.getRepository());
   }
   public static void close(Repository r) {
      RepositoryCache.close(r);
      // assume 2 uses in case reflection fails
      int uses = 2;
      try {
         Field useCnt = Repository.class.getDeclaredField("useCnt");
         useCnt.setAccessible(true);
         uses = ((AtomicInteger) useCnt.get(r)).get();
      } catch (Exception e) {
         e.printStackTrace();
      }
      for (int i = 0; i < uses; i++) {
         r.close();
      }
   }
}