1717package installer
1818
1919import (
20- "context"
2120 "reflect"
2221 "testing"
2322
@@ -42,7 +41,7 @@ func TestMigrateDrivesError(t *testing.T) {
4241
4342 legacyclient .SetDriveClient (legacyclientsetfake .NewSimpleClientset (drive ))
4443 task := migrateTask {client .GetClient (), legacyclient .GetClient ()}
45- driveMap , legacyDriveErrors , driveErrors , err := task .migrateDrives (context . TODO (), false , nil )
44+ driveMap , legacyDriveErrors , driveErrors , err := task .migrateDrives (t . Context (), false , nil )
4645 if len (driveMap ) == 0 && len (legacyDriveErrors ) == 0 && len (driveErrors ) == 0 && err == nil {
4746 t .Fatalf ("expected error, but succeeded\n " )
4847 }
@@ -56,7 +55,7 @@ func TestMigrateDrivesError(t *testing.T) {
5655 },
5756 }
5857 legacyclient .SetDriveClient (legacyclientsetfake .NewSimpleClientset (drive ))
59- driveMap , legacyDriveErrors , driveErrors , err = task .migrateDrives (context . TODO (), false , nil )
58+ driveMap , legacyDriveErrors , driveErrors , err = task .migrateDrives (t . Context (), false , nil )
6059 if len (driveMap ) == 0 && len (legacyDriveErrors ) == 0 && len (driveErrors ) == 0 && err == nil {
6160 t .Fatalf ("expected error, but succeeded\n " )
6261 }
@@ -79,7 +78,7 @@ func TestMigrateDrivesError(t *testing.T) {
7978 },
8079 }
8180 legacyclient .SetDriveClient (legacyclientsetfake .NewSimpleClientset (drive1 , drive2 ))
82- driveMap , legacyDriveErrors , driveErrors , err = task .migrateDrives (context . TODO (), false , nil )
81+ driveMap , legacyDriveErrors , driveErrors , err = task .migrateDrives (t . Context (), false , nil )
8382 if len (driveMap ) == 0 && len (legacyDriveErrors ) == 0 && len (driveErrors ) == 0 && err == nil {
8483 t .Fatalf ("expected error, but succeeded\n " )
8584 }
@@ -90,11 +89,11 @@ func TestMigrateNoDrives(t *testing.T) {
9089 clientset := types .NewExtFakeClientset (clientsetfake .NewSimpleClientset ())
9190 client .SetDriveInterface (clientset .DirectpvLatest ().DirectPVDrives ())
9291 task := migrateTask {client .GetClient (), legacyclient .GetClient ()}
93- _ , legacyDriveErrors , driveErrors , err := task .migrateDrives (context . TODO (), false , nil )
92+ _ , legacyDriveErrors , driveErrors , err := task .migrateDrives (t . Context (), false , nil )
9493 if len (legacyDriveErrors ) != 0 || len (driveErrors ) != 0 || err != nil {
9594 t .Fatalf ("unexpected error; %v\n " , err )
9695 }
97- driveList , err := client .DriveClient ().List (context . TODO (), metav1.ListOptions {})
96+ driveList , err := client .DriveClient ().List (t . Context (), metav1.ListOptions {})
9897 if err != nil && ! apierrors .IsNotFound (err ) {
9998 t .Fatalf ("unexpected error; %v\n " , err )
10099 }
@@ -116,11 +115,11 @@ func TestMigrateNoDrives(t *testing.T) {
116115 clientset = types .NewExtFakeClientset (clientsetfake .NewSimpleClientset ())
117116 client .SetDriveInterface (clientset .DirectpvLatest ().DirectPVDrives ())
118117
119- _ , legacyDriveErrors , driveErrors , err = task .migrateDrives (context . TODO (), false , nil )
118+ _ , legacyDriveErrors , driveErrors , err = task .migrateDrives (t . Context (), false , nil )
120119 if len (legacyDriveErrors ) != 0 || len (driveErrors ) != 0 || err != nil {
121120 t .Fatalf ("unexpected error; %v\n " , err )
122121 }
123- driveList , err = client .DriveClient ().List (context . TODO (), metav1.ListOptions {})
122+ driveList , err = client .DriveClient ().List (t . Context (), metav1.ListOptions {})
124123 if err != nil && ! apierrors .IsNotFound (err ) {
125124 t .Fatalf ("unexpected error; %v\n " , err )
126125 }
@@ -206,15 +205,15 @@ func TestMigrateReadyDrive(t *testing.T) {
206205 clientset := types .NewExtFakeClientset (clientsetfake .NewSimpleClientset ())
207206 client .SetDriveInterface (clientset .DirectpvLatest ().DirectPVDrives ())
208207 task := migrateTask {client .GetClient (), legacyclient .GetClient ()}
209- driveMap , legacyDriveErrors , driveErrors , err := task .migrateDrives (context . TODO (), false , nil )
208+ driveMap , legacyDriveErrors , driveErrors , err := task .migrateDrives (t . Context (), false , nil )
210209 if len (legacyDriveErrors ) != 0 || len (driveErrors ) != 0 || err != nil {
211210 t .Fatalf ("unexpected error; %v, %v, %v\n " , legacyDriveErrors , driveErrors , err )
212211 }
213212 if len (driveMap ) == 0 {
214213 t .Fatalf ("empty drive map\n " )
215214 }
216215
217- result , err := client .DriveClient ().Get (context . TODO (), "08450612-7ab3-40f9-ab83-38645fba6d29" , metav1.GetOptions {})
216+ result , err := client .DriveClient ().Get (t . Context (), "08450612-7ab3-40f9-ab83-38645fba6d29" , metav1.GetOptions {})
218217 if err != nil {
219218 t .Fatalf ("unexpected error; %v\n " , err )
220219 }
@@ -342,15 +341,15 @@ func TestMigrateInUseDrive(t *testing.T) {
342341 clientset := types .NewExtFakeClientset (clientsetfake .NewSimpleClientset ())
343342 client .SetDriveInterface (clientset .DirectpvLatest ().DirectPVDrives ())
344343 task := migrateTask {client .GetClient (), legacyclient .GetClient ()}
345- driveMap , legacyDriveErrors , driveErrors , err := task .migrateDrives (context . TODO (), false , nil )
344+ driveMap , legacyDriveErrors , driveErrors , err := task .migrateDrives (t . Context (), false , nil )
346345 if len (legacyDriveErrors ) != 0 || len (driveErrors ) != 0 || err != nil {
347346 t .Fatalf ("unexpected error; %v, %v, %v\n " , legacyDriveErrors , driveErrors , err )
348347 }
349348 if len (driveMap ) == 0 {
350349 t .Fatalf ("empty drive map\n " )
351350 }
352351
353- result , err := client .DriveClient ().Get (context . TODO (), "08450612-7ab3-40f9-ab83-38645fba6d29" , metav1.GetOptions {})
352+ result , err := client .DriveClient ().Get (t . Context (), "08450612-7ab3-40f9-ab83-38645fba6d29" , metav1.GetOptions {})
354353 if err != nil {
355354 t .Fatalf ("unexpected error; %v\n " , err )
356355 }
@@ -454,7 +453,7 @@ func TestMigrateVolumes(t *testing.T) {
454453 client .SetVolumeInterface (clientset .DirectpvLatest ().DirectPVVolumes ())
455454 task := migrateTask {client .GetClient (), legacyclient .GetClient ()}
456455 legacyVolumeErrors , volumeErrors , err := task .migrateVolumes (
457- context . TODO (),
456+ t . Context (),
458457 map [string ]string {"08450612-7ab3-40f9-ab83-38645fba6d29" : "a9908089-96dd-4e8b-8f06-0c0b5e391f39" },
459458 false ,
460459 nil ,
@@ -463,7 +462,7 @@ func TestMigrateVolumes(t *testing.T) {
463462 t .Fatalf ("unexpected error; %v, %v, %v\n " , legacyVolumeErrors , volumeErrors , err )
464463 }
465464
466- result , err := client .VolumeClient ().Get (context . TODO (), "pvc-009bfc49-4a66-4055-9f19-bd039cc3b4f5" , metav1.GetOptions {})
465+ result , err := client .VolumeClient ().Get (t . Context (), "pvc-009bfc49-4a66-4055-9f19-bd039cc3b4f5" , metav1.GetOptions {})
467466 if err != nil {
468467 t .Fatalf ("unexpected error; %v\n " , err )
469468 }
@@ -493,7 +492,7 @@ func TestMigrateVolumes(t *testing.T) {
493492
494493 // no fsuuid found error
495494 legacyclient .SetVolumeClient (legacyclientsetfake .NewSimpleClientset (volume ))
496- legacyVolumeErrors , volumeErrors , err = task .migrateVolumes (context . TODO (), map [string ]string {}, false , nil )
495+ legacyVolumeErrors , volumeErrors , err = task .migrateVolumes (t . Context (), map [string ]string {}, false , nil )
497496 if len (legacyVolumeErrors ) == 0 && len (volumeErrors ) == 0 && err == nil {
498497 t .Fatalf ("expected error; but succeeded\n " )
499498 }
0 commit comments