Skip to content

Commit

Permalink
chore: initialise timeout factor next to the declaration (elastic#1118)
Browse files Browse the repository at this point in the history
* chore: initialise timeout factor on its own package

* chore: reuse timeout factor from common
  • Loading branch information
mdelapenya committed May 15, 2021
1 parent 8b56088 commit 5fbad34
Show file tree
Hide file tree
Showing 4 changed files with 9 additions and 15 deletions.
1 change: 0 additions & 1 deletion e2e/_suites/fleet/ingest_manager_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,6 @@ func setUpSuite() {
}
common.AgentVersionBase = v

common.TimeoutFactor = shell.GetEnvInteger("TIMEOUT_FACTOR", common.TimeoutFactor)
common.AgentVersion = shell.GetEnv("BEAT_VERSION", common.AgentVersionBase)

// check if version is an alias
Expand Down
7 changes: 1 addition & 6 deletions e2e/_suites/helm/helm_charts_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,10 +37,6 @@ var elasticAPMActive = false

var helmManager helm.Manager

// timeoutFactor a multiplier for the max timeout when doing backoff retries.
// It can be overriden by TIMEOUT_FACTOR env var
var timeoutFactor = 2

//nolint:unused
var kubectlClient kubectl.Kubectl

Expand Down Expand Up @@ -80,7 +76,6 @@ func setupSuite() {
helmVersion = shell.GetEnv("HELM_VERSION", helmVersion)
helmChartVersion = shell.GetEnv("HELM_CHART_VERSION", helmChartVersion)
kubernetesVersion = shell.GetEnv("KUBERNETES_VERSION", kubernetesVersion)
timeoutFactor = shell.GetEnvInteger("TIMEOUT_FACTOR", timeoutFactor)

stackVersion = shell.GetEnv("STACK_VERSION", stackVersion)
v, err := utils.GetElasticArtifactVersion(stackVersion)
Expand Down Expand Up @@ -187,7 +182,7 @@ func (ts *HelmChartTestSuite) aResourceWillExposePods(resourceType string) error
return err
}

maxTimeout := time.Duration(timeoutFactor) * time.Minute
maxTimeout := time.Duration(common.TimeoutFactor) * time.Minute

exp := common.GetExponentialBackOff(maxTimeout)
retryCount := 1
Expand Down
11 changes: 3 additions & 8 deletions e2e/_suites/kubernetes-autodiscover/autodiscover_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,14 +37,6 @@ const defaultBeatVersion = "7.x-SNAPSHOT"
var defaultEventsWaitTimeout = 60 * time.Second
var defaultDeployWaitTimeout = 60 * time.Second

func init() {
// initialise timeout factor
common.TimeoutFactor = shell.GetEnvInteger("TIMEOUT_FACTOR", common.TimeoutFactor)

defaultEventsWaitTimeout = defaultEventsWaitTimeout * time.Duration(common.TimeoutFactor)
defaultDeployWaitTimeout = defaultDeployWaitTimeout * time.Duration(common.TimeoutFactor)
}

type podsManager struct {
kubectl kubernetes.Control
ctx context.Context
Expand Down Expand Up @@ -472,6 +464,9 @@ func InitializeTestSuite(ctx *godog.TestSuiteContext) {
// init logger
config.Init()

defaultEventsWaitTimeout = defaultEventsWaitTimeout * time.Duration(common.TimeoutFactor)
defaultDeployWaitTimeout = defaultDeployWaitTimeout * time.Duration(common.TimeoutFactor)

err := cluster.Initialize(suiteContext, "testdata/kind.yml")
if err != nil {
log.WithError(err).Fatal("Failed to initialize cluster")
Expand Down
5 changes: 5 additions & 0 deletions internal/common/retry.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,12 +8,17 @@ import (
"time"

backoff "github.com/cenkalti/backoff/v4"
"github.com/elastic/e2e-testing/internal/shell"
)

// TimeoutFactor a multiplier for the max timeout when doing backoff retries.
// It can be overriden by TIMEOUT_FACTOR env var
var TimeoutFactor = 3

func init() {
TimeoutFactor = shell.GetEnvInteger("TIMEOUT_FACTOR", TimeoutFactor)
}

// GetExponentialBackOff returns a preconfigured exponential backoff instance
func GetExponentialBackOff(elapsedTime time.Duration) *backoff.ExponentialBackOff {
var (
Expand Down

0 comments on commit 5fbad34

Please sign in to comment.