CBG-4444 create an env var override for bucket op timeout

Author: torcolvin

base/main_test_bucket_pool_config.go

@@ -58,6 +58,9 @@ const (
 
 	// Creates buckets with a specific number of number of replicas
 	tbpEnvBucketNumReplicas = "SG_TEST_BUCKET_NUM_REPLICAS"
+
+	// bucket op timeout is the number of seconds that kv operations will retry. Increasing this can be necessary when flushing buckets rapidly.
+	tbpEnvBucketOpTimeout = "SG_TEST_BUCKET_OP_TIMEOUT"
 )
 
 // TestsUseNamedCollections returns true if the tests use named collections.

base/main_test_cluster.go

@@ -11,6 +11,8 @@ package base
 import (
 	"context"
 	"fmt"
+	"os"
+	"strconv"
 	"strings"
 	"time"
 
@@ -54,11 +56,17 @@ func newTestCluster(ctx context.Context, server string, tbp *TestBucketPool) *tb
 // getGocbClusterForTest makes cluster connection. Callers must close. Returns the cluster and the connection string used to connect.
 func getGocbClusterForTest(ctx context.Context, server string) (*gocb.Cluster, string) {
 
-	testClusterTimeout := 10 * time.Second
 	spec := BucketSpec{
-		Server:          server,
-		TLSSkipVerify:   true,
-		BucketOpTimeout: &testClusterTimeout,
+		Server:        server,
+		TLSSkipVerify: true,
+	}
+	bucketOpTimeout := os.Getenv(tbpEnvBucketOpTimeout)
+	if bucketOpTimeout != "" {
+		secs, err := strconv.Atoi(bucketOpTimeout)
+		if err != nil {
+			FatalfCtx(ctx, "Couldn't parse %s: %v", tbpEnvBucketOpTimeout, err)
+		}
+		spec.BucketOpTimeout = Ptr(time.Duration(secs) * time.Second)
 	}
 	connStr, err := spec.GetGoCBConnString()
 	if err != nil {