|
64 | 64 | import java.nio.ByteOrder; |
65 | 65 | import java.nio.file.Files; |
66 | 66 | import java.time.Duration; |
| 67 | +import java.time.OffsetDateTime; |
| 68 | +import java.time.ZoneOffset; |
67 | 69 | import java.util.ArrayList; |
68 | 70 | import java.util.Arrays; |
69 | 71 | import java.util.Collections; |
@@ -323,7 +325,28 @@ protected void checkDataInIcebergPrimaryKeyTable( |
323 | 325 | for (InternalRow row : expectedRows) { |
324 | 326 | Record record = records.next(); |
325 | 327 | 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))); |
327 | 350 | } |
328 | 351 | assertThat(records.hasNext()).isFalse(); |
329 | 352 | } |
|
0 commit comments