[HUDI-4490] Make AWSDmsAvroPayload class backwards compatible (#6229)

Co-authored-by: Rahil Chertara <rchertar@amazon.com>

hudi-common/src/main/java/org/apache/hudi/common/model/AWSDmsAvroPayload.java

@@ -18,11 +18,10 @@
 
 package org.apache.hudi.common.model;
 
-import org.apache.hudi.common.util.Option;
-
 import org.apache.avro.Schema;
 import org.apache.avro.generic.GenericRecord;
 import org.apache.avro.generic.IndexedRecord;
+import org.apache.hudi.common.util.Option;
 
 import java.io.IOException;
 import java.util.Properties;

hudi-common/src/test/java/org/apache/hudi/common/model/TestAWSDmsAvroPayload.java

@@ -0,0 +1,129 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.hudi.common.model;
+
+import org.apache.avro.Schema;
+import org.apache.avro.generic.GenericData;
+import org.apache.avro.generic.GenericRecord;
+import org.apache.avro.generic.IndexedRecord;
+import org.apache.hudi.common.util.Option;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.junit.jupiter.api.Assertions.fail;
+
+public class TestAWSDmsAvroPayload {
+
+  private static final String AVRO_SCHEMA_STRING = "{\"type\": \"record\","
+      + "\"name\": \"events\"," + "\"fields\": [ "
+      + "{\"name\": \"field1\", \"type\" : \"int\"},"
+      + "{\"name\": \"Op\", \"type\": \"string\"}"
+      + "]}";
+
+  @Test
+  public void testInsert() {
+
+    Schema avroSchema = new Schema.Parser().parse(AVRO_SCHEMA_STRING);
+    GenericRecord record = new GenericData.Record(avroSchema);
+    record.put("field1", 0);
+    record.put("Op", "I");
+
+    AWSDmsAvroPayload payload = new AWSDmsAvroPayload(Option.of(record));
+
+    try {
+      Option<IndexedRecord> outputPayload = payload.getInsertValue(avroSchema);
+      assertTrue((int) outputPayload.get().get(0) == 0);
+      assertTrue(outputPayload.get().get(1).toString().equals("I"));
+    } catch (Exception e) {
+      fail("Unexpected exception");
+    }
+
+  }
+
+  @Test
+  public void testUpdate() {
+    Schema avroSchema = new Schema.Parser().parse(AVRO_SCHEMA_STRING);
+    GenericRecord newRecord = new GenericData.Record(avroSchema);
+    newRecord.put("field1", 1);
+    newRecord.put("Op", "U");
+
+    GenericRecord oldRecord = new GenericData.Record(avroSchema);
+    oldRecord.put("field1", 0);
+    oldRecord.put("Op", "I");
+
+    AWSDmsAvroPayload payload = new AWSDmsAvroPayload(Option.of(newRecord));
+
+    try {
+      Option<IndexedRecord> outputPayload = payload.combineAndGetUpdateValue(oldRecord, avroSchema);
+      assertTrue((int) outputPayload.get().get(0) == 1);
+      assertTrue(outputPayload.get().get(1).toString().equals("U"));
+    } catch (Exception e) {
+      fail("Unexpected exception");
+    }
+
+  }
+
+  @Test
+  public void testDelete() {
+    Schema avroSchema = new Schema.Parser().parse(AVRO_SCHEMA_STRING);
+    GenericRecord deleteRecord = new GenericData.Record(avroSchema);
+    deleteRecord.put("field1", 2);
+    deleteRecord.put("Op", "D");
+
+    GenericRecord oldRecord = new GenericData.Record(avroSchema);
+    oldRecord.put("field1", 2);
+    oldRecord.put("Op", "U");
+
+    AWSDmsAvroPayload payload = new AWSDmsAvroPayload(Option.of(deleteRecord));
+
+    try {
+      Option<IndexedRecord> outputPayload = payload.combineAndGetUpdateValue(oldRecord, avroSchema);
+      // expect nothing to be committed to table
+      assertFalse(outputPayload.isPresent());
+    } catch (Exception e) {
+      fail("Unexpected exception");
+    }
+
+  }
+
+  @Test
+  public void testPreCombineWithDelete() {
+    Schema avroSchema = new Schema.Parser().parse(AVRO_SCHEMA_STRING);
+    GenericRecord deleteRecord = new GenericData.Record(avroSchema);
+    deleteRecord.put("field1", 4);
+    deleteRecord.put("Op", "D");
+
+    GenericRecord oldRecord = new GenericData.Record(avroSchema);
+    oldRecord.put("field1", 4);
+    oldRecord.put("Op", "I");
+
+    AWSDmsAvroPayload payload = new AWSDmsAvroPayload(Option.of(deleteRecord));
+    AWSDmsAvroPayload insertPayload = new AWSDmsAvroPayload(Option.of(oldRecord));
+
+    try {
+      OverwriteWithLatestAvroPayload output = payload.preCombine(insertPayload);
+      Option<IndexedRecord> outputPayload = output.getInsertValue(avroSchema);
+      // expect nothing to be committed to table
+      assertFalse(outputPayload.isPresent());
+    } catch (Exception e) {
+      fail("Unexpected exception");
+    }
+  }
+}