Fix task state check in connector integration tests (#426)

michelin · Aug 5, 2024 · 9dfdf79 · 9dfdf79
1 parent a40e7e3
commit 9dfdf79
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/test/java/com/michelin/ns4kafka/integration/ConnectIntegrationTest.java b/src/test/java/com/michelin/ns4kafka/integration/ConnectIntegrationTest.java
@@ -412,7 +412,8 @@ void shouldRestartConnector() throws InterruptedException {
             .spec(Topic.TopicSpec.builder()
                 .partitions(3)
                 .replicationFactor(1)
-                .configs(Map.of("cleanup.policy", "delete",
+                .configs(Map.of(
+                    "cleanup.policy", "delete",
                     "min.insync.replicas", "1",
                     "retention.ms", "60000"))
                 .build())
@@ -449,7 +450,6 @@ void shouldRestartConnector() throws InterruptedException {
                 .body(connector));
 
         forceConnectorSynchronization();
-        waitForConnectorAndTasksToBeInState("ns1-co1", Connector.TaskState.RUNNING);
 
         ChangeConnectorState restartState = ChangeConnectorState.builder()
             .metadata(Metadata.builder()
@@ -469,6 +469,8 @@ void shouldRestartConnector() throws InterruptedException {
 
         assertEquals(HttpStatus.OK, restartResponse.status());
 
+        waitForConnectorAndTasksToBeInState("ns1-co1", Connector.TaskState.RUNNING);
+
         ConnectorStateInfo actual = connectClient
             .toBlocking()
             .retrieve(HttpRequest.GET("/connectors/ns1-co1/status"), ConnectorStateInfo.class);