James Moger
2011-12-07 7e8873a14ccc2cb25213489d7d7ba97f09673831
tests/com/gitblit/tests/GitBlitSuite.java
@@ -1,29 +1,70 @@
/*
 * Copyright 2011 gitblit.com.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.gitblit.tests;
import java.io.File;
import java.util.concurrent.Executors;
import java.util.concurrent.atomic.AtomicBoolean;
import junit.extensions.TestSetup;
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 org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.runner.RunWith;
import org.junit.runners.Suite;
import org.junit.runners.Suite.SuiteClasses;
public class GitBlitSuite extends TestSetup {
import com.gitblit.GitBlit;
import com.gitblit.GitBlitException;
import com.gitblit.GitBlitServer;
import com.gitblit.models.RepositoryModel;
import com.gitblit.utils.JGitUtils;
/**
 * The GitBlitSuite uses test-gitblit.properties and test-users.conf. The suite
 * is fairly comprehensive for all lower-level functionality. Wicket pages are
 * currently not unit-tested.
 *
 * This suite starts a Gitblit server instance within the same JVM instance as
 * the unit tests. This allows the unit tests to access the GitBlit static
 * singleton while also being able to communicate with the instance via tcp/ip
 * for testing rpc requests, federation requests, and git servlet operations.
 *
 * @author James Moger
 *
 */
@RunWith(Suite.class)
@SuiteClasses({ 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 })
public class GitBlitSuite {
   public static final File REPOSITORIES = new File("git");
   private GitBlitSuite(TestSuite suite) {
      super(suite);
   }
   static int port = 8280;
   static int shutdownPort = 8281;
   public static Test suite() {
      TestSuite suite = new TestSuite();
      suite.addTestSuite(JGitUtilsTest.class);
      return new GitBlitSuite(suite);
   }
   public static String url = "http://localhost:" + port;
   public static String account = "admin";
   public static String password = "admin";
   private static AtomicBoolean started = new AtomicBoolean(false);
   public static Repository getHelloworldRepository() throws Exception {
      return new FileRepository(new File(REPOSITORIES, "helloworld.git"));
   }
@@ -32,33 +73,101 @@
      return new FileRepository(new File(REPOSITORIES, "ticgit.git"));
   }
   @Override
   protected void setUp() throws Exception {
   public static Repository getJGitRepository() throws Exception {
      return new FileRepository(new File(REPOSITORIES, "test/jgit.git"));
   }
   public static Repository getBluezGnomeRepository() throws Exception {
      return new FileRepository(new File(REPOSITORIES, "test/bluez-gnome.git"));
   }
   public static boolean startGitblit() throws Exception {
      if (started.get()) {
         // already started
         return false;
      }
      // 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",
                  "test-users.conf", "--settings", "test-gitblit.properties");
         }
      });
      // Wait a few seconds for it to be running
      Thread.sleep(2500);
      started.set(true);
      return true;
   }
   public static void stopGitblit() throws Exception {
      // Stop Gitblit
      GitBlitServer.main("--stop", "--shutdownPort", "" + shutdownPort);
      // Wait a few seconds for it to be running
      Thread.sleep(2500);
   }
   @BeforeClass
   public static void setUp() throws Exception {
      startGitblit();
      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);
         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 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.");
   @AfterClass
   public static void tearDown() throws Exception {
      stopGitblit();
   }
   private static void cloneOrFetch(String name, String fromUrl) throws Exception {
      System.out.print("Fetching " + name + "... ");
      JGitUtils.cloneRepository(REPOSITORIES, name, fromUrl);
      System.out.println("done.");
   }
   private static 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 static 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 static 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();
      }
   }
}