net.serenitybdd.cucumber.suiteslicing.CucumberSuiteSlicer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of serenity-cucumber Show documentation

There is a newer version: 4.2.8

package net.serenitybdd.cucumber.suiteslicing;

import io.cucumber.gherkin.CucumberScenarioLoader;
import net.serenitybdd.cucumber.util.TagParser;

import java.net.URI;
import java.util.List;
import java.util.function.Predicate;

import static com.google.common.collect.Lists.newArrayList;

public class CucumberSuiteSlicer {

    private final List featurePaths;
    private final TestStatistics statistics;

    public CucumberSuiteSlicer(List featurePaths, TestStatistics statistics) {
        this.featurePaths = featurePaths;
        this.statistics = statistics;
    }

    public WeightedCucumberScenarios scenarios(int batchNumber, int batchCount, int forkNumber, int forkCount, List tagFilters) {
        return new CucumberScenarioLoader(featurePaths, statistics).load()
            .filter(forSuppliedTags(tagFilters))
            .slice(batchNumber).of(batchCount).slice(forkNumber).of(forkCount);
    }

    private Predicate forSuppliedTags(List tagFilters) {
        return cucumberScenario -> TagParser.parseFromTagFilters(tagFilters).evaluate(newArrayList(cucumberScenario.tags));
    }
}