(flink-connector-kudu) 40/44: [FLINK-34930] Fix KuduTableSourceITCase

[martijnvisser]

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");
     }
 }