[FLINK-30257] fix SqlClientITCase#testMatchRecognize

apache · Jan 17, 2023 · f981dd0 · f981dd0
1 parent ae8de97
commit f981dd0
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/flink-end-to-end-tests/flink-sql-client-test/src/test/java/SqlClientITCase.java b/flink-end-to-end-tests/flink-sql-client-test/src/test/java/SqlClientITCase.java
@@ -290,10 +290,15 @@ private void verifyNumberOfResultRecords(String resultFilePath, int expectedNumb
         File tempOutputFile = new File(tempDir, "records.out");
         String tempOutputFilepath = tempOutputFile.toString();
         GenericContainer<?> taskManager = flink.getTaskManagers().get(0);
-        Thread.sleep(5000); // prevent NotFoundException: Status 404
-        taskManager.copyFileFromContainer(resultFilePath, tempOutputFilepath);
-
-        int numberOfResultRecords = UpsertTestFileUtil.getNumberOfRecords(tempOutputFile);
+        int numberOfResultRecords;
+        while (true) {
+            Thread.sleep(5000); // prevent NotFoundException: Status 404
+            taskManager.copyFileFromContainer(resultFilePath, tempOutputFilepath);
+            numberOfResultRecords = UpsertTestFileUtil.getNumberOfRecords(tempOutputFile);
+            if (numberOfResultRecords == expectedNumberOfRecords) {
+                break;
+            }
+        }
         assertThat(numberOfResultRecords).isEqualTo(expectedNumberOfRecords);
     }