client: Client.ImageLoad: close reader on context cancellation

Use a cancelReadCloser to automatically close the reader when the context
is cancelled. Consumers are still recommended to manually close the reader,
but the cancelReadCloser makes the Close idempotent.

Signed-off-by: Sebastiaan van Stijn <github@gone.nl>

integration/build/build_userns_linux_test.go

@@ -106,8 +106,8 @@ func TestBuildUserNamespaceValidateCapabilitiesAreV2(t *testing.T) {
 	loadResp, err := clientNoUserRemap.ImageLoad(ctx, tarReader)
 	assert.NilError(t, err, "failed to load image tar file")
 	defer loadResp.Close()
-	buf = bytes.NewBuffer(nil)
-	err = jsonmessage.DisplayJSONMessagesStream(loadResp, buf, 0, false, nil)
+	var buf2 bytes.Buffer
+	err = jsonmessage.DisplayJSONMessagesStream(loadResp, &buf2, 0, false, nil)
 	assert.NilError(t, err)
 
 	cid := container.Run(ctx, t, clientNoUserRemap,

integration/daemon/migration_test.go

@@ -146,10 +146,10 @@ func TestMigrateSaveLoad(t *testing.T) {
 	assert.Equal(t, info.Images, 0)
 
 	// Import
-	lr, err := apiClient.ImageLoad(ctx, bytes.NewReader(buf.Bytes()), client.ImageLoadWithQuiet(true))
+	resp, err := apiClient.ImageLoad(ctx, bytes.NewReader(buf.Bytes()), client.ImageLoadWithQuiet(true))
 	assert.NilError(t, err)
-	io.Copy(io.Discard, lr)
-	lr.Close()
+	_, _ = io.Copy(io.Discard, resp)
+	_ = resp.Close()
 
 	result := container.RunAttach(ctx, t, apiClient, func(c *container.TestContainerConfig) {
 		c.Name = "Migration-save-load-" + snapshotter

integration/image/save_test.go

@@ -343,11 +343,9 @@ func TestSaveAndLoadPlatform(t *testing.T) {
 			// load the full exported image (all platforms in it)
 			resp, err := apiClient.ImageLoad(ctx, rdr)
 			assert.NilError(t, err)
-			_, err = io.ReadAll(resp)
-			resp.Close()
-			assert.NilError(t, err)
-
-			rdr.Close()
+			_, _ = io.Copy(io.Discard, resp)
+			_ = resp.Close()
+			_ = rdr.Close()
 
 			// verify the loaded image has all the expected platforms
 			for _, p := range tc.expectedSavedPlatforms {
@@ -381,11 +379,9 @@ func TestSaveAndLoadPlatform(t *testing.T) {
 			// load the exported image on the specified platforms only
 			resp, err = apiClient.ImageLoad(ctx, rdr, client.ImageLoadWithPlatforms(tc.loadPlatforms...))
 			assert.NilError(t, err)
-			_, err = io.ReadAll(resp)
-			resp.Close()
-			assert.NilError(t, err)
-
-			rdr.Close()
+			_, _ = io.Copy(io.Discard, resp)
+			_ = resp.Close()
+			_ = rdr.Close()
 
 			// verify the image was loaded for the specified platforms
 			for _, p := range tc.expectedLoadedPlatforms {

integration/internal/image/load.go

@@ -30,7 +30,7 @@ func Load(ctx context.Context, t *testing.T, apiClient client.APIClient, imageFu
 	resp, err := apiClient.ImageLoad(ctx, rc, client.ImageLoadWithQuiet(true))
 	assert.NilError(t, err, "Failed to load dangling image")
 
-	defer resp.Close()
+	defer func() { _ = resp.Close() }()
 
 	if !assert.Check(t, err) {
 		respBody, err := io.ReadAll(resp)

integration/plugin/authz/authz_plugin_test.go

@@ -445,11 +445,12 @@ func imageLoad(ctx context.Context, apiClient client.APIClient, path string) err
 		return err
 	}
 	defer file.Close()
-	response, err := apiClient.ImageLoad(ctx, file, client.ImageLoadWithQuiet(true))
+	resp, err := apiClient.ImageLoad(ctx, file, client.ImageLoadWithQuiet(true))
 	if err != nil {
 		return err
 	}
-	defer response.Close()
+	_, _ = io.Copy(io.Discard, resp)
+	_ = resp.Close()
 	return nil
 }