Proof of concept #1026

Paul Martin

2016-04-16 eecaad8b8e2c447429c31a01d49260ddd6b4ee03

 src/test/java/com/gitblit/tests/GitBlitSuite.java

@@ -33,83 +33,98 @@
import org.junit.runners.Suite;

import org.junit.runners.Suite.SuiteClasses;



import com.gitblit.GitBlit;

import com.gitblit.GitBlitException;

import com.gitblit.GitBlitServer;

import com.gitblit.manager.IRepositoryManager;

import com.gitblit.models.RepositoryModel;

import com.gitblit.servlet.GitblitContext;

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({ ArrayUtilsTest.class, FileUtilsTest.class, TimeUtilsTest.class,

      StringUtilsTest.class, Base64Test.class, JsonUtilsTest.class, ByteFormatTest.class,

      ObjectCacheTest.class, PermissionsTest.class, UserServiceTest.class, LdapUserServiceTest.class,

      UserModelTest.class, UserChoiceTest.class,

      ObjectCacheTest.class, PermissionsTest.class, UserServiceTest.class, LdapAuthenticationTest.class,

      MarkdownUtilsTest.class, JGitUtilsTest.class, SyndicationUtilsTest.class,

      DiffUtilsTest.class, MetricUtilsTest.class, TicgitUtilsTest.class, X509UtilsTest.class,

      DiffUtilsTest.class, MetricUtilsTest.class, X509UtilsTest.class,

      GitBlitTest.class, FederationTests.class, RpcTests.class, GitServletTest.class, GitDaemonTest.class,

      GroovyScriptTest.class, LuceneExecutorTest.class, IssuesTest.class, RepositoryModelTest.class,

      FanoutServiceTest.class })

      SshDaemonTest.class, GroovyScriptTest.class, LuceneExecutorTest.class, RepositoryModelTest.class,

      FanoutServiceTest.class, Issue0259Test.class, Issue0271Test.class, HtpasswdAuthenticationTest.class,

      ModelUtilsTest.class, JnaUtilsTest.class, LdapSyncServiceTest.class, FileTicketServiceTest.class,

      BranchTicketServiceTest.class, RedisTicketServiceTest.class, AuthenticationManagerTest.class,

      SshKeysDispatcherTest.class, UITicketTest.class, PathUtilsTest.class, SshKerberosAuthenticationTest.class,

      GravatarTest.class, FilestoreManagerTest.class, FilestoreServletTest.class })

public class GitBlitSuite {



   public static final File BASEFOLDER = new File("data");



   public static final File REPOSITORIES = new File("data/git");

	


   public static final File SETTINGS = new File("src/test/config/test-gitblit.properties");

	


   public static final File USERSCONF = new File("src/test/config/test-users.conf");



   static int port = 8280;

   static int gitPort = 8300;

   static int shutdownPort = 8281;

   static int sshPort = 39418;



   public static String url = "http://localhost:" + port;

   public static String gitServletUrl = "http://localhost:" + port + "/git";

   public static String gitDaemonUrl = "git://localhost:" + gitPort;

   public static String sshDaemonUrl = "ssh://admin@localhost:" + sshPort;

   public static String account = "admin";

   public static String password = "admin";



   private static AtomicBoolean started = new AtomicBoolean(false);



   public static Repository getHelloworldRepository() throws Exception {

   public static Repository getHelloworldRepository() {

      return getRepository("helloworld.git");

   }



   public static Repository getTicgitRepository() throws Exception {

   public static Repository getTicgitRepository() {

      return getRepository("ticgit.git");

   }



   public static Repository getJGitRepository() throws Exception {

   public static Repository getJGitRepository() {

      return getRepository("test/jgit.git");

   }



   public static Repository getAmbitionRepository() throws Exception {

   public static Repository getAmbitionRepository() {

      return getRepository("test/ambition.git");

   }



   public static Repository getIssuesTestRepository() throws Exception {

      JGitUtils.createRepository(REPOSITORIES, "gb-issues.git").close();

      return getRepository("gb-issues.git");

   }

	
   public static Repository getGitectiveRepository() throws Exception {

   public static Repository getGitectiveRepository() {

      return getRepository("test/gitective.git");

   }

	
   private static Repository getRepository(String name) throws Exception {

      File gitDir = FileKey.resolve(new File(REPOSITORIES, name), FS.DETECTED);

      Repository repository = new FileRepositoryBuilder().setGitDir(gitDir).build();

      return repository;



   public static Repository getTicketsTestRepository() {

      JGitUtils.createRepository(REPOSITORIES, "gb-tickets.git").close();

      return getRepository("gb-tickets.git");

   }



   private static Repository getRepository(String name) {

      try {

         File gitDir = FileKey.resolve(new File(REPOSITORIES, name), FS.DETECTED);

         Repository repository = new FileRepositoryBuilder().setGitDir(gitDir).build();

         return repository;

      } catch (Exception e) {

         e.printStackTrace();

      }

      return null;

   }



   public static boolean startGitblit() throws Exception {

@@ -117,22 +132,28 @@
         // already started

         return false;

      }

		


      GitServletTest.deleteWorkingFolders();

		


      // Start a Gitblit instance

      Executors.newSingleThreadExecutor().execute(new Runnable() {

         @Override

         public void run() {

            GitBlitServer.main("--httpPort", "" + port, "--httpsPort", "0", "--shutdownPort",

                  "" + shutdownPort, "--gitPort", "" + gitPort, "--repositoriesFolder",

                  "\"" + GitBlitSuite.REPOSITORIES.getAbsolutePath() + "\"", "--userService",

                  GitBlitSuite.USERSCONF.getAbsolutePath(), "--settings", GitBlitSuite.SETTINGS.getAbsolutePath(),

            GitBlitServer.main(

                  "--httpPort", "" + port,

                  "--httpsPort", "0",

                  "--shutdownPort", "" + shutdownPort,

                  "--gitPort", "" + gitPort,

                  "--sshPort", "" + sshPort,

                  "--repositoriesFolder", GitBlitSuite.REPOSITORIES.getAbsolutePath(),

                  "--userService", GitBlitSuite.USERSCONF.getAbsolutePath(),

                  "--settings", GitBlitSuite.SETTINGS.getAbsolutePath(),

                  "--baseFolder", "data");

         }

      });



      // Wait a few seconds for it to be running

      Thread.sleep(2500);

      Thread.sleep(5000);



      started.set(true);

      return true;

@@ -157,13 +178,11 @@
         cloneOrFetch("test/helloworld.git", "https://github.com/git/hello-world.git");

         cloneOrFetch("test/ambition.git", "https://github.com/defunkt/ambition.git");

         cloneOrFetch("test/gitective.git", "https://github.com/kevinsawicki/gitective.git");

			
         enableTickets("ticgit.git");

         enableDocs("ticgit.git");



         showRemoteBranches("ticgit.git");

         automaticallyTagBranchTips("ticgit.git");

         showRemoteBranches("test/jgit.git");

         automaticallyTagBranchTips("test/jgit.git");	
         automaticallyTagBranchTips("test/jgit.git");

      }

   }



@@ -182,46 +201,28 @@
      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);

         IRepositoryManager repositoryManager = GitblitContext.getManager(IRepositoryManager.class);

         RepositoryModel model = repositoryManager.getRepositoryModel(repositoryName);

         model.showRemoteBranches = true;

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

         repositoryManager.updateRepositoryModel(model.name, model, false);

      } catch (GitBlitException g) {

         g.printStackTrace();

      }

   }

	


   private static void automaticallyTagBranchTips(String repositoryName) {

      try {

         RepositoryModel model = GitBlit.self().getRepositoryModel(repositoryName);

         IRepositoryManager repositoryManager = GitblitContext.getManager(IRepositoryManager.class);

         RepositoryModel model = repositoryManager.getRepositoryModel(repositoryName);

         model.useIncrementalPushTags = true;

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

         repositoryManager.updateRepositoryModel(model.name, model, false);

      } catch (GitBlitException g) {

         g.printStackTrace();

      }

   }

	


   public static void close(File repository) {

      try {

         File gitDir = FileKey.resolve(repository, FS.detect());

@@ -232,11 +233,11 @@
         e.printStackTrace();

      }

   }

	


   public static void close(Git git) {

      close(git.getRepository());

   }

	


   public static void close(Repository r) {

      RepositoryCache.close(r);