Skip to content

Commit 37749f9

Browse files
committed
[aiconformance]: increase timeout for gang scheduling test
1 parent a34b4a1 commit 37749f9

1 file changed

Lines changed: 11 additions & 12 deletions

File tree

tests/e2e/scenarios/ai-conformance/validators/scheduling-orchestration/gangscheduling_test.go

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -35,23 +35,22 @@ func TestGangScheduling_ViaKueue(t *testing.T) {
3535
h.Skip("Kueue CRDs not found, skipping gang scheduling test via Kueue")
3636
}
3737

38-
h.Logf("# Applying manifest")
38+
h.Logf("# Gang Scheduling (via Kueue)")
3939

40-
jobName := "gangscheduling-kueue"
41-
{
42-
ns := h.TestNamespace()
43-
h.ShellExec(fmt.Sprintf("kubectl apply --namespace %s -f %s", ns, "testdata/gangscheduling/gangscheduling-kueue.yaml"))
44-
}
40+
h.Run("gangscheduling-kueue", func(h *validators.ValidatorHarness) {
41+
jobName := "gangscheduling-kueue"
4542

46-
// Wait for Job completion
47-
{
43+
h.Logf("## Simple gang scheduling test using Kueue")
44+
45+
h.Logf("Creating a Kueue Job that requires gang scheduling")
4846
ns := h.TestNamespace()
49-
h.Logf("# Waiting for Job to complete")
50-
h.ShellExec(fmt.Sprintf("kubectl wait --namespace %s --for=condition=complete job/%s", ns, jobName))
47+
h.ApplyManifest(ns, "testdata/gangscheduling/gangscheduling-kueue.yaml")
5148

52-
}
49+
h.Logf("Waiting for Job to complete")
50+
h.ShellExec(fmt.Sprintf("kubectl wait --namespace %s --for=condition=complete job/%s --timeout=300s", ns, jobName))
5351

54-
h.Success("Gang scheduling via Kueue test completed successfully.")
52+
h.Success("Gang scheduling via Kueue test completed successfully.")
53+
})
5554

5655
h.RecordConformance("schedulingOrchestration/gang_scheduling")
5756
}

0 commit comments

Comments
 (0)