tech.grasshopper.processor.HierarchyProcessor Maven / Gradle / Ivy

Go to download
package tech.grasshopper.processor;

import java.util.Iterator;
import java.util.List;
import java.util.stream.Collectors;

import javax.inject.Inject;
import javax.inject.Singleton;

import tech.grasshopper.pojo.Feature;
import tech.grasshopper.pojo.Scenario;
import tech.grasshopper.pojo.Step;
import tech.grasshopper.properties.ReportProperties;

@Singleton
public class HierarchyProcessor {

	private ReportProperties reportProperties;

	@Inject
	public HierarchyProcessor(ReportProperties reportProperties) {
		this.reportProperties = reportProperties;
	}

	public void process(List features) {
		updateScenarioWithBackgroundSteps(features);
		updateStatusForStrict(features);
	}

	private void updateScenarioWithBackgroundSteps(List features) {

		for (Feature feature : features) {
			if (feature.getElements().get(0).getKeyword().equalsIgnoreCase("background")) {
				List scenarios = feature.getElements();
				Scenario backgroundScenario = null;
				Iterator iterator = scenarios.iterator();

				while (iterator.hasNext()) {
					Scenario scenario = iterator.next();
					if (scenario.getKeyword().equalsIgnoreCase("background")) {
						backgroundScenario = scenario;
						iterator.remove();
					} else
						scenario.getSteps().addAll(0, backgroundScenario.getSteps());
				}
			}
		}
	}

	private void updateStatusForStrict(List features) {

		List steps = features.stream().flatMap(f -> f.getElements().stream()).flatMap(s -> s.getSteps().stream())
				.collect(Collectors.toList());

		for (Step step : steps) {
			//Remove trailing space
			step.setKeyword(step.getKeyword().trim());

			if (reportProperties.isStrictCucumber6Behavior()) {
				String status = step.getResult().getStatus();
				if (status.equalsIgnoreCase("pending") || status.equalsIgnoreCase("undefined"))
					step.getResult().setStatus("failed");
			}
		}
	}
}