From a7282778ceedecedccb298dbf7f87ad88e7facf9 Mon Sep 17 00:00:00 2001
From: Bernardo Pastorelli <13519917+randomswdev@users.noreply.github.com>
Date: Thu, 17 Oct 2024 14:31:19 +0200
Subject: [PATCH] Allow customizing the test server's API timeout

---
 sdk/testutil/retry/retryer.go | 5 +++++
 sdk/testutil/server.go        | 7 +++++--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/sdk/testutil/retry/retryer.go b/sdk/testutil/retry/retryer.go
index 45f41988f2..ba27056473 100644
--- a/sdk/testutil/retry/retryer.go
+++ b/sdk/testutil/retry/retryer.go
@@ -34,3 +34,8 @@ func TwoSeconds() *Timer {
 func ThreeTimes() *Counter {
 	return &Counter{Count: 3, Wait: 25 * time.Millisecond}
 }
+
+// Seconds repeats an operation for the number of input seconds and waits 25ms in between.
+func Seconds(secs time.Duration) *Timer {
+	return &Timer{Timeout: secs * time.Second, Wait: 25 * time.Millisecond}
+}
diff --git a/sdk/testutil/server.go b/sdk/testutil/server.go
index a9da8f3727..e8c3d21cd6 100644
--- a/sdk/testutil/server.go
+++ b/sdk/testutil/server.go
@@ -31,10 +31,11 @@ import (
 	"testing"
 	"time"
 
+	"github.com/pkg/errors"
+
 	"github.com/hashicorp/go-cleanhttp"
 	"github.com/hashicorp/go-uuid"
 	"github.com/hashicorp/go-version"
-	"github.com/pkg/errors"
 
 	"github.com/hashicorp/consul/sdk/freeport"
 	"github.com/hashicorp/consul/sdk/testutil/retry"
@@ -128,6 +129,7 @@ type TestServerConfig struct {
 	SkipLeaveOnInt      bool                   `json:"skip_leave_on_interrupt"`
 	Peering             *TestPeeringConfig     `json:"peering,omitempty"`
 	Autopilot           *TestAutopilotConfig   `json:"autopilot,omitempty"`
+	APITimeout          time.Duration          `json:"-"`
 	ReadyTimeout        time.Duration          `json:"-"`
 	StopTimeout         time.Duration          `json:"-"`
 	Stdout              io.Writer              `json:"-"`
@@ -213,6 +215,7 @@ func defaultServerConfig(t TestingTB, consulVersion *version.Version) *TestServe
 			Server:  ports[5],
 			GRPC:    ports[6],
 		},
+		APITimeout:     2 * time.Second,
 		ReadyTimeout:   10 * time.Second,
 		StopTimeout:    10 * time.Second,
 		SkipLeaveOnInt: true,
@@ -470,7 +473,7 @@ func (s *TestServer) waitForAPI() error {
 	// This retry replicates the logic of retry.Run to allow for nested retries.
 	// By returning an error we can wrap TestServer creation with retry.Run
 	// in makeClientWithConfig.
-	timer := retry.TwoSeconds()
+	timer := retry.Seconds(s.Config.APITimeout)
 	deadline := time.Now().Add(timer.Timeout)
 	for !time.Now().After(deadline) {
 		time.Sleep(timer.Wait)