postgres
diff --git a/‎src/bin/pg_upgrade/t/003_logical_slots.pl
Copy file name to clipboardExpand all lines: src/bin/pg_upgrade/t/003_logical_slots.pl
+6Lines changed: 6 additions & 0 deletions b/‎src/bin/pg_upgrade/t/003_logical_slots.pl
Copy file name to clipboardExpand all lines: src/bin/pg_upgrade/t/003_logical_slots.pl
+6Lines changed: 6 additions & 0 deletions
@@ -163,6 +163,12 @@
 ]);
 $sub->wait_for_subscription_sync($oldpub, 'regress_sub');
 
+# Also wait for two-phase to be enabled
+my $twophase_query =
+  "SELECT count(1) = 0 FROM pg_subscription WHERE subtwophasestate NOT IN ('e');";
+$sub->poll_query_until('postgres', $twophase_query)
+  or die "Timed out while waiting for subscriber to enable twophase";
+
 # 2. Temporarily disable the subscription
 $sub->safe_psql('postgres', "ALTER SUBSCRIPTION regress_sub DISABLE");
 $oldpub->stop;