Skip to content

Commit 426cb39

Browse files
authored
Add test for default queue name injection (#155)
* Add test for default queue name injection * Check user-provided queue name is not overridden by default
1 parent d123c5e commit 426cb39

File tree

2 files changed

+22
-1
lines changed

2 files changed

+22
-1
lines changed

internal/webhook/appwrapper_webhook_test.go

+17
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,23 @@ var _ = Describe("AppWrapper Webhook Tests", func() {
4141
Expect(k8sClient.Delete(ctx, aw)).To(Succeed())
4242
})
4343

44+
It("Default queue name is set", func() {
45+
aw := toAppWrapper(pod(100))
46+
47+
Expect(k8sClient.Create(ctx, aw)).To(Succeed())
48+
Expect(aw.Labels[QueueNameLabel]).Should(BeIdenticalTo(defaultQueueName), "aw should be labeled with the default queue name")
49+
Expect(k8sClient.Delete(ctx, aw)).To(Succeed())
50+
})
51+
52+
It("Provided queue name is not overridden by default queue name", func() {
53+
aw := toAppWrapper(pod(100))
54+
aw.Labels = utilmaps.MergeKeepFirst(map[string]string{QueueNameLabel: userProvidedQueueName}, aw.Labels)
55+
56+
Expect(k8sClient.Create(ctx, aw)).To(Succeed())
57+
Expect(aw.Labels[QueueNameLabel]).Should(BeIdenticalTo(userProvidedQueueName), "queue name should not be overridden")
58+
Expect(k8sClient.Delete(ctx, aw)).To(Succeed())
59+
})
60+
4461
It("User name and ID are set", func() {
4562
aw := toAppWrapper(pod(100))
4663
aw.Labels = utilmaps.MergeKeepFirst(map[string]string{AppWrapperUsernameLabel: "bad", AppWrapperUserIDLabel: "bad"}, aw.Labels)

internal/webhook/suite_test.go

+5-1
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,8 @@ var cancel context.CancelFunc
6262

6363
const limitedUserName = "limited-user"
6464
const limitedUserID = "8da0fcfe-6d7f-4f44-b433-d91d22cc1b8c"
65+
const defaultQueueName = "default-queue"
66+
const userProvidedQueueName = "user-queue"
6567

6668
func TestControllers(t *testing.T) {
6769
RegisterFailHandler(Fail)
@@ -157,7 +159,9 @@ var _ = BeforeSuite(func() {
157159
})
158160
Expect(err).NotTo(HaveOccurred())
159161

160-
err = (&AppWrapperWebhook{Config: config.NewAppWrapperConfig()}).SetupWebhookWithManager(mgr)
162+
conf := config.NewAppWrapperConfig()
163+
conf.QueueName = defaultQueueName // add default queue name
164+
err = (&AppWrapperWebhook{Config: conf}).SetupWebhookWithManager(mgr)
161165
Expect(err).NotTo(HaveOccurred())
162166

163167
//+kubebuilder:scaffold:webhook

0 commit comments

Comments
 (0)