diff --git a/extensions-core/parquet-extensions/src/test/java/org/apache/druid/data/input/parquet/FlattenSpecParquetReaderTest.java b/extensions-core/parquet-extensions/src/test/java/org/apache/druid/data/input/parquet/FlattenSpecParquetReaderTest.java index 726ecde4adf3..6cc155fc52cc 100644 --- a/extensions-core/parquet-extensions/src/test/java/org/apache/druid/data/input/parquet/FlattenSpecParquetReaderTest.java +++ b/extensions-core/parquet-extensions/src/test/java/org/apache/druid/data/input/parquet/FlattenSpecParquetReaderTest.java @@ -19,6 +19,7 @@ package org.apache.druid.data.input.parquet; +import com.fasterxml.jackson.databind.ObjectMapper; import com.google.common.collect.ImmutableList; import org.apache.druid.data.input.ColumnsFilter; import org.apache.druid.data.input.InputEntityReader; @@ -92,8 +93,9 @@ public void testFlat1NoFlattenSpec() throws IOException schema, flattenSpec ); + ObjectMapper objectMapper = new ObjectMapper(); List sampled = sampleAllRows(reader); - Assert.assertEquals(FLAT_JSON, DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues())); + Assert.assertEquals(objectMapper.readTree(FLAT_JSON), objectMapper.readTree(DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues()))); } @Test @@ -125,8 +127,9 @@ public void testFlat1Autodiscover() throws IOException schema, JSONPathSpec.DEFAULT ); + ObjectMapper objectMapper = new ObjectMapper(); List sampled = sampleAllRows(reader); - Assert.assertEquals(FLAT_JSON, DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues())); + Assert.assertEquals(objectMapper.readTree(FLAT_JSON), objectMapper.readTree(DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues()))); } @Test @@ -166,8 +169,9 @@ public void testFlat1Flatten() throws IOException schema, flattenSpec ); + ObjectMapper objectMapper = new ObjectMapper(); List sampled = sampleAllRows(reader); - Assert.assertEquals(FLAT_JSON, DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues())); + Assert.assertEquals(objectMapper.readTree(FLAT_JSON), objectMapper.readTree(DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues()))); } @Test @@ -204,9 +208,9 @@ public void testFlat1FlattenSelectListItem() throws IOException schema, flattenSpec ); + ObjectMapper objectMapper = new ObjectMapper(); List sampled = sampleAllRows(reader); - - Assert.assertEquals(FLAT_JSON, DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues())); + Assert.assertEquals(objectMapper.readTree(FLAT_JSON), objectMapper.readTree(DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues()))); } @@ -242,8 +246,9 @@ public void testNested1NoFlattenSpec() throws IOException schema, flattenSpec ); + ObjectMapper objectMapper = new ObjectMapper(); List sampled = sampleAllRows(reader); - Assert.assertEquals(NESTED_JSON, DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues())); + Assert.assertEquals(objectMapper.readTree(NESTED_JSON), objectMapper.readTree(DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues()))); } @Test @@ -275,8 +280,9 @@ public void testNested1Autodiscover() throws IOException schema, JSONPathSpec.DEFAULT ); + ObjectMapper objectMapper = new ObjectMapper(); List sampled = sampleAllRows(reader); - Assert.assertEquals(NESTED_JSON, DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues())); + Assert.assertEquals(objectMapper.readTree(NESTED_JSON), objectMapper.readTree(DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues()))); } @Test @@ -318,8 +324,9 @@ public void testNested1Flatten() throws IOException schema, flattenSpec ); + ObjectMapper objectMapper = new ObjectMapper(); List sampled = sampleAllRows(reader); - Assert.assertEquals(NESTED_JSON, DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues())); + Assert.assertEquals(objectMapper.readTree(NESTED_JSON), objectMapper.readTree(DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues()))); } @Test @@ -359,8 +366,10 @@ public void testNested1FlattenSelectListItem() throws IOException schema, flattenSpec ); + ObjectMapper objectMapper = new ObjectMapper(); List sampled = sampleAllRows(reader); - Assert.assertEquals(NESTED_JSON, DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues())); + Assert.assertEquals(objectMapper.readTree(NESTED_JSON), objectMapper.readTree(DEFAULT_JSON_WRITER.writeValueAsString(sampled.get(0).getRawValues()))); + } @Test