Add unit test

internal/cli/main.go

@@ -77,7 +77,7 @@ func EscapeIfNeeded(str string) string {
 		// If the str is already unescaped or an error occurred, escape it
 		return url.QueryEscape(str)
 	}
-	// If the str was successfully unescaped, return it as is
+	// If the str was successfully unescaped and is different from the original, return the original
 	return str
 }
 

internal/cli/main_test.go

@@ -0,0 +1,48 @@
+package cli
+
+import (
+	"testing"
+)
+
+func TestEscapeIfNeeded(t *testing.T) {
+	tests := []struct {
+		name     string
+		input    string
+		expected string
+	}{
+		{
+			name:     "AlreadyEscaped",
+			input:    "hello%20world",
+			expected: "hello%20world",
+		},
+		{
+			name:     "Unescaped",
+			input:    "hello world",
+			expected: "hello+world",
+		},
+		{
+			name:     "PartiallyEscaped",
+			input:    "hello%20world!",
+			expected: "hello%20world!",
+		},
+		{
+			name:     "EmptyString",
+			input:    "",
+			expected: "",
+		},
+		{
+			name:     "SpecialCharacters",
+			input:    "hello@world.com",
+			expected: "hello%40world.com",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result := EscapeIfNeeded(tt.input)
+			if result != tt.expected {
+				t.Errorf("EscapeIfNeeded(%q) actual = %q; want = %q", tt.input, result, tt.expected)
+			}
+		})
+	}
+}