FitFixtureReportingSystem.java

package dbfit.fixture.report;

import dbfit.util.MatchResult;
import static dbfit.util.MatchStatus.*;

import fit.Fixture;
import fit.Parse;

import static org.apache.commons.lang3.StringUtils.isEmpty;
import java.util.Objects;

public class FitFixtureReportingSystem implements ReportingSystem {

    protected final Fixture fixture;
    protected final Parse table;

    protected Parse lastClosedRow;
    protected Parse newRow; // new row. attached on 1st cell or on empty row end
    protected Parse newRowTail;

    public FitFixtureReportingSystem(final Fixture fixture, final Parse table) {
        this.fixture = fixture;
        this.table = table;

        lastClosedRow = table.parts.last();
        createNewRow();
    }

    /*
     * Initially "hanging" new row (not attached to main table)
     */
    protected void createNewRow() {
        newRow = new Parse("tr", null, null, null);
        newRowTail = newRow.parts; // null
    }

    /*
     * Ensure new row is attached (but still not closed)
     * OK to call that several times.
     */
    protected Parse newRowAttached() {
        lastClosedRow.more = newRow;
        return newRow;
    }

    protected void closeNewRow() {
        lastClosedRow = newRowAttached();
        createNewRow();
    }

    protected void addCell(final Parse cell) {
        if (newRowTail == null) {
            newRowAttached().parts = cell;
        } else {
            newRowTail.more = cell;
        }

        newRowTail = cell;
    }

    @Override
    public void endRow(final MatchResult res, final String description) {
        String descr = description;

        switch (res.getStatus()) {
        case MISSING:
            descr = Objects.toString(descr, "missing");
            fixture.wrong(newRow);
            break;
        case SURPLUS:
            descr = Objects.toString(descr, "surplus");
            fixture.wrong(newRow);
            break;
        }

        if (!isEmpty(descr)) {
            newRow.parts.addToBody(Fixture.gray(" " + descr));
        }

        closeNewRow();
    }

    @Override
    public void endRow(final MatchResult res) {
        endRow(res, null);
    }

    @Override
    public void addCell(final MatchResult res) {
        Parse cell = new Parse("td", res.getStringValue1(), null, null);

        switch (res.getStatus()) {
        case SUCCESS:
            fixture.right(cell);
            break;
        case WRONG:
            fixture.wrong(cell, res.getStringValue2());
            break;
        case MISSING:
            fixture.wrong(cell);
            break;
        case SURPLUS:
            cell.body = res.getStringValue2();
            fixture.wrong(cell);
            break;
        case EXCEPTION:
            fixture.exception(cell, res.getException());
            break;
        /*
            // TODO: ignored
        */
        default:
            throw new UnsupportedOperationException("Not implemented yet");
        }

        addCell(cell);
    }

    @Override
    public void addException(final Throwable ex) {
        Parse cell = new Parse("td", "", null, null);
        fixture.exception(cell, ex);
        addCell(cell);
        closeNewRow();
    }

    @Override
    public void incRight() {
        ++fixture.counts.right;
    }

    public Parse getTable() {
        return table;
    }
}