com.streamsimple.rt.srcctl.GitSourceControlAgentTest.java Source code

Introduction

Here is the source code for com.streamsimple.rt.srcctl.GitSourceControlAgentTest.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.streamsimple.rt.srcctl;

import java.io.File;
import java.io.IOException;

import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.storage.file.FileRepositoryBuilder;
import org.junit.Assert;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.Description;

import com.streamsimple.commons.lang3.tuple.ImmutablePair;
import com.streamsimple.commons.lang3.tuple.Pair;
import com.streamsimple.javautil.err.ReturnError;
import com.streamsimple.javautils.testutils.DirTestWatcher;
import com.streamsimple.rt.executors.ProcessCommandExecutor;

public class GitSourceControlAgentTest {
    @Rule
    public final GitSourceControlAgentTestWatcher testWatcher = new GitSourceControlAgentTestWatcher();

    public static class GitSourceControlAgentTestWatcher extends DirTestWatcher {
        private SourceControlAgent sourceControlAgent;

        @Override
        protected void starting(Description description) {
            super.starting(description);
            initGitRepo();
            initSourceControlAgent();
        }

        public SourceControlAgent getSourceControlAgent() {
            return sourceControlAgent;
        }

        private void initGitRepo() {
            final String command = "bash -c 'git init'";
            ProcessCommandExecutor commandExecutor = new ProcessCommandExecutor();
            ReturnError returnError = commandExecutor.execute(getDirPath(), command);
            Assert.assertNull(returnError);
        }

        private void initSourceControlAgent() {
            Pair<SourceControlAgent, ReturnError> result = new GitSourceControlAgent.Builder().setDir(getDir())
                    .build();
            Assert.assertNull(result.getRight());
            this.sourceControlAgent = result.getLeft();
        }
    }

    @Test
    public void hasUncommittedTestFalse() {
        SourceControlAgent agent = testWatcher.getSourceControlAgent();

        Assert.assertEquals(new ImmutablePair<>(false, null), agent.hasUncommittedChanges());
    }

    @Test
    public void hasCommittedTestTrue() throws IOException {
        new File(testWatcher.getDir(), "test.txt").createNewFile();
        SourceControlAgent agent = testWatcher.getSourceControlAgent();

        Assert.assertEquals(new ImmutablePair<>(true, null), agent.hasUncommittedChanges());
    }

    @Test
    public void createBranchTest() throws Exception {
        addCommitToRepo();

        SourceControlAgent agent = testWatcher.getSourceControlAgent();

        ReturnError error = agent.createBranchFrom("master", "test1");
        Assert.assertNull(error);

        checkCurrentBranch("master");
    }

    @Test
    public void checkoutBranchTest() throws Exception {
        addCommitToRepo();

        SourceControlAgent agent = testWatcher.getSourceControlAgent();

        ReturnError error = agent.createBranchFrom("master", "test1");
        Assert.assertNull(error);
        error = agent.checkoutBranch("test1");
        Assert.assertNull(error);

        checkCurrentBranch("test1");
    }

    private void checkCurrentBranch(String name) {
        SourceControlAgent agent = testWatcher.getSourceControlAgent();

        Pair<String, ReturnError> result = agent.getCurrentBranch();
        Assert.assertNull(result.getRight());
        Assert.assertEquals(name, result.getLeft());
    }

    private void addCommitToRepo() throws Exception {
        new File(testWatcher.getDir(), "test.txt").createNewFile();

        Repository repository = new FileRepositoryBuilder().readEnvironment().findGitDir(testWatcher.getDir())
                .build();

        try (Git git = new Git(repository)) {
            git.commit().setCommitter("tester", "tester@release-tool.io").setMessage("Testing").call();
        }
    }
}