James Moger
2011-10-23 fe326255202dcfac8b0991ca9d28e3cf4bcc4fe6
tests/com/gitblit/tests/GitBlitSuite.java
@@ -21,14 +21,15 @@
import junit.framework.Test;
import junit.framework.TestSuite;
import org.eclipse.jgit.api.CloneCommand;
import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.storage.file.FileRepository;
import com.gitblit.FileSettings;
import com.gitblit.FileUserService;
import com.gitblit.GitBlit;
import com.gitblit.JettyLoginService;
import com.gitblit.GitBlitException;
import com.gitblit.models.RepositoryModel;
import com.gitblit.utils.JGitUtils;
public class GitBlitSuite extends TestSetup {
   public static final File REPOSITORIES = new File("git");
@@ -39,11 +40,16 @@
   public static Test suite() {
      TestSuite suite = new TestSuite();
      suite.addTestSuite(FileUtilsTest.class);
      suite.addTestSuite(TimeUtilsTest.class);
      suite.addTestSuite(StringUtilsTest.class);
      suite.addTestSuite(ByteFormatTest.class);
      suite.addTestSuite(MarkdownUtilsTest.class);
      suite.addTestSuite(JGitUtilsTest.class);
      suite.addTestSuite(SyndicationUtilsTest.class);
      suite.addTestSuite(DiffUtilsTest.class);
      suite.addTestSuite(MetricUtilsTest.class);
      suite.addTestSuite(TicgitUtilsTest.class);
      suite.addTestSuite(GitBlitTest.class);
      return new GitBlitSuite(suite);
   }
@@ -56,38 +62,69 @@
      return new FileRepository(new File(REPOSITORIES, "ticgit.git"));
   }
   @Override
   protected void setUp() throws Exception {
      if (REPOSITORIES.exists() || REPOSITORIES.mkdirs()) {
         cloneOrFetch("helloworld.git", "https://github.com/git/hello-world.git", true);
         cloneOrFetch("nested/helloworld.git", "https://github.com/git/hello-world.git", true);
         cloneOrFetch("ticgit.git", "https://github.com/jeffWelling/ticgit.git", true);
      }
      FileSettings settings = new FileSettings("distrib/gitblit.properties");
      GitBlit.self().configureContext(settings);
      JettyLoginService loginService = new JettyLoginService(new File("distrib/users.properties"));
      loginService.loadUsers();
      GitBlit.self().setLoginService(loginService);
   public static Repository getJGitRepository() throws Exception {
      return new FileRepository(new File(REPOSITORIES, "test/jgit.git"));
   }
   private void cloneOrFetch(String toFolder, String fromUrl, boolean bare) throws Exception {
      File folder = new File(REPOSITORIES, toFolder + (bare ? "" : "/.git"));
      if (folder.exists()) {
         System.out.print("Updating " + (bare ? "bare " : " ") + toFolder + "... ");
         FileRepository repository = new FileRepository(new File(REPOSITORIES, toFolder));
         Git git = new Git(repository);
         git.fetch().call();
         repository.close();
         System.out.println("done.");
      } else {
         System.out.println("Cloning " + (bare ? "bare " : " ") + toFolder + "... ");
         CloneCommand clone = new CloneCommand();
         clone.setBare(bare);
         clone.setCloneAllBranches(true);
         clone.setURI(fromUrl);
         clone.setDirectory(folder);
         clone.call();
         System.out.println("done.");
   public static Repository getBluezGnomeRepository() throws Exception {
      return new FileRepository(new File(REPOSITORIES, "test/bluez-gnome.git"));
   }
   @Override
   protected void setUp() throws Exception {
      FileSettings settings = new FileSettings("distrib/gitblit.properties");
      GitBlit.self().configureContext(settings, true);
      FileUserService loginService = new FileUserService(new File("distrib/users.properties"));
      GitBlit.self().setUserService(loginService);
      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("test/jgit.git", "https://github.com/eclipse/jgit.git");
         cloneOrFetch("test/helloworld.git", "https://github.com/git/hello-world.git");
         enableTickets("ticgit.git");
         enableDocs("ticgit.git");
         showRemoteBranches("ticgit.git");
         showRemoteBranches("test/jgit.git");
      }
   }
   private void cloneOrFetch(String name, String fromUrl) throws Exception {
      System.out.print("Fetching " + name + "... ");
      JGitUtils.cloneRepository(REPOSITORIES, name, fromUrl);
      System.out.println("done.");
   }
   private void enableTickets(String repositoryName) {
      try {
         RepositoryModel model = GitBlit.self().getRepositoryModel(repositoryName);
         model.useTickets = true;
         GitBlit.self().updateRepositoryModel(model.name, model, false);
      } catch (GitBlitException g) {
         g.printStackTrace();
      }
   }
   private void enableDocs(String repositoryName) {
      try {
         RepositoryModel model = GitBlit.self().getRepositoryModel(repositoryName);
         model.useDocs = true;
         GitBlit.self().updateRepositoryModel(model.name, model, false);
      } catch (GitBlitException g) {
         g.printStackTrace();
      }
   }
   private void showRemoteBranches(String repositoryName) {
      try {
         RepositoryModel model = GitBlit.self().getRepositoryModel(repositoryName);
         model.showRemoteBranches = true;
         GitBlit.self().updateRepositoryModel(model.name, model, false);
      } catch (GitBlitException g) {
         g.printStackTrace();
      }
   }
}