gitea-tea/internal/terminal/safe_writer_test.go

// Copyright 2026 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package terminal

import (
	"bytes"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestSafeWriterKeepsLongSGRSequences(t *testing.T) {
	const colored = "\x1b[38;2;255;255;255;48;2;255;255;255mwhite\x1b[0m\n"
	for _, chunk := range []int{len(colored), 1, 3} {
		var out bytes.Buffer
		w := SafeWriter(&out, true, true)
		for data := []byte(colored); len(data) > 0; {
			n := min(chunk, len(data))
			_, err := w.Write(data[:n])
			require.NoError(t, err)
			data = data[n:]
		}
		require.Equal(t, colored, out.String(), "chunk size %d", chunk)
	}
}

func TestSafeWriter(t *testing.T) {
	long := strings.Repeat("1", 300)
	hostile := "+hello \x1b]0;PWNED\x07 \x1b[1;31mred\x1b[0m\x1b[2J\r\n\u009b2J\u202eend\t\xff\x1b[" + long + "m\n"
	var piped bytes.Buffer
	_, err := SafeWriter(&piped, false, false).Write([]byte(hostile))
	require.NoError(t, err)
	require.Equal(t, hostile, piped.String())
	for _, tc := range []struct {
		name     string
		allowSGR bool
		want     string
	}{
		{"drops every escape", false, "+hello ]0;PWNED [1;31mred[0m[2J\n2Jend\t\uFFFD[" + long + "m\n"},
		{"keeps colors", true, "+hello ]0;PWNED \x1b[1;31mred\x1b[0m[2J\n2Jend\t\uFFFD[" + long + "m\n"},
	} {
		t.Run(tc.name, func(t *testing.T) {
			if !tc.allowSGR {
				require.Equal(t, SafeText(hostile), tc.want)
			}
			for _, chunk := range []int{len(hostile), 1, 3} {
				var out bytes.Buffer
				w := SafeWriter(&out, true, tc.allowSGR)
				for data := []byte(hostile); len(data) > 0; {
					n := min(chunk, len(data))
					written, err := w.Write(data[:n])
					require.NoError(t, err)
					require.Equal(t, n, written)
					data = data[n:]
				}
				require.Equal(t, tc.want, out.String(), "chunk size %d", chunk)
			}
		})
	}
}