Change image_enc_test so that it more reliably delete images

lumjjb · lumjjb · commit c118c45f89d9 · 2019-07-19T10:08:57.000-04:00
Signed-off-by: Brandon Lum &lt;lumjjb@gmail.com&gt;
Signed-off-by: Stefan Berger &lt;stefanb@linux.ibm.com&gt;
diff --git a/image_enc_test.go b/image_enc_test.go
@@ -161,10 +161,15 @@ func TestImageEncryption(t *testing.T) {
 			Parameters: dcparameters,
 		},
 	}
+	// Clean up function cancels lease before deleting the image so the images are
+	// properly deleted
+	defer func() {
+		done(ctx)
+		client.ImageService().Delete(ctx, imageName, images.SynchronousDelete())
+		client.ImageService().Delete(ctx, encImageName, images.SynchronousDelete())
+	}()
 
 	// Perform decryption of image
-	defer client.ImageService().Delete(ctx, imageName, images.SynchronousDelete())
-	defer client.ImageService().Delete(ctx, encImageName, images.SynchronousDelete())
 	lf = func(desc ocispec.Descriptor) bool {
 		return true
 	}
