DBZ-2923 Improving test stability
This commit is contained in:
parent
7144943a8f
commit
aa674842a3
@ -6,8 +6,6 @@
|
||||
|
||||
package io.debezium.connector.postgresql;
|
||||
|
||||
import static org.junit.Assert.assertFalse;
|
||||
|
||||
import java.sql.SQLException;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
@ -143,7 +141,8 @@ public void shouldStopOnPostgresFastShutdown() throws Exception {
|
||||
logger.info("Waiting for Postgres to shut down...");
|
||||
waitForPostgresShutdown();
|
||||
|
||||
assertFalse(isStreamingRunning("postgres", TestHelper.TEST_SERVER));
|
||||
logger.info("Waiting for connector to shut down...");
|
||||
waitForConnectorShutdown("postgres", TestHelper.TEST_SERVER);
|
||||
}
|
||||
|
||||
private void waitForPostgresShutdown() {
|
||||
@ -152,5 +151,4 @@ private void waitForPostgresShutdown() {
|
||||
.atMost(60 * TestHelper.waitTimeForRecords(), TimeUnit.SECONDS)
|
||||
.until(() -> !postgresContainer.isRunning());
|
||||
}
|
||||
|
||||
}
|
||||
|
@ -1064,6 +1064,13 @@ public static void waitForStreamingRunning(String connector, String server, Stri
|
||||
.until(() -> isStreamingRunning(connector, server, contextName));
|
||||
}
|
||||
|
||||
public static void waitForConnectorShutdown(String connector, String server) {
|
||||
Awaitility.await()
|
||||
.pollInterval(200, TimeUnit.MILLISECONDS)
|
||||
.atMost(waitTimeForRecords() * 30, TimeUnit.SECONDS)
|
||||
.until(() -> !isStreamingRunning(connector, server));
|
||||
}
|
||||
|
||||
public static boolean isStreamingRunning(String connector, String server) {
|
||||
return isStreamingRunning(connector, server, "streaming");
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user