zero/internal/tui/command_output_test.go

package tui

import (
	"strings"
	"testing"
)

func TestFormatCommandOutputRendersSectionsFieldsRowsAndHints(t *testing.T) {
	got := formatCommandOutput(commandOutput{
		Title:  "Model",
		Status: commandStatusOK,
		Sections: []commandSection{
			{
				Title: "Active",
				Fields: []commandField{
					{Key: "provider", Value: "openai"},
					{Key: "model", Value: "gpt-5"},
				},
				Rows: []commandRow{
					{Text: "model shell is ready"},
				},
				Hints: []string{"use /model list to compare options"},
			},
			{
				Title: "Available",
				Rows: []commandRow{
					{Text: "gpt-5"},
					{Text: "claude-sonnet-4.5"},
				},
			},
		},
	})

	want := strings.Join([]string{
		"Model",
		"status: ok",
		"Active",
		"  provider: openai",
		"  model: gpt-5",
		"  - model shell is ready",
		"  hint: use /model list to compare options",
		"Available",
		"  - gpt-5",
		"  - claude-sonnet-4.5",
	}, "\n")

	if got != want {
		t.Fatalf("unexpected command output:\nwant:\n%s\n\ngot:\n%s", want, got)
	}
}

func TestFormatCommandOutputSupportsAllStatuses(t *testing.T) {
	tests := []struct {
		name   string
		status commandStatus
		want   string
	}{
		{name: "ok", status: commandStatusOK, want: "Status\nstatus: ok"},
		{name: "warning", status: commandStatusWarning, want: "Status\nstatus: warning"},
		{name: "blocked", status: commandStatusBlocked, want: "Status\nstatus: blocked"},
		{name: "info", status: commandStatusInfo, want: "Status\nstatus: info"},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := formatCommandOutput(commandOutput{
				Title:  "Status",
				Status: tt.status,
			})

			if got != tt.want {
				t.Fatalf("expected %q, got %q", tt.want, got)
			}
		})
	}
}

func TestFormatCommandOutputRendersPlainLinesAndCommandBullets(t *testing.T) {
	got := formatCommandOutput(commandOutput{
		Title:  "Commands",
		Status: commandStatusInfo,
		Sections: []commandSection{
			{
				Title: "Model",
				Lines: []string{
					"/model [list|id] - Show the active model.",
					commandBullet("gpt-5"),
				},
			},
		},
	})

	want := strings.Join([]string{
		"Commands",
		"status: info",
		"Model",
		"  /model [list|id] - Show the active model.",
		"  - gpt-5",
	}, "\n")

	if got != want {
		t.Fatalf("unexpected command lines:\nwant:\n%s\n\ngot:\n%s", want, got)
	}
}

func TestFormatCommandOutputRedactsTokenLikeText(t *testing.T) {
	got := formatCommandOutput(commandOutput{
		Title:  "Permissions",
		Status: commandStatusOK,
		Sections: []commandSection{{
			Title: "Sandbox grants",
			Lines: []string{
				"bash [allow/high] - sk-proj-sensitive-token-value approved shell",
				"anthropic: sk-ant-api03-abcdefghijklmnopqrstuvwxyz",
				"google: AIza1234567890abcdef",
			},
		}},
	})

	for _, secret := range []string{
		"sk-proj-sensitive-token-value",
		"sk-ant-api03-abcdefghijklmnopqrstuvwxyz",
		"AIza1234567890abcdef",
	} {
		if strings.Contains(got, secret) {
			t.Fatalf("expected token-like text to be redacted, got:\n%s", got)
		}
	}
	if !strings.Contains(got, "[REDACTED] approved shell") {
		t.Fatalf("expected token-like text to be redacted, got:\n%s", got)
	}
}

func TestFormatCommandOutputCompactsWhitespaceAndSkipsEmptyItems(t *testing.T) {
	got := formatCommandOutput(commandOutput{
		Title:  "  Doctor  ",
		Status: commandStatusWarning,
		Sections: []commandSection{
			{
				Title: " ",
				Fields: []commandField{
					{Key: " ", Value: "skip me"},
					{Key: "sandbox", Value: "  danger-full-access\r\n(no prompt)  "},
				},
				Rows: []commandRow{
					{Text: " "},
					{Text: "checks complete\nneeds review"},
				},
				Hints: []string{
					"  use /doctor for details\r\nsoon  ",
				},
			},
		},
		Hints: []string{
			"  top-level hint\nis compacted  ",
		},
	})

	want := strings.Join([]string{
		"Doctor",
		"status: warning",
		"  sandbox: danger-full-access (no prompt)",
		"  - checks complete needs review",
		"  hint: use /doctor for details soon",
		"hint: top-level hint is compacted",
	}, "\n")

	if got != want {
		t.Fatalf("unexpected compact command output:\nwant:\n%s\n\ngot:\n%s", want, got)
	}
}