SummaryMessageVisitorTest.java

package pl.touk.sputnik.engine.visitor;

import org.junit.Test;
import pl.touk.sputnik.TestEnvironment;
import pl.touk.sputnik.review.Review;

import static org.assertj.core.api.Assertions.assertThat;

public class SummaryMessageVisitorTest extends TestEnvironment {

    private static final String TOTAL_8_VIOLATIONS_FOUND = "Total 8 violations found";
    private static final String PROBLEM_SOURCE = "PMD";
    private static final String PROBLEM_MESSAGE = "configuration error";
    private static final String PROBLEM_FORMATTED_MESSAGE = "There is a problem with PMD: configuration error";

    @Test
    public void shouldAddSummaryMessage() {
        Review review = review();
        review.setTotalViolationCount(8);

        new SummaryMessageVisitor("Perfect").afterReview(review);

        assertThat(review.getMessages()).containsOnly(TOTAL_8_VIOLATIONS_FOUND);
    }

    @Test
    public void shouldAddSummaryMessageWithOneViolation() {
        Review review = review();
        review.setTotalViolationCount(1);

        new SummaryMessageVisitor("Perfect").afterReview(review);

        assertThat(review.getMessages()).containsOnly("Total 1 violation found");
    }

    @Test
    public void shouldAddPerfectMessageIfThereAreNoViolationsFound() {
        Review review = review();
        review.setTotalViolationCount(0);

        new SummaryMessageVisitor("Perfect").afterReview(review);

        assertThat(review.getMessages()).containsOnly("Perfect");
    }

    @Test
    public void shouldAddProblemMessagesPerfectMessageIfThereAreNoViolationsFound() {
        Review review = review();
        review.setTotalViolationCount(8);
        review.addProblem(PROBLEM_SOURCE, PROBLEM_MESSAGE);

        new SummaryMessageVisitor("Perfect").afterReview(review);

        assertThat(review.getMessages()).containsSequence(TOTAL_8_VIOLATIONS_FOUND, PROBLEM_FORMATTED_MESSAGE);
    }

}