@@ -577,54 +577,56 @@ func TestCreateOrUpdateAffinityAssistantsAndPVCs_Failure(t *testing.T) {
577
577
name : "pvc creation failed - per workspace" ,
578
578
failureType : "pvc" ,
579
579
aaBehavior : aa .AffinityAssistantPerWorkspace ,
580
- expectedErr : fmt .Errorf ("%w: failed to create PVC pvc-b9eea16dce: error creating persistentvolumeclaims" , ErrPvcCreationFailed ),
580
+ expectedErr : fmt .Errorf ("%w for pvc-b9eea16dce: error creating persistentvolumeclaims" , ErrPvcCreationFailed ),
581
581
}, {
582
582
name : "pvc creation failed - disabled" ,
583
583
failureType : "pvc" ,
584
584
aaBehavior : aa .AffinityAssistantDisabled ,
585
- expectedErr : fmt .Errorf ("%w: failed to create PVC pvc-b9eea16dce: error creating persistentvolumeclaims" , ErrPvcCreationFailed ),
585
+ expectedErr : fmt .Errorf ("%w for pvc-b9eea16dce: error creating persistentvolumeclaims" , ErrPvcCreationFailed ),
586
586
}}
587
587
588
588
for _ , tc := range testCases {
589
- ctx := t .Context ()
590
- kubeClientSet := fakek8s .NewSimpleClientset ()
591
- c := Reconciler {
592
- KubeClientSet : kubeClientSet ,
593
- pvcHandler : volumeclaim .NewPVCHandler (kubeClientSet , zap .NewExample ().Sugar ()),
594
- }
589
+ t .Run (tc .name , func (t * testing.T ) {
590
+ ctx := t .Context ()
591
+ kubeClientSet := fakek8s .NewSimpleClientset ()
592
+ c := Reconciler {
593
+ KubeClientSet : kubeClientSet ,
594
+ pvcHandler : volumeclaim .NewPVCHandler (kubeClientSet , zap .NewExample ().Sugar ()),
595
+ }
595
596
596
- switch tc .failureType {
597
- case "pvc" :
598
- c .KubeClientSet .CoreV1 ().(* fake.FakeCoreV1 ).PrependReactor ("create" , "persistentvolumeclaims" ,
599
- func (action testing2.Action ) (handled bool , ret runtime.Object , err error ) {
600
- return true , & corev1.PersistentVolumeClaim {}, errors .New ("error creating persistentvolumeclaims" )
601
- })
602
- case "statefulset" :
603
- c .KubeClientSet .CoreV1 ().(* fake.FakeCoreV1 ).PrependReactor ("create" , "statefulsets" ,
604
- func (action testing2.Action ) (handled bool , ret runtime.Object , err error ) {
605
- return true , & appsv1.StatefulSet {}, errors .New ("error creating statefulsets" )
606
- })
607
- }
597
+ switch tc .failureType {
598
+ case "pvc" :
599
+ c .KubeClientSet .CoreV1 ().(* fake.FakeCoreV1 ).PrependReactor ("create" , "persistentvolumeclaims" ,
600
+ func (action testing2.Action ) (handled bool , ret runtime.Object , err error ) {
601
+ return true , & corev1.PersistentVolumeClaim {}, errors .New ("error creating persistentvolumeclaims" )
602
+ })
603
+ case "statefulset" :
604
+ c .KubeClientSet .CoreV1 ().(* fake.FakeCoreV1 ).PrependReactor ("create" , "statefulsets" ,
605
+ func (action testing2.Action ) (handled bool , ret runtime.Object , err error ) {
606
+ return true , & appsv1.StatefulSet {}, errors .New ("error creating statefulsets" )
607
+ })
608
+ }
608
609
609
- err := c .createOrUpdateAffinityAssistantsAndPVCs (ctx , testPRWithVolumeClaimTemplate , tc .aaBehavior )
610
+ err := c .createOrUpdateAffinityAssistantsAndPVCs (ctx , testPRWithVolumeClaimTemplate , tc .aaBehavior )
610
611
611
- if err == nil {
612
- t .Errorf ("expect error from createOrUpdateAffinityAssistantsAndPVCs but got nil" )
613
- }
612
+ if err == nil {
613
+ t .Errorf ("expect error from createOrUpdateAffinityAssistantsAndPVCs but got nil" )
614
+ }
614
615
615
- switch tc .failureType {
616
- case "pvc" :
617
- if ! errors .Is (err , ErrPvcCreationFailed ) {
618
- t .Errorf ("expected err type mismatching, expecting %v but got: %v" , ErrPvcCreationFailed , err )
616
+ switch tc .failureType {
617
+ case "pvc" :
618
+ if ! errors .Is (err , ErrPvcCreationFailed ) {
619
+ t .Errorf ("expected err type mismatching, expecting %v but got: %v" , ErrPvcCreationFailed , err )
620
+ }
621
+ case "statefulset" :
622
+ if ! errors .Is (err , ErrAffinityAssistantCreationFailed ) {
623
+ t .Errorf ("expected err type mismatching, expecting %v but got: %v" , ErrAffinityAssistantCreationFailed , err )
624
+ }
619
625
}
620
- case "statefulset" :
621
- if ! errors .Is (err , ErrAffinityAssistantCreationFailed ) {
622
- t .Errorf ("expected err type mismatching, expecting %v but got: %v" , ErrAffinityAssistantCreationFailed , err )
626
+ if d := cmp .Diff (tc .expectedErr .Error (), err .Error ()); d != "" {
627
+ t .Errorf ("expected err mismatching: %v" , diff .PrintWantGot (d ))
623
628
}
624
- }
625
- if d := cmp .Diff (tc .expectedErr .Error (), err .Error ()); d != "" {
626
- t .Errorf ("expected err mismatching: %v" , diff .PrintWantGot (d ))
627
- }
629
+ })
628
630
}
629
631
}
630
632
0 commit comments