test: filter out deleted pods when checking workload pod count

Author: 0x5457

internal/controller/tensorfusionworkload_controller_test.go

@@ -260,7 +260,11 @@ var _ = Describe("TensorFusionWorkload Controller", func() {
 				g.Expect(k8sClient.List(ctx, podList,
 					client.InNamespace(key.Namespace),
 					client.MatchingLabels{constants.WorkloadKey: key.Name})).Should(Succeed())
-				g.Expect(podList.Items).Should(HaveLen(1))
+				// Filter out pods that are being deleted
+				activePods := lo.Filter(podList.Items, func(pod corev1.Pod, _ int) bool {
+					return pod.DeletionTimestamp == nil
+				})
+				g.Expect(activePods).Should(HaveLen(1))
 			}, timeout, interval).Should(Succeed())
 
 			// Now check if the pod has the correct GPU