[incubator-paimon] branch master updated: [test] Fix test AppendOnlyTableCompaction stock in future get (#1918)

[yuzelin]

This is an automated email from the ASF dual-hosted git repository.

yuzelin pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-paimon.git


The following commit(s) were added to refs/heads/master by this push:
     new b41572b7e [test] Fix test AppendOnlyTableCompaction stock in future 
get (#1918)
b41572b7e is described below

commit b41572b7e83b1555e1535b8a3e4d0608fe1570ff
Author: YeJunHao <41894543+leaves12...@users.noreply.github.com>
AuthorDate: Thu Aug 31 14:05:26 2023 +0800

    [test] Fix test AppendOnlyTableCompaction stock in future get (#1918)
---
 .../apache/paimon/flink/lookup/FileStoreLookupFunctionTest.java   | 8 ++++++++
 .../flink/sink/AppendOnlyTableCompactionWorkerOperatorTest.java   | 3 +++
 2 files changed, 11 insertions(+)

diff --git 
a/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/lookup/FileStoreLookupFunctionTest.java
 
b/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/lookup/FileStoreLookupFunctionTest.java
index c1e4323d1..b0931f23d 100644
--- 
a/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/lookup/FileStoreLookupFunctionTest.java
+++ 
b/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/lookup/FileStoreLookupFunctionTest.java
@@ -36,6 +36,7 @@ import org.apache.paimon.types.DataTypes;
 import org.apache.paimon.types.RowType;
 import org.apache.paimon.utils.TraceableFileIO;
 
+import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.io.TempDir;
@@ -96,6 +97,13 @@ public class FileStoreLookupFunctionTest {
         fileStoreLookupFunction.open(tempDir.toString());
     }
 
+    @AfterEach
+    public void close() throws Exception {
+        if (fileStoreLookupFunction != null) {
+            fileStoreLookupFunction.close();
+        }
+    }
+
     @Test
     public void testLookupScanLeak() throws Exception {
         commit(writeCommit(1));
diff --git 
a/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/sink/AppendOnlyTableCompactionWorkerOperatorTest.java
 
b/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/sink/AppendOnlyTableCompactionWorkerOperatorTest.java
index dbfbd371d..78802519e 100644
--- 
a/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/sink/AppendOnlyTableCompactionWorkerOperatorTest.java
+++ 
b/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/sink/AppendOnlyTableCompactionWorkerOperatorTest.java
@@ -151,6 +151,9 @@ public class AppendOnlyTableCompactionWorkerOperatorTest 
extends TableTestBase {
 
         for (Future<CommitMessage> f : workerOperator.result()) {
             try {
+                if (!f.isDone()) {
+                    break;
+                }
                 CommitMessage commitMessage = f.get();
                 List<DataFileMeta> fileMetas =
                         ((CommitMessageImpl) 
commitMessage).compactIncrement().compactAfter();