fix case

Author: lxy-9602

paimon-core/src/test/java/org/apache/paimon/mergetree/compact/MergeFunctionTestUtils.java

@@ -107,12 +107,12 @@ public static List<ReusingTestData> getExpectedForAggSum(
                         new ReusingTestData(last.key, last.sequenceNumber, RowKind.INSERT, sum));
             } else {
                 if (group.stream().noneMatch(data -> data.valueKind == RowKind.INSERT)) {
-                    // No insert: fill the pk and left nullable fields to null; sequenceNumber =
-                    // latest
+                    // No insert: initRow now sets all fields including nullable ones,
+                    // so the value is from the last DELETE record
                     ReusingTestData last = group.get(group.size() - 1);
                     expected.add(
                             new ReusingTestData(
-                                    last.key, last.sequenceNumber, RowKind.DELETE, null));
+                                    last.key, last.sequenceNumber, RowKind.DELETE, last.value));
                 } else {
                     RowKind rowKind = null;
                     Long sum = null;
@@ -122,7 +122,7 @@ public static List<ReusingTestData> getExpectedForAggSum(
                             sum = sum == null ? data.value : sum + data.value;
                         } else {
                             rowKind = RowKind.DELETE;
-                            sum = null;
+                            sum = data.value;
                         }
                     }
                     ReusingTestData last = group.get(group.size() - 1);

paimon-core/src/test/java/org/apache/paimon/mergetree/compact/PartialUpdateMergeFunctionTest.java

@@ -919,7 +919,7 @@ public void testInitRowWithNullableFieldOnDelete() {
         RowType rowType =
                 RowType.of(
                         DataTypes.INT().notNull(),
-                        DataTypes.INT(),
+                        DataTypes.INT().notNull(),
                         DataTypes.INT(),
                         DataTypes.INT());
         MergeFunction<KeyValue> func =
@@ -930,9 +930,9 @@ public void testInitRowWithNullableFieldOnDelete() {
         // insert some data first
         add(func, 1, 3, 5, 7);
         // send a DELETE with nullable field as null, triggers initRow
-        add(func, RowKind.DELETE, 1, null, 2, null);
+        add(func, RowKind.DELETE, 1, 2, 2, null);
         // after delete with removeRecordOnDelete, row is re-initialized via initRow
-        validate(func, 1, null, 2, null);
+        validate(func, 1, 2, 2, null);
     }
 
     private void add(MergeFunction<KeyValue> function, Integer... f) {

paimon-core/src/test/java/org/apache/paimon/mergetree/compact/aggregate/AggregateMergeFunctionTest.java

@@ -152,22 +152,23 @@ void testInitRowWithNullableFieldOnDelete() {
                                 RowType.builder()
                                         .fields(
                                                 new DataType[] {
+                                                    DataTypes.INT().notNull(),
                                                     DataTypes.INT().notNull(),
                                                     DataTypes.INT(),
                                                     DataTypes.INT()
                                                 },
-                                                new String[] {"k", "a", "b"})
+                                                new String[] {"k", "a", "b", "c"})
                                         .build(),
                                 Collections.singletonList("k"))
                         .create();
         aggregateFunction.reset();
 
         // insert some data first
-        aggregateFunction.add(value(1, 3, 5));
+        aggregateFunction.add(value(1, 0, 0, 5));
         // send a DELETE with nullable field as null, triggers initRow
-        aggregateFunction.add(deleteValue(1, null, 2));
+        aggregateFunction.add(deleteValue(1, 2, 2, null));
         // after delete with removeRecordOnDelete, row is re-initialized via initRow
-        assertThat(aggregateFunction.getResult().value()).isEqualTo(GenericRow.of(1, null, 2));
+        assertThat(aggregateFunction.getResult().value()).isEqualTo(GenericRow.of(1, 2, 2, null));
     }
 
     private KeyValue value(Integer... values) {