@@ -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
524531func TestCustomizePersistence (t * testing.T ) {
0 commit comments