added unit testing

main_test.go

@@ -0,0 +1,129 @@
+package main
+
+import (
+	"os"
+	"strings"
+	"testing"
+	"time"
+)
+
+func TestRunMainInvalidOnly(t *testing.T) {
+	defer resetMocks()
+
+	var exited bool
+	var exitCode int
+	osExit = func(code int) {
+		exited = true
+		exitCode = code
+	}
+
+	runMain([]string{"bootstrap_environment", "--only", "invalid"})
+
+	if !exited {
+		t.Error("expected runMain with invalid --only value to exit")
+	}
+	if exitCode != 2 {
+		t.Errorf("expected exit code 2, got %d", exitCode)
+	}
+}
+
+func TestRunMainAllInstalled(t *testing.T) {
+	defer resetMocks()
+
+	// All packages are already installed (total packages to install = 0)
+	osStat = func(name string) (os.FileInfo, error) {
+		return nil, nil // all custom paths exist
+	}
+	probe = func(argv []string, timeout time.Duration) (CmdResult, bool) {
+		if argv[0] == "rpm" || argv[0] == "dpkg-query" || argv[0] == "pacman" {
+			return CmdResult{ExitCode: 0, Stdout: []byte("install ok installed")}, true
+		}
+		if argv[0] == "flatpak" {
+			return CmdResult{ExitCode: 0}, true
+		}
+		return CmdResult{ExitCode: 0}, true
+	}
+	hasCmd = func(name string) bool {
+		return true
+	}
+
+	var exited bool
+	osExit = func(code int) {
+		exited = true
+	}
+
+	// We only run custom to keep it short & avoid other logic dependencies
+	runMain([]string{"bootstrap_environment", "--only", "custom"})
+
+	if exited {
+		t.Error("expected program to complete successfully without exiting")
+	}
+}
+
+func TestRunMainInstallAbort(t *testing.T) {
+	defer resetMocks()
+
+	// Simulating some packages to install, but user selects N
+	osStat = func(name string) (os.FileInfo, error) {
+		return nil, os.ErrNotExist // custom paths missing -> need install
+	}
+	probe = func(argv []string, timeout time.Duration) (CmdResult, bool) {
+		return CmdResult{ExitCode: 1}, true // packages not installed
+	}
+	hasCmd = func(name string) bool {
+		return true
+	}
+
+	// Mock stdin to say "n" to abort the prompt "Proceed? [y/N]"
+	stdin = strings.NewReader("n\n")
+
+	var exited bool
+	var exitCode int
+	osExit = func(code int) {
+		exited = true
+		exitCode = code
+	}
+
+	runMain([]string{"bootstrap_environment", "--only", "custom"})
+
+	if !exited {
+		t.Error("expected program to exit on user abort")
+	}
+	if exitCode != 1 {
+		t.Errorf("expected exit code 1, got %d", exitCode)
+	}
+}
+
+func TestRunMainMacos(t *testing.T) {
+	defer resetMocks()
+
+	isMacOS = true
+	pkgMgr = "brew"
+	osStat = func(name string) (os.FileInfo, error) {
+		return nil, nil // brew etc exist
+	}
+	probe = func(argv []string, timeout time.Duration) (CmdResult, bool) {
+		return CmdResult{ExitCode: 0}, true
+	}
+	hasCmd = func(name string) bool {
+		return true
+	}
+	runCmd = func(argv []string, opts CmdOpts) CmdResult {
+		return CmdResult{ExitCode: 0}
+	}
+	runShell = func(cmd string, opts CmdOpts) CmdResult {
+		return CmdResult{ExitCode: 0}
+	}
+
+	var exited bool
+	osExit = func(code int) {
+		exited = true
+	}
+
+	// Run with --only custom, --no-vm, --no-ai
+	runMain([]string{"bootstrap_environment", "--only", "custom", "--no-vm", "--no-ai"})
+
+	if exited {
+		t.Error("expected program to complete successfully on macOS mock run")
+	}
+}