io.qameta.allure.severity.SeverityPlugin Maven / Gradle / Ivy

Go to download
package io.qameta.allure.severity;

import io.qameta.allure.Aggregator;
import io.qameta.allure.core.Configuration;
import io.qameta.allure.core.LaunchResults;
import io.qameta.allure.entity.TestResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.nio.file.Path;
import java.util.List;

import static io.qameta.allure.entity.LabelName.SEVERITY;

/**
 * Plugin that adds severity information to tests results.
 *
 * @since 2.0
 */
public class SeverityPlugin implements Aggregator {

    private static final Logger LOGGER = LoggerFactory.getLogger(SeverityPlugin.class);

    @Override
    public void aggregate(final Configuration configuration,
                          final List launchesResults,
                          final Path outputDirectory) {
        launchesResults.stream()
                .flatMap(results -> results.getResults().stream())
                .forEach(this::setSeverityLevel);

    }

    private void setSeverityLevel(final TestResult result) {
        final SeverityLevel severityLevel = result.findOne(SEVERITY)
                .map(this::getSeverity)
                .orElse(SeverityLevel.NORMAL);
        result.addExtraBlock("severity", severityLevel);
    }

    private SeverityLevel getSeverity(final String value) {
        try {
            return SeverityLevel.fromValue(value);
        } catch (Exception e) {
            LOGGER.error("Unknown severity level {}", value, e);
            return null;
        }
    }
}