package org.drools.guvnor.models.testscenarios.backend;
import org.drools.guvnor.models.testscenarios.shared.Scenario;
import org.junit.internal.AssumptionViolatedException;
import org.junit.internal.runners.model.EachTestNotifier;
import org.junit.runner.Description;
import org.junit.runner.Runner;
import org.junit.runner.notification.RunNotifier;
import org.junit.runners.model.InitializationError;
import org.kie.api.runtime.KieSession;
public class ScenarioRunner4JUnit extends Runner {
// The description of the test suite
private Description descr;
// the actual scenario test to be executed
private Scenario scenario;
private KieSession ksession;
public ScenarioRunner4JUnit( Scenario scenario,
KieSession ksession) throws InitializationError {
this.scenario = scenario;
this.ksession = ksession;
this.descr = Description.createSuiteDescription( "Scenario test case" );
this.descr.addChild( Description.createTestDescription( getClass(),
scenario.getName() ) );
}
@Override
public Description getDescription() {
return descr;
}
@Override
public void run( RunNotifier notifier ) {
Description description = descr.getChildren().get( 0 );
EachTestNotifier eachNotifier = new EachTestNotifier( notifier, description );
try {
eachNotifier.fireTestStarted();
ScenarioRunner runner = new ScenarioRunner( ksession );
runner.run( scenario );
if ( !scenario.wasSuccessful() ) {
StringBuilder builder = new StringBuilder();
for ( String message : scenario.getFailureMessages() ) {
builder.append( message ).append( "\n" );
}
eachNotifier.addFailedAssumption( new AssumptionViolatedException( builder.toString() ) );
}
} catch ( Throwable t ) {
eachNotifier.addFailure( t );
} finally {
// has to always be called as per junit docs
eachNotifier.fireTestFinished();
}
}
}