@@ -48,18 +48,22 @@ import (
4848 "volcano.sh/volcano/pkg/controllers/job/state"
4949)
5050
51- var oneMinute = 1 * time . Minute
52-
53- var twoMinute = 2 * time .Minute
54-
55- var oneCPU = v1. ResourceList { "cpu" : resource . MustParse ( "1000m" )}
51+ var (
52+ oneMinute = 1 * time . Minute
53+ twoMinute = 2 * time .Minute
54+ oneCPU = v1. ResourceList { "cpu" : resource . MustParse ( "1000m" )}
55+ )
5656
5757const (
5858 workerPriority = "worker-pri"
5959 masterPriority = "master-pri"
6060 defaultNginxImage = "nginx:1.14"
6161 defaultBusyBoxImage = "busybox:1.24"
6262 defaultMPIImage = "volcanosh/example-mpi:0.0.1"
63+
64+ defaultNamespace = "test"
65+ defaultQueue1 = "q1"
66+ defaultQueue2 = "q2"
6367)
6468
6569func cpuResource (request string ) v1.ResourceList {
@@ -99,15 +103,14 @@ type context struct {
99103 kbclient * kbver.Clientset
100104 vkclient * vkver.Clientset
101105
102- namespace string
103- queues []string
104- enableNamespaceAsQueue bool
106+ namespace string
107+ queues []string
105108}
106109
107110func initTestContext () * context {
108111 cxt := & context {
109- namespace : "test" ,
110- queues : []string {"q1" , "q2" },
112+ namespace : defaultNamespace ,
113+ queues : []string {defaultQueue1 , defaultQueue2 },
111114 }
112115
113116 home := homeDir ()
@@ -129,10 +132,6 @@ func initTestContext() *context {
129132 Expect (err ).NotTo (HaveOccurred (),
130133 "k8s cluster is required to have one ready worker node at least." )
131134
132- //NOTE(tommylikehu):NamespaceAsQueue feature was removed from kube-batch,
133- //we will eventually remove this logic in test as well.
134- cxt .enableNamespaceAsQueue = false
135-
136135 _ , err = cxt .kubeclient .CoreV1 ().Namespaces ().Create (& v1.Namespace {
137136 ObjectMeta : metav1.ObjectMeta {
138137 Name : cxt .namespace ,
@@ -177,12 +176,7 @@ func queueNotExist(ctx *context) wait.ConditionFunc {
177176 return func () (bool , error ) {
178177 for _ , q := range ctx .queues {
179178 var err error
180- if ctx .enableNamespaceAsQueue {
181- _ , err = ctx .kubeclient .CoreV1 ().Namespaces ().Get (q , metav1.GetOptions {})
182- } else {
183- _ , err = ctx .kbclient .SchedulingV1alpha1 ().Queues ().Get (q , metav1.GetOptions {})
184- }
185-
179+ _ , err = ctx .kbclient .SchedulingV1alpha1 ().Queues ().Get (q , metav1.GetOptions {})
186180 if ! (err != nil && errors .IsNotFound (err )) {
187181 return false , err
188182 }
@@ -234,22 +228,14 @@ func createQueues(cxt *context) {
234228 var err error
235229
236230 for _ , q := range cxt .queues {
237- if cxt .enableNamespaceAsQueue {
238- _ , err = cxt .kubeclient .CoreV1 ().Namespaces ().Create (& v1.Namespace {
239- ObjectMeta : metav1.ObjectMeta {
240- Name : q ,
241- },
242- })
243- } else {
244- _ , err = cxt .kbclient .SchedulingV1alpha1 ().Queues ().Create (& kbv1.Queue {
245- ObjectMeta : metav1.ObjectMeta {
246- Name : q ,
247- },
248- Spec : kbv1.QueueSpec {
249- Weight : 1 ,
250- },
251- })
252- }
231+ _ , err = cxt .kbclient .SchedulingV1alpha1 ().Queues ().Create (& kbv1.Queue {
232+ ObjectMeta : metav1.ObjectMeta {
233+ Name : q ,
234+ },
235+ Spec : kbv1.QueueSpec {
236+ Weight : 1 ,
237+ },
238+ })
253239
254240 Expect (err ).NotTo (HaveOccurred ())
255241 }
@@ -259,17 +245,9 @@ func deleteQueues(cxt *context) {
259245 foreground := metav1 .DeletePropagationForeground
260246
261247 for _ , q := range cxt .queues {
262- var err error
263-
264- if cxt .enableNamespaceAsQueue {
265- err = cxt .kubeclient .CoreV1 ().Namespaces ().Delete (q , & metav1.DeleteOptions {
266- PropagationPolicy : & foreground ,
267- })
268- } else {
269- err = cxt .kbclient .SchedulingV1alpha1 ().Queues ().Delete (q , & metav1.DeleteOptions {
270- PropagationPolicy : & foreground ,
271- })
272- }
248+ err := cxt .kbclient .SchedulingV1alpha1 ().Queues ().Delete (q , & metav1.DeleteOptions {
249+ PropagationPolicy : & foreground ,
250+ })
273251
274252 Expect (err ).NotTo (HaveOccurred ())
275253 }
@@ -305,12 +283,6 @@ func getNS(context *context, job *jobSpec) string {
305283 return job .namespace
306284 }
307285
308- if context .enableNamespaceAsQueue {
309- if len (job .queue ) != 0 {
310- return job .queue
311- }
312- }
313-
314286 return context .namespace
315287}
316288
0 commit comments