From e7a64b8ef5ef518a852135d4535ea8d0813249f1 Mon Sep 17 00:00:00 2001 From: Gary Russell Date: Tue, 21 Oct 2014 17:36:59 -0400 Subject: [PATCH] Speed Up Mongo Build When No Mongo @Rule used a 10 second connection timeout. Polishing --- .../integration/mongodb/rules/MongoDbAvailableRule.java | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/spring-integration-mongodb/src/test/java/org/springframework/integration/mongodb/rules/MongoDbAvailableRule.java b/spring-integration-mongodb/src/test/java/org/springframework/integration/mongodb/rules/MongoDbAvailableRule.java index 60edbbac95..fed15f3e79 100644 --- a/spring-integration-mongodb/src/test/java/org/springframework/integration/mongodb/rules/MongoDbAvailableRule.java +++ b/spring-integration-mongodb/src/test/java/org/springframework/integration/mongodb/rules/MongoDbAvailableRule.java @@ -16,7 +16,6 @@ package org.springframework.integration.mongodb.rules; -import com.mongodb.MongoClient; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.junit.rules.MethodRule; @@ -24,6 +23,9 @@ import org.junit.runners.model.FrameworkMethod; import org.junit.runners.model.Statement; import com.mongodb.Mongo; +import com.mongodb.MongoClient; +import com.mongodb.MongoClientOptions; +import com.mongodb.ServerAddress; /** * A {@link MethodRule} implementation that checks for a running MongoDB process. @@ -36,6 +38,7 @@ public final class MongoDbAvailableRule implements MethodRule { private final Log logger = LogFactory.getLog(this.getClass()); + @Override public Statement apply(final Statement base, final FrameworkMethod method, final Object target) { return new Statement() { @Override @@ -43,7 +46,8 @@ public final class MongoDbAvailableRule implements MethodRule { MongoDbAvailable mongoAvailable = method.getAnnotation(MongoDbAvailable.class); if (mongoAvailable != null) { try { - Mongo mongo = new MongoClient(); + MongoClientOptions options = new MongoClientOptions.Builder().connectTimeout(100).build(); + Mongo mongo = new MongoClient(ServerAddress.defaultHost(), options); mongo.getDatabaseNames(); } catch (Exception e) {