Add optional NIO tweak point in tests

acogoluegnes · acogoluegnes · commit 67bfbb23b9ad · 2017-01-03T15:25:26.000+01:00
diff --git a/src/test/java/com/rabbitmq/client/test/BrokerTestCase.java b/src/test/java/com/rabbitmq/client/test/BrokerTestCase.java
@@ -17,6 +17,7 @@
 package com.rabbitmq.client.test;
 
 import com.rabbitmq.client.*;
+import com.rabbitmq.client.impl.nio.NioParams;
 import com.rabbitmq.tools.Host;
 import org.junit.After;
 import org.junit.Before;
@@ -60,10 +61,17 @@ protected void finished(Description description) {
 
     protected ConnectionFactory newConnectionFactory() {
         ConnectionFactory connectionFactory = TestUtils.connectionFactory();
+        if(TestUtils.USE_NIO) {
+            connectionFactory.setNioParams(nioParams());
+        }
         connectionFactory.setAutomaticRecoveryEnabled(isAutomaticRecoveryEnabled());
         return connectionFactory;
     }
 
+    protected NioParams nioParams() {
+        return new NioParams();
+    }
+
     protected boolean isAutomaticRecoveryEnabled() {
         return true;
     }
diff --git a/src/test/java/com/rabbitmq/client/test/server/PersistenceGuarantees.java b/src/test/java/com/rabbitmq/client/test/server/PersistenceGuarantees.java
@@ -19,6 +19,7 @@
 
 import java.io.IOException;
 
+import com.rabbitmq.client.impl.nio.NioParams;
 import org.junit.Test;
 
 import com.rabbitmq.client.MessageProperties;
@@ -28,6 +29,14 @@ public class PersistenceGuarantees extends BrokerTestCase {
     private static final int COUNT = 10000;
     private String queue;
 
+    @Override
+    protected NioParams nioParams() {
+        NioParams nioParams = super.nioParams();
+        // may need a higher enqueuing timeout on slow environments
+        return nioParams
+            .setWriteEnqueuingTimeoutInMs(nioParams.getWriteEnqueuingTimeoutInMs() * 2);
+    }
+
     protected void declareQueue() throws IOException {
         queue = channel.queueDeclare("", true, false, false, null).getQueue();
     }
