diff --git a/airbyte-tests/src/automaticMigrationAcceptanceTest/java/io/airbyte/test/automaticMigrationAcceptance/MigrationAcceptanceTest.java b/airbyte-tests/src/automaticMigrationAcceptanceTest/java/io/airbyte/test/automaticMigrationAcceptance/MigrationAcceptanceTest.java index 13b67b4eabc22..9b1297b78d687 100644 --- a/airbyte-tests/src/automaticMigrationAcceptanceTest/java/io/airbyte/test/automaticMigrationAcceptance/MigrationAcceptanceTest.java +++ b/airbyte-tests/src/automaticMigrationAcceptanceTest/java/io/airbyte/test/automaticMigrationAcceptance/MigrationAcceptanceTest.java @@ -44,6 +44,7 @@ import org.junit.jupiter.api.Test; import org.slf4j.Logger; import org.slf4j.LoggerFactory; +import org.testcontainers.utility.ComparableVersion; /** * This class contains an e2e test simulating what a user encounter when trying to upgrade Airybte. @@ -242,11 +243,9 @@ private static void assertDestinationDefinitionInformation(final ApiClient apiCl foundLocalCSVDestinationDefinition = true; } case "424892c4-daac-4491-b35d-c6688ba547ba" -> { - final String[] tagBrokenAsArray = destinationDefinitionRead.getDockerImageTag().replace(".", ",").split(","); - assertEquals(3, tagBrokenAsArray.length); - assertTrue(Integer.parseInt(tagBrokenAsArray[0]) >= 0); - assertTrue(Integer.parseInt(tagBrokenAsArray[1]) >= 3); - assertTrue(Integer.parseInt(tagBrokenAsArray[2]) >= 9); + final String tag = destinationDefinitionRead.getDockerImageTag(); + final ComparableVersion version = new ComparableVersion(tag); + assertTrue(version.compareTo(new ComparableVersion("0.3.9")) >= 0); assertTrue(destinationDefinitionRead.getName().contains("Snowflake")); foundSnowflakeDestinationDefinition = true; }