io.qameta.allure.environment.Allure1EnvironmentPlugin Maven / Gradle / Ivy

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

import io.qameta.allure.Widget;
import io.qameta.allure.core.Configuration;
import io.qameta.allure.core.LaunchResults;
import io.qameta.allure.entity.EnvironmentItem;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Supplier;

import static io.qameta.allure.allure1.Allure1Plugin.ENVIRONMENT_BLOCK_NAME;
import static java.util.stream.Collectors.groupingBy;
import static java.util.stream.Collectors.toList;

/**
 * @author Egor Borisov [email protected]
 */
public class Allure1EnvironmentPlugin implements Widget {

    @Override
    public List getData(final Configuration configuration,
                                         final List launches) {
        final List> launchEnvironments = launches.stream()
                .flatMap(launch -> launch.getExtra(ENVIRONMENT_BLOCK_NAME,
                        (Supplier>) HashMap::new).entrySet().stream())
                .collect(toList());

        return launchEnvironments.stream()
                .collect(groupingBy(Map.Entry::getKey, toList()))
                .entrySet().stream().map(this::aggregateItem).collect(toList());
    }

    private EnvironmentItem aggregateItem(final Map.Entry>> entry) {
        return new EnvironmentItem()
                .withName(entry.getKey())
                .withValues(entry.getValue().stream().map(Map.Entry::getValue).collect(toList()));
    }

    @Override
    public String getName() {
        return "environment";
    }
}