MatchersPrinter.java

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

import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;

import org.mockito.ArgumentMatcher;
import org.mockito.internal.matchers.ContainsExtraTypeInfo;
import org.mockito.internal.reporting.PrintSettings;

@SuppressWarnings("unchecked")
public class MatchersPrinter {

    public String getArgumentsLine(List<ArgumentMatcher> matchers, PrintSettings printSettings) {
        Iterator args = applyPrintSettings(matchers, printSettings);
        return ValuePrinter.printValues("(", ", ", ");", args);
    }

    public String getArgumentsBlock(List<ArgumentMatcher> matchers, PrintSettings printSettings) {
        Iterator args = applyPrintSettings(matchers, printSettings);
        return ValuePrinter.printValues("(\n    ", ",\n    ", "\n);", args);
    }

    private Iterator<FormattedText> applyPrintSettings(
            List<ArgumentMatcher> matchers, PrintSettings printSettings) {
        List<FormattedText> out = new LinkedList<>();
        int i = 0;
        for (final ArgumentMatcher matcher : matchers) {
            if (matcher instanceof ContainsExtraTypeInfo) {
                ContainsExtraTypeInfo typeInfoMatcher = (ContainsExtraTypeInfo) matcher;
                Object wanted = typeInfoMatcher.getWanted();
                String simpleNameOfArgument =
                        wanted != null ? wanted.getClass().getSimpleName() : "";
                String fullyQualifiedClassName =
                        wanted != null ? wanted.getClass().getCanonicalName() : "";

                if (printSettings.extraTypeInfoFor(i)) {
                    out.add(
                            new FormattedText(
                                    typeInfoMatcher.toStringWithType(simpleNameOfArgument)));
                } else if (printSettings.fullyQualifiedNameFor(simpleNameOfArgument)) {
                    out.add(
                            new FormattedText(
                                    typeInfoMatcher.toStringWithType(fullyQualifiedClassName)));
                } else {
                    out.add(new FormattedText(MatcherToString.toString(matcher)));
                }
            } else {
                out.add(new FormattedText(MatcherToString.toString(matcher)));
            }
            i++;
        }
        return out.iterator();
    }
}