test(cli): add coverage for exit codes, output parsing and NO_COLOR

Author: almeidazs

cmd/root_test.go

@@ -0,0 +1,40 @@
+package cmd
+
+import (
+	"testing"
+
+	"github.com/RewriteToday/cli/internal/clierr"
+)
+
+func TestResolveOutputFormat(t *testing.T) {
+	tests := []struct {
+		name string
+		args []string
+		want string
+	}{
+		{name: "default", args: []string{}, want: "text"},
+		{name: "long flag", args: []string{"--output", "json"}, want: "json"},
+		{name: "short flag", args: []string{"-o", "json"}, want: "json"},
+		{name: "equals", args: []string{"--output=json"}, want: "json"},
+		{name: "invalid falls back", args: []string{"--output=xml"}, want: "text"},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := ResolveOutputFormat(tt.args); got != tt.want {
+				t.Fatalf("expected %q, got %q", tt.want, got)
+			}
+		})
+	}
+}
+
+func TestNormalizeOutputFormat_InvalidReturnsUsageError(t *testing.T) {
+	_, err := normalizeOutputFormat("xml")
+	if err == nil {
+		t.Fatal("expected an error")
+	}
+
+	if got := clierr.ExitCode(err); got != int(clierr.CodeUsage) {
+		t.Fatalf("expected code %d, got %d", clierr.CodeUsage, got)
+	}
+}

internal/clierr/clierr_test.go

@@ -0,0 +1,31 @@
+package clierr
+
+import (
+	"context"
+	"errors"
+	"testing"
+)
+
+func TestExitCode_WithWrappedCode(t *testing.T) {
+	err := Wrap(CodeUsage, errors.New("bad args"))
+
+	if got := ExitCode(err); got != int(CodeUsage) {
+		t.Fatalf("expected %d, got %d", CodeUsage, got)
+	}
+}
+
+func TestExitCode_WithNetworkDeadline(t *testing.T) {
+	err := context.DeadlineExceeded
+
+	if got := ExitCode(err); got != int(CodeNetwork) {
+		t.Fatalf("expected %d, got %d", CodeNetwork, got)
+	}
+}
+
+func TestExitCode_WithUnknownError(t *testing.T) {
+	err := errors.New("boom")
+
+	if got := ExitCode(err); got != int(CodeInternal) {
+		t.Fatalf("expected %d, got %d", CodeInternal, got)
+	}
+}

internal/render/colors_test.go

@@ -0,0 +1,17 @@
+package render
+
+import "testing"
+
+func TestIsColorEnabled_RespectsNoColor(t *testing.T) {
+	t.Setenv("NO_COLOR", "1")
+	if IsColorEnabled() {
+		t.Fatal("expected colors to be disabled when NO_COLOR is set")
+	}
+}
+
+func TestIsColorEnabled_DefaultEnabled(t *testing.T) {
+	t.Setenv("NO_COLOR", "")
+	if !IsColorEnabled() {
+		t.Fatal("expected colors to be enabled when NO_COLOR is empty")
+	}
+}