org.mockito.internal.reporting.SmartPrinter Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of mockito-core Show documentation

Mockito mock objects library core API and implementation

There is a newer version: 5.12.0

/*
 * Copyright (c) 2007 Mockito contributors
 * This program is made available under the terms of the MIT License.
 */
package org.mockito.internal.reporting;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Set;

import org.mockito.invocation.Invocation;
import org.mockito.invocation.MatchableInvocation;

/**
 * Makes sure both wanted and actual are printed consistently (single line or multiline)
 * 
 * Makes arguments printed with types if necessary
 */
public class SmartPrinter {

    private final String wanted;
    private final List actuals;

    public SmartPrinter(
            MatchableInvocation wanted,
            Invocation actual,
            Integer... indexesOfMatchersToBeDescribedWithExtraTypeInfo) {
        this(
                wanted,
                Collections.singletonList(actual),
                indexesOfMatchersToBeDescribedWithExtraTypeInfo,
                Collections.emptySet());
    }

    public SmartPrinter(
            MatchableInvocation wanted,
            List allActualInvocations,
            Integer[] indexesOfMatchersToBeDescribedWithExtraTypeInfo,
            Set classNamesToBeDescribedWithFullName) {
        PrintSettings printSettings = new PrintSettings();
        printSettings.setMultiline(isMultiLine(wanted, allActualInvocations));
        printSettings.setMatchersToBeDescribedWithExtraTypeInfo(
                indexesOfMatchersToBeDescribedWithExtraTypeInfo);
        printSettings.setMatchersToBeDescribedWithFullName(classNamesToBeDescribedWithFullName);

        this.wanted = printSettings.print(wanted);

        List actuals = new ArrayList<>();
        for (Invocation actual : allActualInvocations) {
            actuals.add(printSettings.print(actual));
        }
        this.actuals = Collections.unmodifiableList(actuals);
    }

    public String getWanted() {
        return wanted;
    }

    public List getActuals() {
        return actuals;
    }

    private static boolean isMultiLine(
            MatchableInvocation wanted, List allActualInvocations) {
        boolean isWantedMultiline = wanted.toString().contains("\n");
        boolean isAnyActualMultiline = false;
        for (Invocation invocation : allActualInvocations) {
            isAnyActualMultiline |= invocation.toString().contains("\n");
        }
        return isWantedMultiline || isAnyActualMultiline;
    }
}