hanzo-ml/trainer
1
0
Fork 0
mirror of https://github.com/hanzo-ml/trainer.git synced 2026-09-24 00:31:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: deflake test to ensure runtime is created before creating trainjob (#2807)

Browse Source 
Signed-off-by: Tsubasa Nagasawa <[email protected]>
This commit is contained in:
Tsubasa Nagasawa
2025-08-29 06:12:22 +00:00
committed by GitHub
parent 357f0c52aa
commit d05f5d797d
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/integration/controller/clustertrainingruntime_controller_test.go
+3
View File
@@ -107,6 +107,9 @@ var _ = ginkgo.Describe("ClusterTrainingRuntime controller", ginkgo.Ordered, fun
ginkgo.It("ClusterTrainingRuntime can be deleted once referenced TrainJob is deleted", func() {
ginkgo.By("Creating a ClusterTrainingRuntime and TrainJob")
gomega.Expect(k8sClient.Create(ctx, clTrainingRuntime)).Should(gomega.Succeed())
gomega.Eventually(func(g gomega.Gomega) {
g.Expect(k8sClient.Get(ctx, client.ObjectKeyFromObject(clTrainingRuntime), clTrainingRuntime)).Should(gomega.Succeed())
}, util.Timeout, util.Interval).Should(gomega.Succeed())
gomega.Expect(k8sClient.Create(ctx, trainJob)).Should(gomega.Succeed())
ginkgo.By("Checking if the ClusterTrainingRuntime obtains finalizers")
test/integration/controller/trainingruntime_controller_test.go
+3
View File
@@ -99,6 +99,9 @@ var _ = ginkgo.Describe("TrainingRuntime controller", ginkgo.Ordered, func() {
ginkgo.It("TrainingRuntime can be deleted once referenced TrainJob is deleted", func() {
ginkgo.By("Creating a TrainingRuntime and TrainJob")
gomega.Expect(k8sClient.Create(ctx, trainingRuntime)).Should(gomega.Succeed())
gomega.Eventually(func(g gomega.Gomega) {
g.Expect(k8sClient.Get(ctx, client.ObjectKeyFromObject(trainingRuntime), trainingRuntime)).Should(gomega.Succeed())
}, util.Timeout, util.Interval).Should(gomega.Succeed())
gomega.Expect(k8sClient.Create(ctx, trainJob)).Should(gomega.Succeed())
ginkgo.By("Checking if the TrainingRuntime obtains finalizers")