14895 Update tests instance type to ATLAS (#15434)

airbytehq · Aug 9, 2022 · 8b43a92 · 8b43a92
1 parent 558c50c
commit 8b43a92
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/...rbyte/integrations/destination/mongodb/MongodbDestinationStrictEncryptAcceptanceTest.java b/...rbyte/integrations/destination/mongodb/MongodbDestinationStrictEncryptAcceptanceTest.java
@@ -47,9 +47,8 @@ static void setupConfig() throws IOException {
     final JsonNode credentialsJson = Jsons.deserialize(credentialsJsonString);
 
     final JsonNode instanceConfig = Jsons.jsonNode(ImmutableMap.builder()
-        .put("instance", MongoInstanceType.STANDALONE.getType())
-        .put(JdbcUtils.HOST_KEY, credentialsJson.get(JdbcUtils.HOST_KEY).asText())
-        .put(JdbcUtils.PORT_KEY, credentialsJson.get(JdbcUtils.PORT_KEY).asInt())
+        .put("instance", MongoInstanceType.ATLAS.getType())
+        .put("cluster_url", credentialsJson.get("cluster_url").asText())
         .build());
 
     final JsonNode authConfig = Jsons.jsonNode(ImmutableMap.builder()
@@ -105,11 +104,11 @@ protected List<JsonNode> retrieveRecords(final TestDestinationEnv testEnv,
 
   @Override
   protected void setup(final TestDestinationEnv testEnv) {
-    final String connectionString = String.format("mongodb://%s:%s@%s:%s/%s?authSource=admin&ssl=true",
-        config.get(AUTH_TYPE).get(JdbcUtils.USERNAME_KEY).asText(),
-        config.get(AUTH_TYPE).get(JdbcUtils.PASSWORD_KEY).asText(),
-        config.get(INSTANCE_TYPE).get(JdbcUtils.HOST_KEY).asText(),
-        config.get(INSTANCE_TYPE).get(JdbcUtils.PORT_KEY).asText(),
+    var credentials = String.format("%s:%s@", config.get(AUTH_TYPE).get(JdbcUtils.USERNAME_KEY).asText(),
+        config.get(AUTH_TYPE).get(JdbcUtils.PASSWORD_KEY).asText());
+    final String connectionString = String.format("mongodb+srv://%s%s/%s?retryWrites=true&w=majority&tls=true",
+        credentials,
+        config.get(INSTANCE_TYPE).get("cluster_url").asText(),
         config.get(JdbcUtils.DATABASE_KEY).asText());
 
     mongoDatabase = new MongoDatabase(connectionString, config.get(JdbcUtils.DATABASE_KEY).asText());