diff --git a/airbyte-integrations/connectors/destination-tidb/src/test/java/io/airbyte/integrations/destination/tidb/TiDBDestinationTest.java b/airbyte-integrations/connectors/destination-tidb/src/test/java/io/airbyte/integrations/destination/tidb/TiDBDestinationTest.java index 6854c84581795..6292108c52736 100644 --- a/airbyte-integrations/connectors/destination-tidb/src/test/java/io/airbyte/integrations/destination/tidb/TiDBDestinationTest.java +++ b/airbyte-integrations/connectors/destination-tidb/src/test/java/io/airbyte/integrations/destination/tidb/TiDBDestinationTest.java @@ -5,6 +5,7 @@ package io.airbyte.integrations.destination.tidb; import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertTrue; import com.fasterxml.jackson.databind.JsonNode; import com.google.common.collect.ImmutableMap; @@ -25,7 +26,9 @@ import io.airbyte.protocol.models.Field; import io.airbyte.protocol.models.JsonSchemaType; import java.time.Instant; +import java.util.HashMap; import java.util.List; +import java.util.Map; import java.util.stream.Collectors; import java.util.stream.IntStream; import org.junit.jupiter.api.AfterEach; @@ -97,10 +100,13 @@ public void sanityTest() throws Exception { final List actualRecords = database.bufferedResultSetQuery( connection -> connection.createStatement().executeQuery("SELECT * FROM public._airbyte_raw_id_and_name;"), JdbcUtils.getDefaultSourceOperations()::rowToJson); - - assertEquals( - expectedRecords.stream().map(AirbyteMessage::getRecord).map(AirbyteRecordMessage::getData).collect(Collectors.toList()), - actualRecords.stream().map(o -> o.get("_airbyte_data").asText()).map(Jsons::deserialize).collect(Collectors.toList())); + final Map expectedRecordsWithId = new HashMap<>(); + expectedRecords.stream().map(AirbyteMessage::getRecord).map(AirbyteRecordMessage::getData) + .forEach(data -> expectedRecordsWithId.put(data.get("id").asInt(), data)); + actualRecords.stream().map(o -> o.get("_airbyte_data").asText()).map(Jsons::deserialize).forEach(actual -> { + assertTrue(expectedRecordsWithId.containsKey(actual.get("id").asInt())); + assertEquals(expectedRecordsWithId.get(actual.get("id").asInt()), actual); + }); } private List getNRecords(final int n) {