[FLINK-24331][tests] Forbid to finish the TestEventSource if it has n…

…ot sent at least one value
mbalassi · Oct 22, 2021 · f070f0d · f070f0d
1 parent 63bd6a3
commit f070f0d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...sts/src/test/java/org/apache/flink/runtime/operators/lifecycle/graph/TestEventSource.java b/...sts/src/test/java/org/apache/flink/runtime/operators/lifecycle/graph/TestEventSource.java
@@ -73,7 +73,8 @@ public void open(Configuration parameters) throws Exception {
  public void run(SourceContext<TestDataElement> ctx) {
  long lastSent = 0;
  while (isRunning) {
- TestCommand cmd = scheduledCommands.poll();
+ // Don't finish the source if it has not sent at least one value.
+ TestCommand cmd = lastSent == 0 ? null : scheduledCommands.poll();
  if (cmd == FINISH_SOURCES) {
  ack(cmd);
  isRunning = false;