Updated CreateLocalQueue function parameters to use variadic arguments of type *kueuev1beta1.localQueue

Author: abhijeet-dhumal

support/kueue.go

@@ -18,7 +18,6 @@ package support
 
 import (
 	"github.com/onsi/gomega"
-
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	kueuev1beta1 "sigs.k8s.io/kueue/apis/kueue/v1beta1"
 )
@@ -65,7 +64,25 @@ func CreateKueueClusterQueue(t Test, clusterQueueSpec kueuev1beta1.ClusterQueueS
 	return clusterQueue
 }
 
-func CreateKueueLocalQueue(t Test, namespace, clusterQueueName string, annotations map[string]string) *kueuev1beta1.LocalQueue {
+func LocalQueueWithNamespace(namespace string) LocalQueueOption {
+	return func(lq *kueuev1beta1.LocalQueue) {
+		lq.ObjectMeta.Namespace = namespace
+	}
+}
+
+func LocalQueuewithClusterQueue(clusterQueueName string) LocalQueueOption {
+	return func(lq *kueuev1beta1.LocalQueue) {
+		lq.Spec.ClusterQueue = kueuev1beta1.ClusterQueueReference(clusterQueueName)
+	}
+}
+
+func LocalQueuewithAnnotations(annotations map[string]string) LocalQueueOption {
+	return func(lq *kueuev1beta1.LocalQueue) {
+		lq.ObjectMeta.Annotations = annotations
+	}
+}
+
+func CreateKueueLocalQueue(t Test, options ...LocalQueueOption) *kueuev1beta1.LocalQueue {
 	t.T().Helper()
 
 	localQueue := &kueuev1beta1.LocalQueue{
@@ -75,15 +92,16 @@ func CreateKueueLocalQueue(t Test, namespace, clusterQueueName string, annotatio
 		},
 		ObjectMeta: metav1.ObjectMeta{
 			GenerateName: "lq-",
-			Namespace:    namespace,
-			Annotations:  annotations,
-		},
-		Spec: kueuev1beta1.LocalQueueSpec{
-			ClusterQueue: kueuev1beta1.ClusterQueueReference(clusterQueueName),
 		},
+		Spec: kueuev1beta1.LocalQueueSpec{},
+	}
+
+	//Apply options
+	for _, opt := range options {
+		opt(localQueue)
 	}
 
-	localQueue, err := t.Client().Kueue().KueueV1beta1().LocalQueues(namespace).Create(t.Ctx(), localQueue, metav1.CreateOptions{})
+	localQueue, err := t.Client().Kueue().KueueV1beta1().LocalQueues(localQueue.Namespace).Create(t.Ctx(), localQueue, metav1.CreateOptions{})
 	t.Expect(err).NotTo(gomega.HaveOccurred())
 	t.T().Logf("Created Kueue LocalQueue %s/%s successfully", localQueue.Namespace, localQueue.Name)
 

support/kueue_test.go

@@ -55,7 +55,7 @@ func TestCreateKueueLocalQueue(t *testing.T) {
 	annotations := map[string]string{
 		"kueue.x-k8s.io/default-queue": "true",
 	}
-	lq := CreateKueueLocalQueue(test, "ns-1", "cq-1", annotations)
+	lq := CreateKueueLocalQueue(test, LocalQueueWithNamespace("ns-1"), LocalQueuewithClusterQueue("cq-1"), LocalQueuewithAnnotations(annotations))
 
 	test.Expect(lq).To(gomega.Not(gomega.BeNil()))
 	test.Expect(lq.GenerateName).To(gomega.Equal("lq-"))

support/utils.go

@@ -22,6 +22,7 @@ import (
 	"path"
 
 	"github.com/onsi/gomega"
+	kueuev1beta1 "sigs.k8s.io/kueue/apis/kueue/v1beta1"
 )
 
 func Ptr[T any](v T) *T {
@@ -30,6 +31,8 @@ func Ptr[T any](v T) *T {
 
 type OutputType string
 
+type LocalQueueOption func(*kueuev1beta1.LocalQueue)
+
 const (
 	Log OutputType = "log"
 )