From c727a0fec7c7a36952dc705f911863428f4ab0aa Mon Sep 17 00:00:00 2001
From: ROKUMATE <rohitkumawat0110@gmail.com>
Date: Thu, 11 Jun 2026 00:15:40 +0530
Subject: [PATCH] pkg/copy: add tests for ParseSourceAndDestination

Fixes: #28907

Signed-off-by: ROKUMATE <rohitkumawat0110@gmail.com>
---
 pkg/copy/parse_test.go | 160 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 160 insertions(+)
 create mode 100644 pkg/copy/parse_test.go

diff --git a/pkg/copy/parse_test.go b/pkg/copy/parse_test.go
new file mode 100644
index 0000000000..85b1cb886e
--- /dev/null
+++ b/pkg/copy/parse_test.go
@@ -0,0 +1,160 @@
+package copy
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestParseSourceAndDestination(t *testing.T) {
+	tests := []struct {
+		name        string
+		source      string
+		destination string
+		srcCtr      string
+		srcPath     string
+		destCtr     string
+		destPath    string
+	}{
+		{
+			name:        "container source, local destination",
+			source:      "ctr:/etc/hostname",
+			destination: "/tmp/hostname",
+			srcCtr:      "ctr",
+			srcPath:     "/etc/hostname",
+			destCtr:     "",
+			destPath:    "/tmp/hostname",
+		},
+		{
+			name:        "local source, container destination",
+			source:      "/tmp/hostname",
+			destination: "ctr:/etc/hostname",
+			srcCtr:      "",
+			srcPath:     "/tmp/hostname",
+			destCtr:     "ctr",
+			destPath:    "/etc/hostname",
+		},
+		{
+			name:        "colon in a path starting with a dot is not a container",
+			source:      "./weird:name",
+			destination: "ctr:/dst",
+			srcCtr:      "",
+			srcPath:     "./weird:name",
+			destCtr:     "ctr",
+			destPath:    "/dst",
+		},
+		{
+			name:        "colon in a path starting with a slash is not a container",
+			source:      "/abs/weird:name",
+			destination: "ctr:/dst",
+			srcCtr:      "",
+			srcPath:     "/abs/weird:name",
+			destCtr:     "ctr",
+			destPath:    "/dst",
+		},
+		{
+			name:        "relative path without a colon has no container",
+			source:      "relative/path",
+			destination: "ctr:/dst",
+			srcCtr:      "",
+			srcPath:     "relative/path",
+			destCtr:     "ctr",
+			destPath:    "/dst",
+		},
+		{
+			name:        "bare filename without a colon has no container",
+			source:      "myfile.txt",
+			destination: "/dst",
+			srcCtr:      "",
+			srcPath:     "myfile.txt",
+			destCtr:     "",
+			destPath:    "/dst",
+		},
+		{
+			name:        "only the first colon separates container from path",
+			source:      "ctr:/path/with:colon",
+			destination: "/dst",
+			srcCtr:      "ctr",
+			srcPath:     "/path/with:colon",
+			destCtr:     "",
+			destPath:    "/dst",
+		},
+		{
+			name:        "spaces in the path are preserved",
+			source:      "ctr:/path with spaces/file",
+			destination: "/dst",
+			srcCtr:      "ctr",
+			srcPath:     "/path with spaces/file",
+			destCtr:     "",
+			destPath:    "/dst",
+		},
+		{
+			name:        "no container on either side",
+			source:      "/src",
+			destination: "/dst",
+			srcCtr:      "",
+			srcPath:     "/src",
+			destCtr:     "",
+			destPath:    "/dst",
+		},
+		{
+			name: "container on both sides",
+			// Multi-letter names so they are not mistaken for Windows drive
+			// letters (e.g. "a:" is an absolute path on Windows).
+			source:      "ctra:/x",
+			destination: "ctrb:/y",
+			srcCtr:      "ctra",
+			srcPath:     "/x",
+			destCtr:     "ctrb",
+			destPath:    "/y",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			srcCtr, srcPath, destCtr, destPath, err := ParseSourceAndDestination(tt.source, tt.destination)
+			require.NoError(t, err)
+			assert.Equal(t, tt.srcCtr, srcCtr, "source container")
+			assert.Equal(t, tt.srcPath, srcPath, "source path")
+			assert.Equal(t, tt.destCtr, destCtr, "destination container")
+			assert.Equal(t, tt.destPath, destPath, "destination path")
+		})
+	}
+}
+
+func TestParseSourceAndDestinationErrors(t *testing.T) {
+	tests := []struct {
+		name        string
+		source      string
+		destination string
+	}{
+		{
+			name:        "missing source path",
+			source:      "ctr:",
+			destination: "/tmp/hostname",
+		},
+		{
+			name:        "missing destination path",
+			source:      "/tmp/hostname",
+			destination: "ctr:",
+		},
+		{
+			name:        "empty source with a valid destination",
+			source:      "",
+			destination: "/dst",
+		},
+		{
+			name:        "both empty",
+			source:      "",
+			destination: "",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			_, _, _, _, err := ParseSourceAndDestination(tt.source, tt.destination)
+			assert.Error(t, err)
+		})
+	}
+}