diff --git a/src/test/java/software/amazon/documentdb/jdbc/DocumentDbConnectionTest.java b/src/test/java/software/amazon/documentdb/jdbc/DocumentDbConnectionTest.java index a2e2d89b..a01cdcfe 100644 --- a/src/test/java/software/amazon/documentdb/jdbc/DocumentDbConnectionTest.java +++ b/src/test/java/software/amazon/documentdb/jdbc/DocumentDbConnectionTest.java @@ -29,6 +29,7 @@ import software.amazon.documentdb.jdbc.common.test.DocumentDbTestEnvironment; import software.amazon.documentdb.jdbc.common.test.DocumentDbTestEnvironmentFactory; import software.amazon.documentdb.jdbc.common.utilities.SqlError; +import software.amazon.documentdb.jdbc.metadata.DocumentDbMetadataServiceImpl; import software.amazon.documentdb.jdbc.metadata.DocumentDbSchema; import software.amazon.documentdb.jdbc.persist.DocumentDbSchemaReader; import software.amazon.documentdb.jdbc.persist.DocumentDbSchemaWriter; @@ -565,7 +566,7 @@ public void run() { final Instant timeoutTime = Instant.now().plus(timeToWaitSECS, ChronoUnit.SECONDS); DocumentDbConnection connection = null; try { - connection = new DocumentDbConnection(properties); + connection = new DocumentDbConnection(properties, new DocumentDbMetadataServiceImpl()); while (timeoutTime.isAfter(Instant.now())) { connection.isValid(1); TimeUnit.MILLISECONDS.sleep(100);