fix image pull stream error handling

Signed-off-by: puneeth_aditya_5656 <myakampuneeth@gmail.com>

Author: mugiwaraluffy56

pkg/connectors/container_client.go

@@ -112,7 +112,9 @@ func (cli *containerClient) CreateContainer(opts ContainerOpts) (string, error)
 		return "", err
 	}
 	defer out.Close()
-	io.Copy(os.Stdout, out)
+	if err := copyImagePullOutput(out); err != nil {
+		return "", err
+	}
 
 	resp, err := cli.cli.ContainerCreate(
 		ctx,
@@ -148,3 +150,11 @@ func (cli *containerClient) StopContainer(containerId string) error {
 func (cli *containerClient) CloseClient() error {
 	return cli.cli.Close()
 }
+
+func copyImagePullOutput(out io.Reader) error {
+	_, err := io.Copy(os.Stdout, out)
+	if err != nil {
+		return fmt.Errorf("failed to read image pull output: %w", err)
+	}
+	return nil
+}

pkg/connectors/container_client_test.go

@@ -0,0 +1,35 @@
+package connectors
+
+import (
+	"errors"
+	"io"
+	"strings"
+	"testing"
+)
+
+type failingReader struct{}
+
+func (f failingReader) Read(p []byte) (int, error) {
+	return 0, errors.New("read failed")
+}
+
+func TestCopyImagePullOutputReturnsReadError(t *testing.T) {
+	err := copyImagePullOutput(failingReader{})
+
+	if err == nil {
+		t.Fatal("expected error")
+	}
+	if !strings.Contains(err.Error(), "failed to read image pull output") {
+		t.Fatalf("unexpected error: %v", err)
+	}
+}
+
+func TestCopyImagePullOutputSucceeds(t *testing.T) {
+	err := copyImagePullOutput(strings.NewReader(""))
+
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+}
+
+var _ io.Reader = failingReader{}