com.gitblit.tests.GitBlitSuite.java Source code

Java tutorial

Introduction

Here is the source code for com.gitblit.tests.GitBlitSuite.java

Source

/*
 * 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.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.FileRepositoryBuilder;
import org.eclipse.jgit.util.FS;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.runner.RunWith;
import org.junit.runners.Suite;
import org.junit.runners.Suite.SuiteClasses;

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, UserModelTest.class, UserChoiceTest.class,
        ObjectCacheTest.class, PermissionsTest.class, UserServiceTest.class, LdapAuthenticationTest.class,
        MarkdownUtilsTest.class, JGitUtilsTest.class, SyndicationUtilsTest.class, DiffUtilsTest.class,
        MetricUtilsTest.class, X509UtilsTest.class, GitBlitTest.class, FederationTests.class, RpcTests.class,
        GitServletTest.class, GitDaemonTest.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, TicketReferenceTest.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() {
        return getRepository("helloworld.git");
    }

    public static Repository getTicgitRepository() {
        return getRepository("ticgit.git");
    }

    public static Repository getJGitRepository() {
        return getRepository("test/jgit.git");
    }

    public static Repository getAmbitionRepository() {
        return getRepository("test/ambition.git");
    }

    public static Repository getGitectiveRepository() {
        return getRepository("test/gitective.git");
    }

    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);
            if (gitDir == null)
                gitDir = new File("data/git/helloworld.git");
            Repository repository = new FileRepositoryBuilder().setGitDir(gitDir).build();
            return repository;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    public static boolean startGitblit() throws Exception {
        if (started.get()) {
            // 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, "--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(5000);

        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(5000);
    }

    @BeforeClass
    public static void setUp() throws Exception {
        startGitblit();

        if (REPOSITORIES.exists() || REPOSITORIES.mkdirs()) {
            cloneOrFetch("helloworld.git", "https://github.com/carstenartur/hello-world.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/carstenartur/hello-world.git");
            cloneOrFetch("test/ambition.git", "https://github.com/defunkt/ambition.git");
            cloneOrFetch("test/gitective.git", "https://github.com/kevinsawicki/gitective.git");

            showRemoteBranches("ticgit.git");
            automaticallyTagBranchTips("ticgit.git");
            showRemoteBranches("test/jgit.git");
            automaticallyTagBranchTips("test/jgit.git");
        }
    }

    @AfterClass
    public static void tearDown() throws Exception {
        stopGitblit();
    }

    private static void cloneOrFetch(String name, String fromUrl) throws Exception {
        System.out.print("Fetching " + name + "... ");
        try {
            JGitUtils.cloneRepository(REPOSITORIES, name, fromUrl);
        } catch (Throwable t) {
            System.out.println("Error: " + t.getMessage());
        }
        System.out.println("done.");
    }

    private static void showRemoteBranches(String repositoryName) {
        try {
            IRepositoryManager repositoryManager = GitblitContext.getManager(IRepositoryManager.class);
            RepositoryModel model = repositoryManager.getRepositoryModel(repositoryName);
            model.showRemoteBranches = true;
            repositoryManager.updateRepositoryModel(model.name, model, false);
        } catch (GitBlitException g) {
            g.printStackTrace();
        }
    }

    private static void automaticallyTagBranchTips(String repositoryName) {
        try {
            IRepositoryManager repositoryManager = GitblitContext.getManager(IRepositoryManager.class);
            RepositoryModel model = repositoryManager.getRepositoryModel(repositoryName);
            model.useIncrementalPushTags = true;
            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());
            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();
        }
    }
}