johnyangk commented on a change in pull request #137: [NEMO-232] Implement InputWatermarkManager URL: https://github.com/apache/incubator-nemo/pull/137#discussion_r229598953
########## File path: runtime/executor/src/test/java/org/apache/nemo/runtime/executor/task/TaskExecutorTest.java ########## @@ -601,6 +600,91 @@ public void close() { } } + /** + * Source vertex for unbounded source test. + */ + private final class TestUnboundedSourceVertex extends SourceVertex { + + @Override + public boolean isBounded() { + return false; + } + + @Override + public List<Readable> getReadables(int desiredNumOfSplits) throws Exception { + return null; + } + + @Override + public void clearInternalStates() { + + } + + @Override + public IRVertex getClone() { + return null; + } + } + + private final class TestUnboundedSourceReadable implements Readable { + int pointer = 0; + final int middle = elements.size() / 2; + final int end = elements.size(); + boolean watermarkEmitted = false; + final List<Long> watermarks; Review comment: Maybe use a queue, and use the queue size instead of `numEmittedWatermarks`? ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services