[test][int][kubeflow] Cleanup resource flavors (kubernetes-sigs#3178)

KunWuLuan · Oct 2, 2024 · 1b0c05c · 1b0c05c
1 parent 1247fa3
commit 1b0c05c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/integration/controller/jobs/kubeflow/kubeflowjob.go b/test/integration/controller/jobs/kubeflow/kubeflowjob.go
@@ -124,6 +124,10 @@ func ShouldReconcileJob(ctx context.Context, k8sClient client.Client, job, creat
 	gomega.Expect(k8sClient.Create(ctx, onDemandFlavor)).Should(gomega.Succeed())
 	spotFlavor := testing.MakeResourceFlavor("spot").NodeLabel(instanceKey, "spot").Obj()
 	gomega.Expect(k8sClient.Create(ctx, spotFlavor)).Should(gomega.Succeed())
+	defer func() {
+		util.ExpectObjectToBeDeleted(ctx, k8sClient, onDemandFlavor, true)
+		util.ExpectObjectToBeDeleted(ctx, k8sClient, spotFlavor, true)
+	}()
 	clusterQueue := testing.MakeClusterQueue("cluster-queue").
 		ResourceGroup(
 			*testing.MakeFlavorQuotas("on-demand").Resource(corev1.ResourceCPU, "5").Obj(),