[FLINK-30257] fix SqlClientITCase#testMatchRecognize

apache · May 25, 2023 · 853976e · 853976e
1 parent 23030f6
commit 853976e
Showing 1 changed file with 12 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,18 @@ 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(50); // prevent NotFoundException: Status 404
+            try {
+                taskManager.copyFileFromContainer(resultFilePath, tempOutputFilepath);
+                numberOfResultRecords = UpsertTestFileUtil.getNumberOfRecords(tempOutputFile);
+                if (numberOfResultRecords == expectedNumberOfRecords) {
+                    break;
+                }
+            } catch (Exception ignored) {
+            }
+        }
         assertThat(numberOfResultRecords).isEqualTo(expectedNumberOfRecords);
     }