fix test.workspace

kbase · Feb 16, 2024 · e1a3897 · e1a3897
1 parent 4b59fd4
commit e1a3897
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/src/us/kbase/workspace/test/workspace/WorkspaceIntegrationWithGridFSTest.java b/src/us/kbase/workspace/test/workspace/WorkspaceIntegrationWithGridFSTest.java
@@ -32,7 +32,8 @@
 import org.junit.Test;
 import org.slf4j.LoggerFactory;
 
-import com.mongodb.MongoClient;
+import com.mongodb.client.MongoClient;
+import com.mongodb.client.MongoClients;
 import com.mongodb.client.MongoDatabase;
 
 import ch.qos.logback.classic.Level;
@@ -114,7 +115,7 @@ public static void setUpClass() throws Exception {
 		System.out.println("Started test mongo instance at localhost: " + MONGO.getServerPort());
 
 		@SuppressWarnings("resource")
-		final MongoClient mcli = new MongoClient("localhost:" + MONGO.getServerPort());
+		final MongoClient mcli = MongoClients.create("mongodb://localhost:" + MONGO.getServerPort());
 		WSDB = mcli.getDatabase(WSDB_NAME);
 		TYPEDB = mcli.getDatabase(TYPEDB_NAME);
 		TestCommon.destroyDB(WSDB);
@@ -145,8 +146,8 @@ public static void tearDownClass() throws Exception {
 
 	@Before
 	public void clearDB() throws Exception {
-		try (final MongoClient mongoClient = new MongoClient(
-				"localhost:" + MONGO.getServerPort())) {
+		try (final MongoClient mongoClient = MongoClients.create(
+				"mongodb://localhost:" + MONGO.getServerPort())) {
 			TestCommon.destroyDB(mongoClient.getDatabase(WSDB_NAME));
 		}
 	}

diff --git a/src/us/kbase/workspace/test/workspace/WorkspaceTester.java b/src/us/kbase/workspace/test/workspace/WorkspaceTester.java
@@ -99,7 +99,8 @@
 import com.fasterxml.jackson.databind.JsonMappingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.google.common.collect.ImmutableMap;
-import com.mongodb.MongoClient;
+import com.mongodb.client.MongoClient;
+import com.mongodb.client.MongoClients;
 import com.mongodb.client.MongoDatabase;
 
 @RunWith(Parameterized.class)
@@ -173,7 +174,7 @@ public static void tearDownClass() throws Exception {
 
 	@Before
 	public void clearDB() throws Exception {
-		try (final MongoClient cli = new MongoClient("localhost:" + mongo.getServerPort())) {
+		try (final MongoClient cli = MongoClients.create("mongodb://localhost:" + mongo.getServerPort())) {
 			TestCommon.destroyDB(cli.getDatabase(DB_WS_NAME));
 		}
 		ws.setConfig(DynamicConfigUpdate.getDefault());
@@ -204,7 +205,7 @@ public WorkspaceTester(
 		}
 		if (!CONFIGS.containsKey(config)) {
 			@SuppressWarnings("resource")
-			final MongoClient mcli = new MongoClient("localhost:" + mongo.getServerPort());
+			final MongoClient mcli = MongoClients.create("mongodb://localhost:" + mongo.getServerPort());
 			final MongoDatabase wsdb = mcli.getDatabase(DB_WS_NAME);
 			final MongoDatabase tdb = mcli.getDatabase(DB_TYPE_NAME);
 			TestCommon.destroyDB(wsdb);