com.github.nagyesta.abortmission.testkit.LaunchEvaluationUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of abort.testkit Show documentation
A collection of classes we can use to test Abort Mission Boosters.
There is a newer version: 5.0.281
package com.github.nagyesta.abortmission.testkit;

import com.github.nagyesta.abortmission.core.healthcheck.MissionHealthCheckEvaluator;
import com.github.nagyesta.abortmission.core.healthcheck.ReadOnlyStageStatistics;
import com.github.nagyesta.abortmission.core.telemetry.StageResult;
import com.github.nagyesta.abortmission.core.telemetry.StageTimeMeasurement;

import java.util.List;
import java.util.function.Consumer;
import java.util.stream.Collectors;

/**
 * Utility class for helping with the evaluation of launch statistics.
 */
public final class LaunchEvaluationUtil {

    private LaunchEvaluationUtil() {
        throw new IllegalCallerException("Utility class.");
    }

    /**
     * Returns the exception classes causing the failures for the given evaluator.
     *
     * @param evaluator The evaluator we want to check.
     * @return The exception classes causing the failures.
     */
    public static List findExceptionsForMissionFailuresOf(final MissionHealthCheckEvaluator evaluator) {
        final ReadOnlyStageStatistics statistics = evaluator.getStats().getReadOnlyMission();
        return findExceptions(statistics);
    }

    /**
     * Returns the exception classes causing the failures for the given evaluator.
     *
     * @param evaluator The evaluator we want to check.
     * @return The exception classes causing the failures.
     */
    public static List findExceptionsForCountdownFailuresOf(final MissionHealthCheckEvaluator evaluator) {
        final ReadOnlyStageStatistics statistics = evaluator.getStats().getReadOnlyCountdown();
        return findExceptions(statistics);
    }

    /**
     * Returns the messages of the exceptions causing the failures for the given evaluator.
     *
     * @param evaluator The evaluator we want to check.
     * @return The messages of the exceptions causing the failures.
     */
    public static List findThrowableMessagesForMissionFailuresOf(final MissionHealthCheckEvaluator evaluator) {
        final ReadOnlyStageStatistics statistics = evaluator.getStats().getReadOnlyMission();
        return findThrowableMessages(statistics);
    }

    /**
     * Returns the messages of the exceptions causing the failures for the given evaluator.
     *
     * @param evaluator The evaluator we want to check.
     * @return The messages of the exceptions causing the failures.
     */
    public static List findThrowableMessagesForCountdownFailuresOf(final MissionHealthCheckEvaluator evaluator) {
        final ReadOnlyStageStatistics statistics = evaluator.getStats().getReadOnlyCountdown();
        return findThrowableMessages(statistics);
    }

    /**
     * Returns the display names of the countdown measurements for the given evaluator.
     *
     * @param evaluator The evaluator we want to check.
     * @return The display names of the countdown measurements.
     */
    public static List findCountdownDisplayNamesForMeasurementsOf(final MissionHealthCheckEvaluator evaluator) {
        return evaluator.getStats().getReadOnlyCountdown().timeSeriesStream()
                .map(StageTimeMeasurement::getDisplayName)
                .collect(Collectors.toList());
    }

    /**
     * Returns the display names of the mission measurements for the given evaluator.
     *
     * @param evaluator The evaluator we want to check.
     * @return The display names of the mission measurements.
     */
    public static List findMissionDisplayNamesForMeasurementsOf(final MissionHealthCheckEvaluator evaluator) {
        return evaluator.getStats().getReadOnlyMission().timeSeriesStream()
                .map(StageTimeMeasurement::getDisplayName)
                .collect(Collectors.toList());
    }

    /**
     * Applies the given action to the stack trace elements of the mission failures.
     *
     * @param evaluator The evaluator we want to check.
     * @param action    The action we want to apply.
     */
    public static void forEachNonFilteredStackTraceElementOfMissionFailures(
            final MissionHealthCheckEvaluator evaluator, final Consumer action) {
        evaluator.getStats().getReadOnlyMission().timeSeriesStream()
                .filter(s -> s.getResult() == StageResult.FAILURE)
                .map(StageTimeMeasurement::getStackTrace)
                .flatMap(List::stream)
                .filter(e -> !e.equals("< filtered >"))
                .forEach(action);
    }

    private static List findExceptions(final ReadOnlyStageStatistics statistics) {
        return statistics.timeSeriesStream()
                .filter(s -> s.getResult() == StageResult.FAILURE)
                .map(StageTimeMeasurement::getThrowableClass)
                .collect(Collectors.toList());
    }

    private static List findThrowableMessages(final ReadOnlyStageStatistics statistics) {
        return statistics.timeSeriesStream()
                .filter(s -> s.getResult() == StageResult.FAILURE)
                .map(StageTimeMeasurement::getThrowableMessage)
                .collect(Collectors.toList());
    }
}