|
2 | 2 |
|
3 | 3 | import apoc.graph.Graphs; |
4 | 4 | import apoc.util.TestUtil; |
| 5 | +import com.mongodb.ConnectionString; |
| 6 | +import com.mongodb.MongoClientSettings; |
5 | 7 | import com.mongodb.client.MongoClient; |
6 | 8 | import com.mongodb.client.MongoClients; |
7 | 9 | import com.mongodb.client.MongoCollection; |
|
33 | 35 | import java.util.Date; |
34 | 36 | import java.util.Map; |
35 | 37 | import java.util.Set; |
| 38 | +import java.util.concurrent.TimeUnit; |
36 | 39 | import java.util.function.Consumer; |
37 | 40 |
|
38 | 41 | import static apoc.mongodb.MongoDBColl.ERROR_MESSAGE; |
@@ -72,7 +75,16 @@ static void beforeClassCommon(MongoVersion mongoVersion) throws Exception { |
72 | 75 | final String host = mongo.getHost(); |
73 | 76 | final Integer port = mongo.getMappedPort(MONGO_DEFAULT_PORT); |
74 | 77 | final String format = String.format("mongodb://admin:pass@%s:%s", host, port); |
75 | | - try (MongoClient mongoClient = MongoClients.create(format)) { |
| 78 | + |
| 79 | + MongoClientSettings settings = MongoClientSettings.builder() |
| 80 | + .applyConnectionString(new ConnectionString(format)) |
| 81 | + .applyToConnectionPoolSettings(builder -> builder |
| 82 | + .maxSize(10) // Reduce max pool size for testing purpose |
| 83 | + .maxConnectionIdleTime(10, TimeUnit.SECONDS) // Faster idle cleanup |
| 84 | + ) |
| 85 | + .build(); |
| 86 | + |
| 87 | + try (MongoClient mongoClient = MongoClients.create(settings)) { |
76 | 88 | String uriPrefix = String.format("mongodb://admin:pass@%s:%s", mongo.getContainerIpAddress(), mongo.getMappedPort(MONGO_DEFAULT_PORT)); |
77 | 89 | PERSON_URI = uriPrefix + "/test.person?authSource=admin"; |
78 | 90 | TEST_URI = uriPrefix + "/test.test?authSource=admin"; |
|
0 commit comments