From 84bef236981f5e7415c5f472c83c99fab4a7607e Mon Sep 17 00:00:00 2001
From: Nikolay Martynov <mar.kolya@gmail.com>
Date: Wed, 26 Sep 2018 12:55:57 -0400
Subject: [PATCH] Increase timeouts in Couchbase to make tests more stable

---
 .../src/test/groovy/util/AbstractCouchbaseTest.groovy      | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/dd-java-agent/instrumentation/couchbase-2.0/src/test/groovy/util/AbstractCouchbaseTest.groovy b/dd-java-agent/instrumentation/couchbase-2.0/src/test/groovy/util/AbstractCouchbaseTest.groovy
index edf8042b22..95c1324c87 100644
--- a/dd-java-agent/instrumentation/couchbase-2.0/src/test/groovy/util/AbstractCouchbaseTest.groovy
+++ b/dd-java-agent/instrumentation/couchbase-2.0/src/test/groovy/util/AbstractCouchbaseTest.groovy
@@ -115,7 +115,7 @@ abstract class AbstractCouchbaseTest extends AgentTestRunner {
   }
 
   private DefaultCouchbaseEnvironment.Builder envBuilder(BucketSettings bucketSettings) {
-    def timeout = TimeUnit.SECONDS.toMillis(5)
+    def timeout = TimeUnit.SECONDS.toMillis(10)
     return DefaultCouchbaseEnvironment.builder()
       .bootstrapCarrierDirectPort(mock.getCarrierPort(bucketSettings.name()))
       .bootstrapHttpDirectPort(port)
@@ -124,9 +124,14 @@ abstract class AbstractCouchbaseTest extends AgentTestRunner {
       .networkLatencyMetricsCollectorConfig(DefaultLatencyMetricsCollectorConfig.create(0, TimeUnit.DAYS))
       .computationPoolSize(1)
       .connectTimeout(timeout)
+      .disconnectTimeout(timeout)
       .kvTimeout(timeout)
       .managementTimeout(timeout)
       .queryTimeout(timeout)
       .viewTimeout(timeout)
+      .keepAliveTimeout(timeout)
+      .searchTimeout(timeout)
+      .analyticsTimeout(timeout)
+      .socketConnectTimeout(timeout.intValue())
   }
 }