From 58792c4a7c166b8746f5c366f37cfefcfb8866d5 Mon Sep 17 00:00:00 2001
From: Chris Cranford <chris@hibernate.org>
Date: Mon, 31 Jul 2023 03:25:08 -0400
Subject: [PATCH] DBZ-6730 Exit consumption loop if engine has stopped

---
 .../test/java/io/debezium/embedded/AbstractConnectorTest.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/debezium-embedded/src/test/java/io/debezium/embedded/AbstractConnectorTest.java b/debezium-embedded/src/test/java/io/debezium/embedded/AbstractConnectorTest.java
index 91283956f..2e54089cc 100644
--- a/debezium-embedded/src/test/java/io/debezium/embedded/AbstractConnectorTest.java
+++ b/debezium-embedded/src/test/java/io/debezium/embedded/AbstractConnectorTest.java
@@ -542,7 +542,7 @@ protected int consumeRecordsUntil(BiPredicate<Integer, SourceRecord> condition,
         int recordsConsumed = 0;
         int nullReturn = 0;
         boolean isLastRecord = false;
-        while (!isLastRecord) {
+        while (!isLastRecord && engine.isRunning()) {
             SourceRecord record = consumedLines.poll(pollTimeoutInMs, TimeUnit.MILLISECONDS);
             if (record != null) {
                 nullReturn = 0;