net.thucydides.junit.pipeline.ThucydidesPipelineRunner Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of thucydides-extensions Show documentation

There is a newer version: 0.9.93

Show newest version

package net.thucydides.junit.pipeline;

import com.google.common.collect.Lists;
import net.thucydides.core.model.TestOutcome;
import net.thucydides.core.model.TestStep;
import net.thucydides.junit.runners.ThucydidesRunner;
import org.junit.runners.model.InitializationError;

import java.util.List;

class ThucydidesPipelineRunner extends ThucydidesRunner {
    ThucydidesPipelineRunner(Class klass) throws InitializationError {
        super(klass);
    }

    @Override
    public List getTestOutcomes() {
        List flatTestOutcomes = super.getTestOutcomes();
        List expandedTestOutcomes = Lists.newArrayList();

        for(TestOutcome testOutcome : flatTestOutcomes) {
            expandedTestOutcomes.addAll(extractPipelineTestsFrom(testOutcome));
        }
        return expandedTestOutcomes;
    }

    private List extractPipelineTestsFrom(TestOutcome baseTestOutcome) {

        List steps = baseTestOutcome.getTestSteps();

        int numberOfOutcomes = countFirstRound(steps);
        List testOutcomePerPipeline = createTargetTestOutcomes(baseTestOutcome, numberOfOutcomes);

        int stepCounter = 0;
        for(TestStep step : steps) {
            TestOutcome currentTestOutcome = testOutcomePerPipeline.get(stepCounter % numberOfOutcomes);
            currentTestOutcome.recordStep(step);
            stepCounter++;
        }
        return testOutcomePerPipeline;
    }

    private List createTargetTestOutcomes(TestOutcome baseTestOutcome, int testOutcomeCount) {
        List targetTestOutcomes = Lists.newArrayList();
        for(int outcomeIndex = 1; outcomeIndex <= testOutcomeCount; outcomeIndex++) {
            String qualifiedTitle = baseTestOutcome.getTitle() + " [" + outcomeIndex + "]";
            TestOutcome newTestOutcome = TestOutcome.emptyCopyOf(baseTestOutcome);
            newTestOutcome.setTitle(qualifiedTitle);
            targetTestOutcomes.add(newTestOutcome);
        }
        return targetTestOutcomes;
    }

    private int countFirstRound(List steps) {
        int firstRoundStepCount = 0;
        String firstStepName = steps.get(0).getDescription();
        for(TestStep step : steps) {
            if (step.getDescription().equals(firstStepName)) {
                firstRoundStepCount++;
            } else {
                break;
            }
        }
        return firstRoundStepCount;
    }

}