From 3527479cd8bebb3038783502dd5b2c00baa60be0 Mon Sep 17 00:00:00 2001
From: Vojtech Juranek <vjuranek@redhat.com>
Date: Thu, 14 Jul 2022 14:33:16 +0200
Subject: [PATCH] DBZ-5408 Make sure streaming is fully running before dropping
 connection

---
 .../debezium/connector/postgresql/PostgresConnectorIT.java  | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/debezium-connector-postgres/src/test/java/io/debezium/connector/postgresql/PostgresConnectorIT.java b/debezium-connector-postgres/src/test/java/io/debezium/connector/postgresql/PostgresConnectorIT.java
index 1d85aa4c4..6e44bbc53 100644
--- a/debezium-connector-postgres/src/test/java/io/debezium/connector/postgresql/PostgresConnectorIT.java
+++ b/debezium-connector-postgres/src/test/java/io/debezium/connector/postgresql/PostgresConnectorIT.java
@@ -1022,10 +1022,14 @@ public void shouldRecoverFromRetriableException() throws Exception {
 
         assertRecordsFromSnapshot(2, 1, 1);
 
+        // DBZ-5408 make sure regular streaming is fully running
+        TestHelper.execute(INSERT_STMT);
+        assertRecordsAfterInsert(2, 2, 2);
+
         // kill all opened connections to the database
         TestHelper.execute("SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE backend_type='walsender'");
         TestHelper.execute(INSERT_STMT);
-        assertRecordsAfterInsert(2, 2, 2);
+        assertRecordsAfterInsert(2, 3, 3);
     }
 
     @Test