fix: image load should close stream after copy

Signed-off-by: Cezar Rata <[email protected]>

internal/service/image/load.go

@@ -42,11 +42,16 @@ func (s *service) Load(ctx context.Context, inStream io.Reader, outStream io.Wri
 		return err
 	}
 	defer func() {
-		rw.Close()
 		os.Remove(img)
 	}()
 	go func() {
-		io.Copy(rw, inStream)
+		written, err := io.Copy(rw, inStream)
+		if err != nil {
+			s.logger.Errorf("failed to copy: %s", err)
+		} else {
+			s.logger.Debugf("copied %d bytes", written)
+		}
+		rw.Close()
 	}()
 	if err = s.nctlImageSvc.LoadImage(ctx, img, outStream, quiet); err != nil {
 		s.logger.Errorf("failed to load image %s: %s", img, err)

internal/service/image/load_test.go

@@ -42,6 +42,7 @@ var _ = Describe("Image Load API", func() {
 	})
 	Context("service", func() {
 		It("should return no errors upon success", func() {
+			logger.EXPECT().Debugf(gomock.Any(), gomock.Any())
 			ncClient.EXPECT().GetDataStore().
 				Return(name, nil)
 			ncClient.EXPECT().LoadImage(gomock.Any(), gomock.Any(), nil, gomock.Any()).
@@ -53,6 +54,7 @@ var _ = Describe("Image Load API", func() {
 		})
 		It("should return an error if load image method returns an error", func() {
 			logger.EXPECT().Errorf(gomock.Any(), gomock.Any())
+			logger.EXPECT().Debugf(gomock.Any(), gomock.Any())
 			ncClient.EXPECT().GetDataStore().
 				Return(name, nil)
 			ncClient.EXPECT().LoadImage(gomock.Any(), gomock.Any(), nil, gomock.Any()).