diff --git a/support/kueue.go b/support/kueue.go index 98fafd9..3f582be 100644 --- a/support/kueue.go +++ b/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 +} diff --git a/support/kueue_test.go b/support/kueue_test.go index 04e1124..54f5ea1 100644 --- a/support/kueue_test.go +++ b/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")) +}