io.github.olib963.javatest.fixtures.internal.FixtureRunner Maven / Gradle / Ivy

Go to download
package io.github.olib963.javatest.fixtures.internal;

import io.github.olib963.javatest.TestResults;
import io.github.olib963.javatest.TestRunner;
import io.github.olib963.javatest.fixtures.FixtureDefinition;

import java.util.Optional;
import java.util.function.Function;

public class FixtureRunner implements TestRunner {

    private final String fixtureName;
    private final FixtureDefinition fixtureDefinition;
    private final Function testFunction;

    public FixtureRunner(String fixtureName, FixtureDefinition fixtureDefinition, Function testFunction) {
        this.fixtureName = fixtureName;
        this.fixtureDefinition = fixtureDefinition;
        this.testFunction = testFunction;
    }

    @Override
    public TestResults run() {
        return fixtureDefinition.create()
                .mapError(e -> new Exception("Could not create fixture \"" + fixtureName + '"', e))
                .map(this::runWithFixture)
                .recoverWith(e -> TestResults.empty().failBecause(flattenMessages(e)));

    }

    private TestResults runWithFixture(Fixture fixture) {
        var results = testFunction.apply(fixture).run();
        return fixtureDefinition.destroy(fixture)
                .map(unit -> results)
                .recoverWith(e -> results.failBecause(
                        flattenMessages(new Exception("Could not destroy fixture \"" + fixtureName + '"', e))));
    }

    private String flattenMessages(Throwable t) {
        return Optional.ofNullable(t.getCause())
                .map(c -> t.getMessage() + System.lineSeparator() + flattenMessages(c))
                .orElseGet(t::getMessage);
    }
}