Skip to content

Commit c967575

Browse files
committed
remove enableNamespaceAsQueue in e2e
1 parent b9ea8d5 commit c967575

2 files changed

Lines changed: 27 additions & 55 deletions

File tree

test/e2e/queue.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ var _ = Describe("Queue E2E Test", func() {
4343
}
4444

4545
spec.name = "q1-qj-1"
46-
spec.queue = "q1"
46+
spec.queue = defaultQueue1
4747
job1 := createJob(context, spec)
4848
err := waitJobReady(context, job1)
4949
Expect(err).NotTo(HaveOccurred())
@@ -58,7 +58,7 @@ var _ = Describe("Queue E2E Test", func() {
5858
}
5959

6060
spec.name = "q2-qj-2"
61-
spec.queue = "q2"
61+
spec.queue = defaultQueue2
6262
job2 := createJob(context, spec)
6363
err = waitTasksReady(context, job2, expected)
6464
Expect(err).NotTo(HaveOccurred())

test/e2e/util.go

Lines changed: 25 additions & 53 deletions
Original file line numberDiff line numberDiff line change
@@ -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

5757
const (
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

6569
func 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

107110
func 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

Comments
 (0)