@@ -157,10 +157,8 @@ func TestApplyOperatorOverrides(t *testing.T) {
157157 },
158158 },
159159 {
160- Name : "nova" ,
161- ControllerManager : operatorv1beta1.ContainerSpec {
162- Tolerations : customTolerations ,
163- },
160+ Name : "nova" ,
161+ Tolerations : customTolerations ,
164162 },
165163 }
166164
@@ -363,32 +361,26 @@ func TestTolerationsOverride(t *testing.T) {
363361 {
364362 name : "Add tolerations to empty list" ,
365363 operatorSpec : operatorv1beta1.OperatorSpec {
366- Name : "test-operator" ,
367- ControllerManager : operatorv1beta1.ContainerSpec {
368- Tolerations : testTolerations ,
369- },
364+ Name : "test-operator" ,
365+ Tolerations : testTolerations ,
370366 },
371367 initialTolerations : nil ,
372368 expectedTolerations : testTolerations ,
373369 },
374370 {
375371 name : "No custom tolerations, keep defaults unchanged" ,
376372 operatorSpec : operatorv1beta1.OperatorSpec {
377- Name : "test-operator" ,
378- ControllerManager : operatorv1beta1.ContainerSpec {
379- // No tolerations specified
380- },
373+ Name : "test-operator" ,
374+ // No tolerations specified
381375 },
382376 initialTolerations : defaultTolerations ,
383377 expectedTolerations : defaultTolerations ,
384378 },
385379 {
386380 name : "Merge custom tolerations with defaults (different keys)" ,
387381 operatorSpec : operatorv1beta1.OperatorSpec {
388- Name : "test-operator" ,
389- ControllerManager : operatorv1beta1.ContainerSpec {
390- Tolerations : testTolerations , // Different keys than defaults
391- },
382+ Name : "test-operator" ,
383+ Tolerations : testTolerations , // Different keys than defaults
392384 },
393385 initialTolerations : defaultTolerations ,
394386 expectedTolerations : append (defaultTolerations , testTolerations ... ),
@@ -397,14 +389,12 @@ func TestTolerationsOverride(t *testing.T) {
397389 name : "Override default tolerations (same key)" ,
398390 operatorSpec : operatorv1beta1.OperatorSpec {
399391 Name : "test-operator" ,
400- ControllerManager : operatorv1beta1.ContainerSpec {
401- Tolerations : []corev1.Toleration {
402- {
403- Key : corev1 .TaintNodeNotReady , // "node.kubernetes.io/not-ready", // Same key as default
404- Operator : corev1 .TolerationOpExists ,
405- Effect : corev1 .TaintEffectNoExecute ,
406- TolerationSeconds : ptr.To [int64 ](600 ), // Different value
407- },
392+ Tolerations : []corev1.Toleration {
393+ {
394+ Key : corev1 .TaintNodeNotReady , // "node.kubernetes.io/not-ready", // Same key as default
395+ Operator : corev1 .TolerationOpExists ,
396+ Effect : corev1 .TaintEffectNoExecute ,
397+ TolerationSeconds : ptr.To [int64 ](600 ), // Different value
408398 },
409399 },
410400 },
@@ -428,20 +418,18 @@ func TestTolerationsOverride(t *testing.T) {
428418 name : "Mixed scenario: override one default, add new custom" ,
429419 operatorSpec : operatorv1beta1.OperatorSpec {
430420 Name : "test-operator" ,
431- ControllerManager : operatorv1beta1.ContainerSpec {
432- Tolerations : []corev1.Toleration {
433- {
434- Key : corev1 .TaintNodeNotReady , // "node.kubernetes.io/not-ready", // Override default
435- Operator : corev1 .TolerationOpExists ,
436- Effect : corev1 .TaintEffectNoExecute ,
437- TolerationSeconds : ptr.To [int64 ](300 ),
438- },
439- {
440- Key : "node.example.com/gpu" , // Add new
441- Operator : corev1 .TolerationOpEqual ,
442- Value : "nvidia" ,
443- Effect : corev1 .TaintEffectNoSchedule ,
444- },
421+ Tolerations : []corev1.Toleration {
422+ {
423+ Key : corev1 .TaintNodeNotReady , // "node.kubernetes.io/not-ready", // Override default
424+ Operator : corev1 .TolerationOpExists ,
425+ Effect : corev1 .TaintEffectNoExecute ,
426+ TolerationSeconds : ptr.To [int64 ](300 ),
427+ },
428+ {
429+ Key : "node.example.com/gpu" , // Add new
430+ Operator : corev1 .TolerationOpEqual ,
431+ Value : "nvidia" ,
432+ Effect : corev1 .TaintEffectNoSchedule ,
445433 },
446434 },
447435 },
0 commit comments