DBZ-8461 Don't shut down engine which is already being shut down

spieps · Nov 28, 2024 · cdb29d3 · cdb29d3
1 parent 4f418cd
commit cdb29d3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/...ctor-postgres/src/test/java/io/debezium/connector/postgresql/RecordsStreamProducerIT.java b/...ctor-postgres/src/test/java/io/debezium/connector/postgresql/RecordsStreamProducerIT.java
@@ -2709,7 +2709,8 @@ public void stopInTheMiddleOfTxAndResume() throws Exception {
             VerifyRecord.isValidInsert(record, PK_FIELD, i + 2);
         }
 
-        stopConnector();
+        waitForEngineShutdown();
+        cleanupTestFwkState();
 
         startConnector(Function.identity(), false);
         consumer.expects(expectSecondRun);

diff --git a/...nnector-sqlserver/src/test/java/io/debezium/connector/sqlserver/SqlServerConnectorIT.java b/...nnector-sqlserver/src/test/java/io/debezium/connector/sqlserver/SqlServerConnectorIT.java
@@ -764,7 +764,8 @@ public void updatePrimaryKeyTwiceWithRestartInMiddleOfTx() throws Exception {
 
         final SourceRecords records1 = consumeRecordsByTopic(14);
 
-        stopConnector();
+        waitForEngineShutdown();
+        cleanupTestFwkState();
 
         start(SqlServerConnector.class, config);
         assertConnectorIsRunning();