You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@bahir.apache.org by es...@apache.org on 2022/07/17 12:29:24 UTC
[bahir-flink] 03/16: feat: add KuduRowDataInputFormatTest
This is an automated email from the ASF dual-hosted git repository.
eskabetxe pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/bahir-flink.git
commit 62600e6681227b03b803b9dfc624bbc53e46e57a
Author: min <40...@users.noreply.github.com>
AuthorDate: Thu May 19 16:49:15 2022 +0800
feat: add KuduRowDataInputFormatTest
---
.../connectors/kudu/connector/KuduTestBase.java | 22 ++++
.../kudu/format/KuduRowDataInputFormatTest.java | 121 +++++++++++++++++++++
2 files changed, 143 insertions(+)
diff --git a/flink-connector-kudu/src/test/java/org/apache/flink/connectors/kudu/connector/KuduTestBase.java b/flink-connector-kudu/src/test/java/org/apache/flink/connectors/kudu/connector/KuduTestBase.java
index d42692e..90b0746 100644
--- a/flink-connector-kudu/src/test/java/org/apache/flink/connectors/kudu/connector/KuduTestBase.java
+++ b/flink-connector-kudu/src/test/java/org/apache/flink/connectors/kudu/connector/KuduTestBase.java
@@ -18,6 +18,7 @@ package org.apache.flink.connectors.kudu.connector;
import org.apache.flink.api.java.tuple.Tuple5;
import org.apache.flink.connectors.kudu.connector.convertor.RowResultRowConvertor;
+import org.apache.flink.connectors.kudu.connector.convertor.RowResultRowDataConvertor;
import org.apache.flink.connectors.kudu.connector.reader.KuduInputSplit;
import org.apache.flink.connectors.kudu.connector.reader.KuduReader;
import org.apache.flink.connectors.kudu.connector.reader.KuduReaderConfig;
@@ -298,6 +299,27 @@ public class KuduTestBase {
return rows;
}
+ protected List<RowData> readRowDatas(KuduTableInfo tableInfo) throws Exception {
+ String masterAddresses = getMasterAddress();
+ KuduReaderConfig readerConfig = KuduReaderConfig.Builder.setMasters(masterAddresses).build();
+ KuduReader<RowData> reader = new KuduReader<>(tableInfo, readerConfig, new RowResultRowDataConvertor());
+
+ KuduInputSplit[] splits = reader.createInputSplits(1);
+ List<RowData> rows = new ArrayList<>();
+ for (KuduInputSplit split : splits) {
+ KuduReaderIterator<RowData> resultIterator = reader.scanner(split.getScanToken());
+ while (resultIterator.hasNext()) {
+ RowData row = resultIterator.next();
+ if (row != null) {
+ rows.add(row);
+ }
+ }
+ }
+ reader.close();
+
+ return rows;
+ }
+
protected void kuduRowsTest(List<Row> rows) {
for (Row row : rows) {
Integer rowId = (Integer) row.getField(0);
diff --git a/flink-connector-kudu/src/test/java/org/apache/flink/connectors/kudu/format/KuduRowDataInputFormatTest.java b/flink-connector-kudu/src/test/java/org/apache/flink/connectors/kudu/format/KuduRowDataInputFormatTest.java
new file mode 100644
index 0000000..aa391fc
--- /dev/null
+++ b/flink-connector-kudu/src/test/java/org/apache/flink/connectors/kudu/format/KuduRowDataInputFormatTest.java
@@ -0,0 +1,121 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.flink.connectors.kudu.format;
+
+import org.apache.flink.connectors.kudu.connector.KuduTableInfo;
+import org.apache.flink.connectors.kudu.connector.KuduTestBase;
+import org.apache.flink.connectors.kudu.connector.convertor.RowResultRowConvertor;
+import org.apache.flink.connectors.kudu.connector.convertor.RowResultRowDataConvertor;
+import org.apache.flink.connectors.kudu.connector.reader.KuduInputSplit;
+import org.apache.flink.connectors.kudu.connector.reader.KuduReaderConfig;
+import org.apache.flink.table.data.GenericRowData;
+import org.apache.flink.table.data.RowData;
+import org.apache.flink.types.Row;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+class KuduRowDataInputFormatTest extends KuduTestBase {
+
+ @Test
+ void testInvalidKuduMaster() {
+ KuduTableInfo tableInfo = booksTableInfo("books", false);
+ Assertions.assertThrows(NullPointerException.class, () -> new KuduRowDataInputFormat(null,
+ new RowResultRowDataConvertor(), tableInfo));
+ }
+
+ @Test
+ void testInvalidTableInfo() {
+ String masterAddresses = getMasterAddress();
+ KuduReaderConfig readerConfig = KuduReaderConfig.Builder.setMasters(masterAddresses).build();
+ Assertions.assertThrows(NullPointerException.class, () -> new KuduRowDataInputFormat(readerConfig,
+ new RowResultRowDataConvertor(), null));
+ }
+
+ @Test
+ void testInputFormat() throws Exception {
+ KuduTableInfo tableInfo = booksTableInfo("books", true);
+ setUpDatabase(tableInfo);
+
+ List<RowData> rows = readRowDatas(tableInfo);
+ Assertions.assertEquals(5, rows.size());
+
+ cleanDatabase(tableInfo);
+ }
+
+ @Test
+ void testInputFormatWithProjection() throws Exception {
+ KuduTableInfo tableInfo = booksTableInfo("books", true);
+ setUpDatabase(tableInfo);
+
+ List<RowData> rows = readRowDatas(tableInfo, "title", "id");
+ Assertions.assertEquals(5, rows.size());
+
+ for (RowData row : rows) {
+ Assertions.assertEquals(2, row.getArity());
+ }
+
+ cleanDatabase(tableInfo);
+ }
+
+ private List<RowData> readRowDatas(KuduTableInfo tableInfo, String... fieldProjection) throws Exception {
+ String masterAddresses = getMasterAddress();
+ KuduReaderConfig readerConfig = KuduReaderConfig.Builder.setMasters(masterAddresses).build();
+ KuduRowDataInputFormat inputFormat = new KuduRowDataInputFormat(readerConfig, new RowResultRowDataConvertor(),
+ tableInfo, new ArrayList<>(), fieldProjection == null ? null : Arrays.asList(fieldProjection));
+
+ KuduInputSplit[] splits = inputFormat.createInputSplits(1);
+ List<RowData> rows = new ArrayList<>();
+ for (KuduInputSplit split : splits) {
+ inputFormat.open(split);
+ while (!inputFormat.reachedEnd()) {
+ RowData row = inputFormat.nextRecord(new GenericRowData(5));
+ if (row != null) {
+ rows.add(row);
+ }
+ }
+ }
+ inputFormat.close();
+
+ return rows;
+ }
+
+ private List<Row> readRows(KuduTableInfo tableInfo, String... fieldProjection) throws Exception {
+ String masterAddresses = getMasterAddress();
+ KuduReaderConfig readerConfig = KuduReaderConfig.Builder.setMasters(masterAddresses).build();
+ KuduRowInputFormat inputFormat = new KuduRowInputFormat(readerConfig, new RowResultRowConvertor(), tableInfo,
+ new ArrayList<>(), fieldProjection == null ? null : Arrays.asList(fieldProjection));
+
+ KuduInputSplit[] splits = inputFormat.createInputSplits(1);
+ List<Row> rows = new ArrayList<>();
+ for (KuduInputSplit split : splits) {
+ inputFormat.open(split);
+ while (!inputFormat.reachedEnd()) {
+ Row row = inputFormat.nextRecord(new Row(5));
+ if (row != null) {
+ rows.add(row);
+ }
+ }
+ }
+ inputFormat.close();
+
+ return rows;
+ }
+}