From 6948b09611280ef051bf580b960bfca71136926f Mon Sep 17 00:00:00 2001 From: Wencong Liu Date: Tue, 17 Jan 2023 11:47:54 +0800 Subject: [PATCH] [FLINK-30257] fix SqlClientITCase#testMatchRecognize --- .../src/test/java/SqlClientITCase.java | 13 +++++++++---- 1 file changed, 9 insertions(+), 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 index de56f79491ff5d..f37528480f22e3 100644 --- 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); }