fix test

memiiso · Jun 11, 2024 · 514864f · 514864f
1 parent a95a120
commit 514864f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/...zium-server-iceberg-sink/src/main/java/io/debezium/server/iceberg/IcebergChangeEvent.java b/...zium-server-iceberg-sink/src/main/java/io/debezium/server/iceberg/IcebergChangeEvent.java
@@ -285,7 +285,7 @@ private static Map.Entry<Integer, Types.NestedField> debeziumFieldToIcebergField
   private static List<Types.NestedField> icebergSchemaFields(JsonNode schemaNode) {
     List<Types.NestedField> schemaColumns = new ArrayList<>();
     AtomicReference<Integer> fieldId = new AtomicReference<>(1);
-    if (schemaNode != null && schemaNode.has("fields") && schemaNode.get("fields").isArray()) {
+    if (schemaNode != null && !schemaNode.isNull() && schemaNode.has("fields") && schemaNode.get("fields").isArray()) {
       LOGGER.debug("Converting iceberg schema to debezium:{}", schemaNode);
       schemaNode.get("fields").forEach(field -> {
         Map.Entry<Integer, Types.NestedField> df = debeziumFieldToIcebergField(field, field.get("field").textValue(), fieldId.get());
@@ -329,7 +329,7 @@ public int hashCode() {
 
     private Schema icebergSchema() {
 
-      if (this.valueSchema == null) {
+      if (this.valueSchema.isNull()) {
         throw new RuntimeException("Failed to get schema from debezium event, event schema is null");
       }
 

diff --git a/...ink/src/test/java/io/debezium/server/iceberg/testresources/IcebergChangeEventBuilder.java b/...ink/src/test/java/io/debezium/server/iceberg/testresources/IcebergChangeEventBuilder.java
@@ -106,8 +106,14 @@ public IcebergChangeEventBuilder addKeyField(String name, String val) {
   public IcebergChangeEvent build() {
     return new IcebergChangeEvent(
         this.destination,
-            payload.asText().getBytes(StandardCharsets.UTF_8),
-            keyPayload.asText().getBytes(StandardCharsets.UTF_8)
+        ("{" +
+            "\"schema\":" + this.valueSchema() + "," +
+            "\"payload\":" + payload.toString() +
+            "} ").getBytes(StandardCharsets.UTF_8),
+        ("{" +
+            "\"schema\":" + this.keySchema() + "," +
+            "\"payload\":" + keyPayload.toString() +
+            "} ").getBytes(StandardCharsets.UTF_8)
     );
   }