Add skip metadata test

snowflakedb · Oct 23, 2024 · ea5505b · ea5505b
1 parent e30c7e0
commit ea5505b
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/...est/java/com/snowflake/kafka/connector/records/IcebergTableStreamingRecordMapperTest.java b/...est/java/com/snowflake/kafka/connector/records/IcebergTableStreamingRecordMapperTest.java
@@ -12,6 +12,7 @@
 import java.util.stream.Stream;
 import net.snowflake.client.jdbc.internal.fasterxml.jackson.core.JsonProcessingException;
 import net.snowflake.client.jdbc.internal.fasterxml.jackson.databind.ObjectMapper;
+import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.Arguments;
 import org.junit.jupiter.params.provider.MethodSource;
@@ -113,6 +114,20 @@ void shouldMapMetadata(String description, SnowflakeTableRow row, Map<String, Ob
     assertThat(resultSchematized.get(Utils.TABLE_COLUMN_METADATA)).isEqualTo(expected);
   }
 
+  @Test
+  void shouldSkipMapMetadata() throws JsonProcessingException {
+    // Given
+    SnowflakeTableRow row = buildRow(primitiveJsonExample);
+
+    // When
+    Map<String, Object> result = mapper.processSnowflakeRecord(row, false, false);
+    Map<String, Object> resultSchematized = mapper.processSnowflakeRecord(row, true, false);
+
+    // Then
+    assertThat(result).doesNotContainKey(Utils.TABLE_COLUMN_METADATA);
+    assertThat(resultSchematized).doesNotContainKey(Utils.TABLE_COLUMN_METADATA);
+  }
+
   @ParameterizedTest(name = "{0}")
   @MethodSource("prepareNoSchematizationData")
   void shouldMapRecord_schematizationDisabled(