Skip to content

Commit daf8fab

Browse files
committed
nit
1 parent 6ae88d2 commit daf8fab

File tree

1 file changed

+24
-1
lines changed

1 file changed

+24
-1
lines changed

fluss-lake/fluss-lake-iceberg/src/test/java/com/alibaba/fluss/lake/iceberg/testutils/FlinkIcebergTieringTestBase.java

Lines changed: 24 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,8 @@
6464
import java.nio.ByteOrder;
6565
import java.nio.file.Files;
6666
import java.time.Duration;
67+
import java.time.OffsetDateTime;
68+
import java.time.ZoneOffset;
6769
import java.util.ArrayList;
6870
import java.util.Arrays;
6971
import java.util.Collections;
@@ -323,7 +325,28 @@ protected void checkDataInIcebergPrimaryKeyTable(
323325
for (InternalRow row : expectedRows) {
324326
Record record = records.next();
325327
assertThat(record.get(0)).isEqualTo(row.getBoolean(0));
326-
assertThat(record.get(1)).isEqualTo(row.getByte(1));
328+
assertThat(record.get(1)).isEqualTo((int) row.getByte(1));
329+
assertThat(record.get(2)).isEqualTo((int) row.getShort(2));
330+
assertThat(record.get(3)).isEqualTo(row.getInt(3));
331+
assertThat(record.get(4)).isEqualTo(row.getLong(4));
332+
assertThat(record.get(5)).isEqualTo(row.getFloat(5));
333+
assertThat(record.get(6)).isEqualTo(row.getDouble(6));
334+
assertThat(record.get(7)).isEqualTo(row.getString(7).toString());
335+
// Iceberg expects BigDecimal for decimal types.
336+
assertThat(record.get(8)).isEqualTo(row.getDecimal(8, 5, 2).toBigDecimal());
337+
assertThat(record.get(9)).isEqualTo(row.getDecimal(9, 20, 0).toBigDecimal());
338+
assertThat(record.get(10))
339+
.isEqualTo(
340+
OffsetDateTime.ofInstant(
341+
row.getTimestampLtz(10, 3).toInstant(), ZoneOffset.UTC));
342+
assertThat(record.get(11))
343+
.isEqualTo(
344+
OffsetDateTime.ofInstant(
345+
row.getTimestampLtz(11, 6).toInstant(), ZoneOffset.UTC));
346+
assertThat(record.get(12)).isEqualTo(row.getTimestampNtz(12, 6).toLocalDateTime());
347+
assertThat(record.get(13)).isEqualTo(row.getTimestampNtz(13, 6).toLocalDateTime());
348+
// Iceberg's Record interface expects ByteBuffer for binary types.
349+
assertThat(record.get(14)).isEqualTo(ByteBuffer.wrap(row.getBinary(14, 4)));
327350
}
328351
assertThat(records.hasNext()).isFalse();
329352
}

0 commit comments

Comments
 (0)