org.eclipse.egit.ui.gitflow.FeatureFinishSquashHandlerTest.java Source code

Introduction

Here is the source code for org.eclipse.egit.ui.gitflow.FeatureFinishSquashHandlerTest.java

Source

/*******************************************************************************
 * Copyright (C) 2015, Max Hohenegger <eclipse@hohenegger.eu>
 *
 * 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
 *******************************************************************************/
package org.eclipse.egit.ui.gitflow;

import static org.eclipse.egit.gitflow.ui.internal.UIPreferences.FEATURE_FINISH_KEEP_BRANCH;
import static org.eclipse.egit.gitflow.ui.internal.UIPreferences.FEATURE_FINISH_SQUASH;
import static org.eclipse.swtbot.swt.finder.waits.Conditions.shellIsActive;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotEquals;
import static org.junit.Assert.assertTrue;

import java.io.IOException;
import java.util.Iterator;

import org.eclipse.egit.gitflow.GitFlowRepository;
import org.eclipse.egit.gitflow.ui.Activator;
import org.eclipse.egit.gitflow.ui.internal.UIText;
import org.eclipse.jface.preference.IPreferenceStore;
import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.api.errors.GitAPIException;
import org.eclipse.jgit.api.errors.NoHeadException;
import org.eclipse.jgit.revwalk.RevCommit;
import org.eclipse.swtbot.swt.finder.junit.SWTBotJunit4ClassRunner;
import org.junit.Test;
import org.junit.runner.RunWith;

/**
 * Tests for the Team->Gitflow->Feature Finish action with squash option
 */
@RunWith(SWTBotJunit4ClassRunner.class)
public class FeatureFinishSquashHandlerTest extends AbstractFeatureFinishHandlerTest {

    private static final String SQUASHED_COMMENT_SUMMARY = "Hello World";

    @Test
    public void testFeatureFinishSquash() throws Exception {
        int expectedCommitCount = 2;

        init();

        setContentAddAndCommit("bar");
        expectedCommitCount++;

        createFeature(FEATURE_NAME);
        RevCommit commit1 = setContentAddAndCommit("commit 1");
        expectedCommitCount++;
        RevCommit commit2 = setContentAddAndCommit("commit 2");
        expectedCommitCount++;

        checkoutBranch(DEVELOP);

        checkoutFeature(FEATURE_NAME);

        finishFeature();
        expectedCommitCount--;

        RevCommit developHead = new GitFlowRepository(repository).findHead();
        assertNotEquals(developHead, commit1);
        assertNotEquals(developHead, commit2);

        assertEquals(expectedCommitCount, countCommits());

        assertTrue(developHead.getFullMessage().startsWith(SQUASHED_COMMENT_SUMMARY));

        IPreferenceStore prefStore = Activator.getDefault().getPreferenceStore();
        assertTrue(prefStore.getBoolean(FEATURE_FINISH_SQUASH));
        assertFalse(prefStore.getBoolean(FEATURE_FINISH_KEEP_BRANCH));
    }

    private int countCommits() throws GitAPIException, NoHeadException, IOException {
        Iterable<RevCommit> commits = Git.wrap(repository).log().all().call();
        Iterator<RevCommit> iterator = commits.iterator();
        int count = 0;

        while (iterator.hasNext()) {
            iterator.next();
            count++;
        }
        return count;
    }

    @Override
    protected void preFinish() {
        bot.waitUntil(shellIsActive(UIText.FeatureFinishHandler_rewordSquashedCommitMessage));
        String text = bot.styledText().getText();
        text = text.substring(text.indexOf('\n'));
        bot.styledText().setText(SQUASHED_COMMENT_SUMMARY + text);
        bot.button("OK").click();
    }

    @Override
    protected void selectOptions() {
        bot.checkBox(UIText.FinishFeatureDialog_squashCheck).click();
    }
}