use of teetime.framework.TestConfiguration in project TeeTime by teetime-framework.
the class IntraStageCollectorTest method testVisitedStages.
@Test
public void testVisitedStages() {
TestConfiguration config = new TestConfiguration();
IntraStageCollector collector = new IntraStageCollector(config.init);
Traverser traversor = new Traverser(collector);
traversor.traverse(config.init);
assertThat(collector.getIntraStages(), containsInAnyOrder(config.init, config.f2b, config.distributor));
}
use of teetime.framework.TestConfiguration in project TeeTime by teetime-framework.
the class A1ThreadableStageCollectorTest method testVisit.
@Test
public void testVisit() throws Exception {
TestConfiguration config = new TestConfiguration();
Collection<AbstractStage> startStages = ConfigurationFacade.INSTANCE.getStartStages(config);
A1ThreadableStageCollector stageCollector = new A1ThreadableStageCollector();
Traverser traversor = new Traverser(stageCollector);
for (AbstractStage startStage : startStages) {
traversor.traverse(startStage);
}
Set<AbstractStage> newThreadableStages = stageCollector.getThreadableStages();
assertThat(newThreadableStages, hasSize(4));
}
Aggregations