test: make tick time configurable to reduce test duration

camunda · Dec 8, 2022 · 07cd4a6 · 07cd4a6
1 parent d591077
commit 07cd4a6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/go-chaos/internal/pods.go b/go-chaos/internal/pods.go
@@ -121,12 +121,12 @@ func (c K8Client) RestartPodWithGracePeriod(podName string, gracePeriodSec *int6
 }
 
 func (c K8Client) AwaitReadiness() error {
-	return c.AwaitReadinessWithTimeout(5 * time.Minute)
+	return c.AwaitReadinessWithTimeout(5*time.Minute, 1*time.Second)
 }
 
-func (c K8Client) AwaitReadinessWithTimeout(timeout time.Duration) error {
+func (c K8Client) AwaitReadinessWithTimeout(timeout time.Duration, tickTime time.Duration) error {
 	timedOut := time.After(timeout)
-	ticker := time.Tick(1 * time.Second)
+	ticker := time.Tick(tickTime)
 
 	// Keep checking until we're timed out
 	for {

diff --git a/go-chaos/internal/pods_test.go b/go-chaos/internal/pods_test.go
@@ -283,7 +283,7 @@ func Test_ShouldReturnTrueWhenBrokersAreReady(t *testing.T) {
 	k8Client.CreateDeploymentWithLabelsAndName(t, gatewaySelector, "gateway")
 
 	// when
-	err = k8Client.AwaitReadinessWithTimeout(2 * time.Second)
+	err = k8Client.AwaitReadinessWithTimeout(100*time.Millisecond, 1*time.Millisecond)
 
 	// then
 	require.NoError(t, err)
@@ -304,7 +304,7 @@ func Test_ShouldReturnErrorWhenAtleastOneBrokerIsNotReady(t *testing.T) {
 	k8Client.CreateDeploymentWithLabelsAndName(t, gatewaySelector, "gateway")
 
 	// when
-	err = k8Client.AwaitReadinessWithTimeout(2 * time.Second)
+	err = k8Client.AwaitReadinessWithTimeout(100*time.Millisecond, 1*time.Second)
 
 	// then
 	require.Error(t, err)
@@ -325,7 +325,7 @@ func Test_ShouldReturnErrorWhenAtleastOneBrokerIsNotRunning(t *testing.T) {
 	k8Client.CreateDeploymentWithLabelsAndName(t, gatewaySelector, "gateway")
 
 	// when
-	err = k8Client.AwaitReadinessWithTimeout(2 * time.Second)
+	err = k8Client.AwaitReadinessWithTimeout(100*time.Millisecond, 1*time.Second)
 
 	// then
 	require.Error(t, err)