mockit.coverage.reporting.sourceFiles.NeutralOutput Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmockit Show documentation

JMockit is a Java toolkit for automated developer testing. It contains APIs for the creation of the objects to be tested, for mocking dependencies, and for faking external APIs; JUnit (4 & 5) and TestNG test runners are supported. It also contains an advanced code coverage tool.

The newest version!

/*
 * Copyright (c) 2006 JMockit developers
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.coverage.reporting.sourceFiles;

import edu.umd.cs.findbugs.annotations.NonNull;
import edu.umd.cs.findbugs.annotations.Nullable;

import java.io.PrintWriter;

import mockit.coverage.reporting.parsing.LineElement;
import mockit.coverage.reporting.parsing.LineParser;

final class NeutralOutput {
    @NonNull
    private final PrintWriter output;
    private boolean previousLineInImports;
    private boolean previousLineInComments;
    @Nullable
    private String lineIndentation;
    private boolean blankLinesPending;

    NeutralOutput(@NonNull PrintWriter output) {
        this.output = output;
    }

    boolean writeLineWithoutCoverageInfo(@NonNull LineParser lineParser) {
        if (previousLineInComments || !previousLineInImports) {
            if (writeLineInComments(lineParser) || writeLineInImports(lineParser)) {
                return true;
            }
        } else if (writeLineInImports(lineParser) || writeLineInComments(lineParser)) {
            return true;
        }

        if (lineParser.isBlankLine()) {
            blankLinesPending = true;
            return true;
        }

        writeBlankLineIfPending();
        return false;
    }

    private boolean writeLineInComments(@NonNull LineParser lineParser) {
        LineElement initialElement = lineParser.getInitialElement();

        if (lineParser.isInComments()
                || previousLineInComments && initialElement.isComment() && initialElement.getNext() == null) {
            String lineText = initialElement.toString();

            if (previousLineInComments) {
                output.println();
            } else {
                writeOpeningForCollapsibleBlockOfLines();
                output.write("      ");
                extractLineIndentation(lineText);
                previousLineInComments = true;
            }

            output.write(lineText);
            return true;
        }
        if (previousLineInComments) {
            output.append("
").append(lineIndentation).println("/*...*/");
            output.println("    ");
            previousLineInComments = false;
        }

        return false;
    }

    private void writeOpeningForCollapsibleBlockOfLines() {
        writeBlankLineIfPending();
        output.println("    ");
        output.println("      ");
    }

    private void writeBlankLineIfPending() {
        if (blankLinesPending) {
            output.println("    ");
            blankLinesPending = false;
        }
    }

    private void extractLineIndentation(@NonNull String lineText) {
        int indentationSize = 0;

        for (int i = 0; i < lineText.length(); i++, indentationSize++) {
            if (lineText.charAt(i) > ' ') {
                break;
            }
        }

        lineIndentation = lineText.substring(0, indentationSize);
    }

    private boolean writeLineInImports(@NonNull LineParser lineParser) {
        LineElement initialElement = lineParser.getInitialElement();
        boolean isImport = initialElement.isKeyword("import");

        if (!previousLineInImports && isImport || previousLineInImports && (isImport || lineParser.isBlankLine())) {
            String lineText = initialElement.toString();

            if (previousLineInImports) {
                output.println();
                blankLinesPending = !isImport;
            } else {
                writeOpeningForCollapsibleBlockOfLines();
                output.write("      ");
                previousLineInImports = true;
            }

            output.write(lineText);
            return true;
        }
        if (previousLineInImports) {
            output.println("
import ...");
            output.println("    ");
            previousLineInImports = false;
        }

        return false;
    }
}