Set timeout and interval to all Eventually tests

Signed-off-by: Yuki Iwai <[email protected]>
kubeflow · Jul 3, 2023 · cab8cf3 · cab8cf3
1 parent b88ed3d
commit cab8cf3
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/pkg/controller.v1/mpi/mpijob_controller_test.go b/pkg/controller.v1/mpi/mpijob_controller_test.go
@@ -17,6 +17,7 @@ package mpi
 import (
 	"context"
 	"fmt"
+	"strings"
 
 	common "github.com/kubeflow/training-operator/pkg/apis/kubeflow.org/v1"
 	. "github.com/onsi/ginkgo/v2"
@@ -27,7 +28,6 @@ import (
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/utils/pointer"
 	ctrl "sigs.k8s.io/controller-runtime"
-	"strings"
 
 	kubeflowv1 "github.com/kubeflow/training-operator/pkg/apis/kubeflow.org/v1"
 	"github.com/kubeflow/training-operator/pkg/util/testutil"

diff --git a/pkg/controller.v1/pytorch/pytorchjob_controller_test.go b/pkg/controller.v1/pytorch/pytorchjob_controller_test.go
@@ -31,7 +31,7 @@ import (
 )
 
 var _ = Describe("PyTorchJob controller", func() {
-	// Define utility constants for object names and testing timeouts/durations and intervals.
+	// Define utility constants for object names.
 	const (
 		expectedPort = int32(8080)
 	)

diff --git a/pkg/controller.v1/tensorflow/job_test.go b/pkg/controller.v1/tensorflow/job_test.go
@@ -648,11 +648,11 @@ var _ = Describe("Test for controller.v1/common", func() {
 				Eventually(func() bool {
 					gotErr := testK8sClient.Get(ctx, client.ObjectKeyFromObject(tc.tfJob), &kubeflowv1.TFJob{})
 					return errors.IsNotFound(gotErr)
-				}).Should(BeTrue())
+				}, testutil.Timeout, testutil.Interval).Should(BeTrue())
 			} else {
 				Eventually(func() error {
 					return testK8sClient.Get(ctx, client.ObjectKeyFromObject(tc.tfJob), &kubeflowv1.TFJob{})
-				}).Should(BeNil())
+				}, testutil.Timeout, testutil.Interval).Should(BeNil())
 			}
 		},
 		Entry("TFJob shouldn't be removed since TTL is nil", &cleanUpCases{
@@ -766,7 +766,7 @@ var _ = Describe("Test for controller.v1/common", func() {
 					svc := &corev1.Service{}
 					Expect(testK8sClient.Get(ctx, client.ObjectKeyFromObject(wantSvc), svc)).Should(Succeed())
 					return svc
-				}).Should(BeComparableTo(wantSvc,
+				}, testutil.Timeout, testutil.Interval).Should(BeComparableTo(wantSvc,
 					cmpopts.IgnoreFields(metav1.ObjectMeta{}, "UID", "ResourceVersion", "Generation", "CreationTimestamp", "ManagedFields")))
 			}
 		},

diff --git a/pkg/controller.v1/tensorflow/pod_test.go b/pkg/controller.v1/tensorflow/pod_test.go
@@ -17,6 +17,7 @@ package tensorflow
 import (
 	"context"
 	"fmt"
+	"os"
 
 	commonv1 "github.com/kubeflow/training-operator/pkg/apis/kubeflow.org/v1"
 	. "github.com/onsi/ginkgo/v2"
@@ -26,7 +27,6 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/apimachinery/pkg/util/uuid"
-	"os"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 
 	kubeflowv1 "github.com/kubeflow/training-operator/pkg/apis/kubeflow.org/v1"