This is an automated email from the ASF dual-hosted git repository. martijnvisser pushed a commit to branch main in repository https://gitbox.apache.org/repos/asf/flink-connector-kudu.git
commit abc4a3b965e54406eaddcf06261116916440c4c5 Author: Ferenc Csaky <ferenc.cs...@pm.me> AuthorDate: Thu Mar 28 15:05:46 2024 +0100 [FLINK-34930] Fix KuduTableSourceITCase --- .../connector/kudu/table/KuduTableSourceITCase.java | 20 ++++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/flink-connector-kudu/src/test/java/org/apache/flink/connector/kudu/table/KuduTableSourceITCase.java b/flink-connector-kudu/src/test/java/org/apache/flink/connector/kudu/table/KuduTableSourceITCase.java index 2468e77..7971bda 100644 --- a/flink-connector-kudu/src/test/java/org/apache/flink/connector/kudu/table/KuduTableSourceITCase.java +++ b/flink-connector-kudu/src/test/java/org/apache/flink/connector/kudu/table/KuduTableSourceITCase.java @@ -23,6 +23,7 @@ import org.apache.flink.table.api.TableEnvironment; import org.apache.flink.types.Row; import org.apache.flink.util.CloseableIterator; +import org.junit.jupiter.api.AfterEach; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; @@ -36,9 +37,11 @@ public class KuduTableSourceITCase extends KuduTestBase { private TableEnvironment tableEnv; private KuduCatalog catalog; + private KuduTableInfo tableInfo = null; + @BeforeEach - public void init() { - KuduTableInfo tableInfo = booksTableInfo("books", true); + void init() { + tableInfo = booksTableInfo("books", true); setUpDatabase(tableInfo); tableEnv = KuduTableTestUtils.createTableEnvWithBlinkPlannerBatchMode(); catalog = new KuduCatalog(getMasterAddress()); @@ -46,6 +49,14 @@ public class KuduTableSourceITCase extends KuduTestBase { tableEnv.useCatalog("kudu"); } + @AfterEach + void cleanup() { + if (tableInfo != null) { + cleanDatabase(tableInfo); + tableInfo = null; + } + } + @Test void testFullBatchScan() throws Exception { CloseableIterator<Row> it = @@ -53,7 +64,8 @@ public class KuduTableSourceITCase extends KuduTestBase { List<Row> results = new ArrayList<>(); it.forEachRemaining(results::add); assertEquals(5, results.size()); - assertEquals("1001,Java for dummies,Tan Ah Teck,11.11,11", results.get(0).toString()); + assertEquals( + "+I[1001, Java for dummies, Tan Ah Teck, 11.11, 11]", results.get(0).toString()); tableEnv.executeSql("DROP TABLE books"); } @@ -68,7 +80,7 @@ public class KuduTableSourceITCase extends KuduTestBase { List<Row> results = new ArrayList<>(); it.forEachRemaining(results::add); assertEquals(1, results.size()); - assertEquals("More Java for more dummies", results.get(0).toString()); + assertEquals("+I[More Java for more dummies]", results.get(0).toString()); tableEnv.executeSql("DROP TABLE books"); } }