org.eclipse.egit.core.test.op.SquashCommitsOperationTest.java Source code

Introduction

Here is the source code for org.eclipse.egit.core.test.op.SquashCommitsOperationTest.java

Source

/*******************************************************************************
 * Copyright (c) 2014 Maik Schreiber
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Maik Schreiber - initial implementation
 *******************************************************************************/
package org.eclipse.egit.core.test.op;

import static org.junit.Assert.assertEquals;

import java.io.File;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;

import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.egit.core.op.SquashCommitsOperation;
import org.eclipse.egit.core.test.GitTestCase;
import org.eclipse.egit.core.test.TestRepository;
import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.api.LogCommand;
import org.eclipse.jgit.api.RebaseCommand.InteractiveHandler;
import org.eclipse.jgit.api.errors.GitAPIException;
import org.eclipse.jgit.lib.Constants;
import org.eclipse.jgit.lib.RebaseTodoLine;
import org.eclipse.jgit.revwalk.RevCommit;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class SquashCommitsOperationTest extends GitTestCase {
    private TestRepository testRepository;
    private RevCommit commit1;
    private RevCommit commit2;
    private RevCommit commit3;

    @Override
    @Before
    public void setUp() throws Exception {
        super.setUp();
        gitDir = new File(project.getProject().getLocationURI().getPath(), Constants.DOT_GIT);
        testRepository = new TestRepository(gitDir);
        testRepository.connect(project.getProject());
        testRepository.createInitialCommit("initial");

        File file = testRepository.createFile(project.getProject(), "file-1");
        commit1 = testRepository.addAndCommit(project.getProject(), file, "commit 1");
        testRepository.appendFileContent(file, "file-2");
        commit2 = testRepository.addAndCommit(project.getProject(), file, "commit 2");
        testRepository.appendFileContent(file, "file-3");
        commit3 = testRepository.addAndCommit(project.getProject(), file, "commit 3");
    }

    @Override
    @After
    public void tearDown() throws Exception {
        testRepository.dispose();
        super.tearDown();
    }

    @Test
    public void squash() throws Exception {
        InteractiveHandler messageHandler = new InteractiveHandler() {
            @Override
            public void prepareSteps(List<RebaseTodoLine> steps) {
                // not used
            }

            @Override
            public String modifyCommitMessage(String commit) {
                return "squashed";
            }
        };

        List<RevCommit> commits = Arrays.asList(commit1, commit2, commit3);
        SquashCommitsOperation op = new SquashCommitsOperation(testRepository.getRepository(), commits,
                messageHandler);
        op.execute(new NullProgressMonitor());

        assertEquals(2, countCommitsInHead());

        LogCommand log = new Git(testRepository.getRepository()).log();
        Iterable<RevCommit> logCommits = log.call();
        RevCommit latestCommit = logCommits.iterator().next();
        assertEquals("squashed", latestCommit.getFullMessage());
    }

    private int countCommitsInHead() throws GitAPIException {
        LogCommand log = new Git(testRepository.getRepository()).log();
        Iterable<RevCommit> commits = log.call();
        int result = 0;
        for (Iterator i = commits.iterator(); i.hasNext();) {
            i.next();
            result++;
        }
        return result;
    }
}