From 11fc8b06ab9c3b8043acde228f2ad35cb47c48d7 Mon Sep 17 00:00:00 2001
From: Ruobing1997 <ruobing2@andrew.cmu.edu>
Date: Sat, 28 Mar 2026 22:35:38 -0700
Subject: [PATCH] tiny fix to prevent NPE

---
 .../test/java/org/apache/fluss/flink/sink/FlussSinkITCase.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/fluss-flink/fluss-flink-common/src/test/java/org/apache/fluss/flink/sink/FlussSinkITCase.java b/fluss-flink/fluss-flink-common/src/test/java/org/apache/fluss/flink/sink/FlussSinkITCase.java
index 1a5e98eebb..7e3fabbfb7 100644
--- a/fluss-flink/fluss-flink-common/src/test/java/org/apache/fluss/flink/sink/FlussSinkITCase.java
+++ b/fluss-flink/fluss-flink-common/src/test/java/org/apache/fluss/flink/sink/FlussSinkITCase.java
@@ -382,12 +382,13 @@ public void testPartialUpdateWithTwoWriters() throws Exception {
             for (TableBucket bucket : scanRecords.buckets()) {
                 for (ScanRecord record : scanRecords.records(bucket)) {
                     InternalRow row = record.getRow();
+                    int amount = row.isNullAt(2) ? -1 : row.getInt(2);
                     String address = row.getString(3) != null ? row.getString(3).toString() : null;
                     TestOrder order =
                             new TestOrder(
                                     row.getLong(0),
                                     row.getLong(1),
-                                    row.getInt(2),
+                                    amount,
                                     address,
                                     toFlinkRowKind(record.getChangeType()));
                     actual.add(order);