[FLINK-30257] fix SqlClientITCase#testMatchRecognize

apache · Feb 10, 2023 · 6948b09 · 6948b09
1 parent ae89c99
commit 6948b09
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(50); // prevent NotFoundException: Status 404
+            taskManager.copyFileFromContainer(resultFilePath, tempOutputFilepath);
+            numberOfResultRecords = UpsertTestFileUtil.getNumberOfRecords(tempOutputFile);
+            if (numberOfResultRecords == expectedNumberOfRecords) {
+                break;
+            }
+        }
         assertThat(numberOfResultRecords).isEqualTo(expectedNumberOfRecords);
     }