Skip to content

Commit b955ff5

Browse files
committed
fix test
1 parent 03a20f3 commit b955ff5

File tree

1 file changed

+34
-1
lines changed

1 file changed

+34
-1
lines changed

fluss-flink/fluss-flink-common/src/test/java/org/apache/fluss/flink/utils/FlussRowToFlinkRowConverterTest.java

Lines changed: 34 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,15 @@
1818
package org.apache.fluss.flink.utils;
1919

2020
import org.apache.fluss.client.table.scanner.ScanRecord;
21+
import org.apache.fluss.flink.row.FlinkAsFlussArray;
2122
import org.apache.fluss.record.ChangeType;
2223
import org.apache.fluss.record.LogRecord;
24+
import org.apache.fluss.row.BinaryString;
2325
import org.apache.fluss.row.InternalRow;
2426
import org.apache.fluss.row.indexed.IndexedRow;
2527
import org.apache.fluss.row.indexed.IndexedRowWriter;
2628
import org.apache.fluss.types.DataType;
29+
import org.apache.fluss.types.DataTypes;
2730
import org.apache.fluss.types.RowType;
2831
import org.apache.fluss.utils.DateTimeUtils;
2932

@@ -36,6 +39,7 @@
3639
import java.time.LocalDate;
3740
import java.time.LocalTime;
3841

42+
import static org.apache.fluss.row.BinaryString.fromString;
3943
import static org.apache.fluss.row.TestInternalRowGenerator.createAllRowType;
4044
import static org.apache.fluss.row.TestInternalRowGenerator.createAllTypes;
4145
import static org.apache.fluss.row.indexed.IndexedRowTest.genRecordForAllTypes;
@@ -93,7 +97,36 @@ void testConverter() throws Exception {
9397

9498
assertThat(flinkRow.getTimestamp(17, 1).toString())
9599
.isEqualTo("2023-10-25T12:01:13.182");
96-
assertThat(flinkRow.isNullAt(18)).isTrue();
100+
assertThat(flinkRow.getTimestamp(18, 5).toString())
101+
.isEqualTo("2023-10-25T12:01:13.182");
102+
103+
// array of int
104+
Integer[] array1 =
105+
new FlinkAsFlussArray(flinkRow.getArray(19)).toObjectArray(DataTypes.INT());
106+
assertThat(array1).isEqualTo(new Integer[] {1, 2, 3, 4, 5, -11, null, 444, 102234});
107+
108+
// array of float
109+
Float[] array2 =
110+
new FlinkAsFlussArray(flinkRow.getArray(20)).toObjectArray(DataTypes.FLOAT());
111+
assertThat(array2)
112+
.isEqualTo(
113+
new Float[] {
114+
0.1f, 1.1f, -0.5f, 6.6f, Float.MAX_VALUE, Float.MIN_VALUE
115+
});
116+
117+
// array of string
118+
assertThat(flinkRow.getArray(21).size()).isEqualTo(3);
119+
BinaryString[] stringArray1 =
120+
new FlinkAsFlussArray(flinkRow.getArray(21).getArray(0))
121+
.toObjectArray(DataTypes.STRING());
122+
assertThat(stringArray1)
123+
.isEqualTo(new BinaryString[] {fromString("a"), null, fromString("c")});
124+
assertThat(flinkRow.getArray(21).isNullAt(1)).isTrue();
125+
BinaryString[] stringArray2 =
126+
new FlinkAsFlussArray(flinkRow.getArray(21).getArray(2))
127+
.toObjectArray(DataTypes.STRING());
128+
assertThat(stringArray2)
129+
.isEqualTo(new BinaryString[] {fromString("hello"), fromString("world")});
97130
}
98131
}
99132

0 commit comments

Comments
 (0)