Skip to content

Commit 8a8861d

Browse files
committed
Added clearing to unit tets
1 parent cb0c1de commit 8a8861d

1 file changed

Lines changed: 12 additions & 5 deletions

File tree

utils/utils_test.go

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -505,20 +505,27 @@ func TestCustomizePodSpecPriorityClassName(t *testing.T) {
505505

506506
pts, runtime := &corev1.PodTemplateSpec{}, createRuntimeComponent(name, namespace, spec)
507507
CustomizePodSpec(pts, runtime)
508-
nilResult := pts.Spec.PriorityClassName
508+
defaultPriorityClassName := pts.Spec.PriorityClassName
509509

510510
priorityClassName := "high-priority"
511511
spec.PriorityClassName = &priorityClassName
512512
pts, runtime = &corev1.PodTemplateSpec{}, createRuntimeComponent(name, namespace, spec)
513513
CustomizePodSpec(pts, runtime)
514-
setResult := pts.Spec.PriorityClassName
514+
setPriorityClassName := pts.Spec.PriorityClassName
515+
516+
spec.PriorityClassName = nil
517+
pts = &corev1.PodTemplateSpec{}
518+
pts.Spec.PriorityClassName = priorityClassName
519+
runtime = createRuntimeComponent(name, namespace, spec)
520+
CustomizePodSpec(pts, runtime)
521+
clearPriorityClassName := pts.Spec.PriorityClassName
515522

516523
testCPS := []Test{
517-
{"Default PriorityClassName", "", nilResult},
518-
{"Set PriorityClassName", priorityClassName, setResult},
524+
{"Default PriorityClassName (not set)", "", defaultPriorityClassName},
525+
{"Set PriorityClassName", priorityClassName, setPriorityClassName},
526+
{"Clearing PriorityClassName", "", clearPriorityClassName},
519527
}
520528
verifyTests(testCPS, t)
521-
522529
}
523530

524531
func TestCustomizePersistence(t *testing.T) {

0 commit comments

Comments
 (0)