org.pitest.cucumber.CucumberTestUnitFinder Maven / Gradle / Ivy

Go to download
package org.pitest.cucumber;


import java.util.ArrayList;
import java.util.List;

import org.junit.runner.RunWith;
import org.pitest.testapi.TestUnit;
import org.pitest.testapi.TestUnitFinder;
import org.pitest.util.Log;

import cucumber.api.junit.Cucumber;
import cucumber.runtime.RuntimeOptions;
import cucumber.runtime.RuntimeOptionsFactory;
import cucumber.runtime.io.MultiLoader;
import cucumber.runtime.io.ResourceLoader;
import cucumber.runtime.model.CucumberExamples;
import cucumber.runtime.model.CucumberFeature;
import cucumber.runtime.model.CucumberScenario;
import cucumber.runtime.model.CucumberScenarioOutline;
import cucumber.runtime.model.CucumberTagStatement;

public class CucumberTestUnitFinder implements TestUnitFinder {

    public List findTestUnits(Class junitTestClass) {
        List result = new ArrayList();
        RunWith annotation = junitTestClass.getAnnotation(RunWith.class);
        if (annotation!= null && Cucumber.class.isAssignableFrom(annotation.value())) {
            RuntimeOptionsFactory runtimeOptionsFactory = new RuntimeOptionsFactory(junitTestClass);
            RuntimeOptions runtimeOptions = runtimeOptionsFactory.create();
            ClassLoader classLoader = junitTestClass.getClassLoader();
            ResourceLoader resourceLoader = new MultiLoader(classLoader);
            final List cucumberFeatures = runtimeOptions.cucumberFeatures(resourceLoader);
            for (CucumberFeature feature : cucumberFeatures) {
                Log.getLogger().fine("Found feature \"" + feature.getGherkinFeature().getName() + "\"");
                List featureElements = feature.getFeatureElements();
                for (CucumberTagStatement element : featureElements) {
                    if (element instanceof CucumberScenario) {
                        CucumberScenario scenario = (CucumberScenario) element;
                        Log.getLogger().fine("Found \"" + scenario.getVisualName() + "\"");
                        ScenarioTestUnit testUnit = new ScenarioTestUnit(junitTestClass, scenario);
                        result.add(testUnit);
                    } else if (element instanceof CucumberScenarioOutline) {
                        CucumberScenarioOutline scenarioOutline = (CucumberScenarioOutline) element;
                        Log.getLogger().fine("Found \"" + scenarioOutline.getVisualName() + "\"");
                        for (CucumberExamples examples : scenarioOutline.getCucumberExamplesList()) {
                            for (CucumberScenario scenario : examples.createExampleScenarios()) {
                                ScenarioTestUnit testUnit = new ScenarioTestUnit(junitTestClass, scenario);
                                result.add(testUnit);
                            }
                        }

                    } else {
                        Log.getLogger().warning("Ignoring unknown cucumber tag statement " + element.getVisualName());
                    }
                }
            }
        }
        return result;
    }
}