1 files changed, 34 insertions, 0 deletions
diff --git a/internal/cli/cli_test.go b/internal/cli/cli_test.go
new file mode 100644
index 0000000..71ff6cb
--- /dev/null
+++ b/internal/cli/cli_test.go
@@ -0,0 +1,34 @@
+package cli
+
+import (
+	"testing"
+)
+
+func TestAppRun_ProfileDirInjection(t *testing.T) {
+	// Set up a temporary directory to simulate XDG_CONFIG_HOME/wg-wrap/profiles
+	tmpDir := t.TempDir()
+
+	tests := []struct {
+		name    string
+		args    []string
+		wantErr bool
+	}{
+		{
+			name:    "valid profile with injected dir",
+			args:    []string{"wg-wrap", "--profile", "test-vpn", "curl", "google.com"},
+			wantErr: false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			app := NewApp(tt.args)
+			app.ConfigDir = tmpDir // Inject temporary directory
+
+			err := app.Run()
+			if (err != nil) != tt.wantErr {
+				t.Errorf("App.Run() error = %v, wantErr %v", err, tt.wantErr)
+			}
+		})
+	}
+}