test: fix flaky TestDockerEngine (aquasecurity#6054)

DmitriyLewen · Feb 2, 2024 · 27d35ba · 27d35ba
1 parent c3a66da
commit 27d35ba
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/integration/docker_engine_test.go b/integration/docker_engine_test.go
@@ -245,23 +245,25 @@ func TestDockerEngine(t *testing.T) {
 				// load image into docker engine
 				res, err := cli.ImageLoad(ctx, testfile, true)
 				require.NoError(t, err, tt.name)
-				io.Copy(io.Discard, res.Body)
+				if _, err := io.Copy(io.Discard, res.Body); err != nil {
+					require.NoError(t, err, tt.name)
+				}
+				defer res.Body.Close()
 
 				// tag our image to something unique
 				err = cli.ImageTag(ctx, tt.imageTag, tt.input)
 				require.NoError(t, err, tt.name)
 
 				// cleanup
 				t.Cleanup(func() {
-					_, err = cli.ImageRemove(ctx, tt.input, api.ImageRemoveOptions{
+					_, _ = cli.ImageRemove(ctx, tt.input, api.ImageRemoveOptions{
 						Force:         true,
 						PruneChildren: true,
 					})
-					_, err = cli.ImageRemove(ctx, tt.imageTag, api.ImageRemoveOptions{
+					_, _ = cli.ImageRemove(ctx, tt.imageTag, api.ImageRemoveOptions{
 						Force:         true,
 						PruneChildren: true,
 					})
-					assert.NoError(t, err, tt.name)
 				})
 			}