diff --git a/src/main/java/teetime/framework/AbstractCompositeStage.java b/src/main/java/teetime/framework/AbstractCompositeStage.java index ab55c4b314ffad52d5fd488d348cff84e63a1e38..342d057e579a3308ac26ac2cdc9691b91c0a46e2 100644 --- a/src/main/java/teetime/framework/AbstractCompositeStage.java +++ b/src/main/java/teetime/framework/AbstractCompositeStage.java @@ -54,6 +54,7 @@ public abstract class AbstractCompositeStage { AbstractRunnableStage runnable = AbstractRunnableStage.create(stage); Thread newThread = new TeeTimeThread(runnable, threadName); stage.setOwningThread(newThread); + stage.setActive(true); } /** diff --git a/src/main/java/teetime/framework/Stage.java b/src/main/java/teetime/framework/Stage.java index 4cc0cf7673b735c4641265ac624e68d0204a1970..4a246b5916a804f42838ceb51a0479a02a6ae884 100644 --- a/src/main/java/teetime/framework/Stage.java +++ b/src/main/java/teetime/framework/Stage.java @@ -50,6 +50,8 @@ public abstract class Stage { /** The owning thread of this stage if this stage is directly executed by a {@link AbstractRunnableStage}, <code>null</code> otherwise. */ private Thread owningThread; + private boolean isActive; + private ConfigurationContext owningContext; ConfigurationContext getOwningContext() { @@ -181,4 +183,12 @@ public abstract class Stage { protected abstract void removeDynamicPort(InputPort<?> inputPort); + public boolean isActive() { + return isActive; + } + + void setActive(final boolean isActive) { + this.isActive = isActive; + } + } diff --git a/src/test/java/teetime/framework/AbstractCompositeStageTest.java b/src/test/java/teetime/framework/AbstractCompositeStageTest.java index 414e0f1cc12b5db28ef8ad298f1529b68fec1fb6..418d999454536be19d70c55b8a3f1a4ac8cbc25d 100644 --- a/src/test/java/teetime/framework/AbstractCompositeStageTest.java +++ b/src/test/java/teetime/framework/AbstractCompositeStageTest.java @@ -27,17 +27,17 @@ public class AbstractCompositeStageTest { @Ignore @Test public void testNestedStages() { - Execution<NestesConfig> exec = new Execution<NestesConfig>(new NestesConfig()); + Execution<NestedConf> exec = new Execution<NestedConf>(new NestedConf()); // assertThat(exec.getConfiguration().getContext().getThreadableStages().size(), is(3)); } - private class NestesConfig extends Configuration { + private class NestedConf extends Configuration { private final InitialElementProducer<Object> init; private final Sink sink; private final TestNestingCompositeStage compositeStage; - public NestesConfig() { + public NestedConf() { init = new InitialElementProducer<Object>(new Object()); sink = new Sink(); compositeStage = new TestNestingCompositeStage();