Add function to get Kueue workloads

Author: sutaakar

support/kueue.go

@@ -88,3 +88,26 @@ func CreateKueueLocalQueue(t Test, namespace, clusterQueueName string) *kueuev1b
 
 	return localQueue
 }
+
+func GetKueueWorkloads(t Test, namespace string) []*kueuev1beta1.Workload {
+	t.T().Helper()
+
+	workloads, err := t.Client().Kueue().KueueV1beta1().Workloads(namespace).List(t.Ctx(), metav1.ListOptions{})
+	t.Expect(err).NotTo(gomega.HaveOccurred())
+
+	workloadsp := []*kueuev1beta1.Workload{}
+	for _, v := range workloads.Items {
+		workloadsp = append(workloadsp, &v)
+	}
+
+	return workloadsp
+}
+
+func KueueWorkloadAdmitted(workload *kueuev1beta1.Workload) bool {
+	for _, v := range workload.Status.Conditions {
+		if v.Type == "Admitted" && v.Status == "True" {
+			return true
+		}
+	}
+	return false
+}

support/kueue_test.go

@@ -59,3 +59,27 @@ func TestCreateKueueLocalQueue(t *testing.T) {
 	test.Expect(lq.Namespace).To(gomega.Equal("ns-1"))
 	test.Expect(lq.Spec.ClusterQueue).To(gomega.Equal(kueuev1beta1.ClusterQueueReference("cq-1")))
 }
+
+func TestGetKueueWorkloads(t *testing.T) {
+	test := NewTest(t)
+
+	wl := &kueuev1beta1.Workload{
+		TypeMeta: metav1.TypeMeta{
+			APIVersion: kueuev1beta1.SchemeGroupVersion.String(),
+			Kind:       "Workload",
+		},
+		ObjectMeta: metav1.ObjectMeta{
+			Name: "wl1",
+		},
+	}
+
+	_, err := test.Client().Kueue().KueueV1beta1().Workloads("ns-1").Create(test.ctx, wl, metav1.CreateOptions{})
+	test.Expect(err).To(gomega.BeNil())
+
+	wls := GetKueueWorkloads(test, "ns-1")
+
+	test.Expect(wls).To(gomega.Not(gomega.BeNil()))
+	test.Expect(wls).To(gomega.HaveLen(1))
+	test.Expect(wls[0].Name).To(gomega.Equal("wl1"))
+	test.Expect(wls[0].Namespace).To(gomega.Equal("ns-1"))
+}