Function TestCopyGuardedContent

pkg/iostreams/content_test.go:53–148

(t *testing.T)

}

func TestCopyGuardedContent(t *testing.T) {
	png := append([]byte("\x89PNG\r\n\x1a\n"), make([]byte, 16)...)
	readErr := errors.New("boom")

	tests := []struct {
		name    string
		content []byte
		// reader overrides content for cases a byte slice cannot express, such
		// as a mid-stream read failure.
		reader  io.Reader
		isTTY   bool
		wantOut []byte
		// wantErrIs matches a sentinel error with errors.Is; wantErrAs matches a
		// typed error (e.g. BinaryTerminalError, which carries a MIME field) with
		// errors.As, so its value must be a pointer to that error type.
		wantErrIs error
		wantErrAs any
	}{
		{
			name:    "clean text is written",
			content: []byte("hello world\n"),
			isTTY:   true,
			wantOut: []byte("hello world\n"),
		},
		{
			name:      "text with escape is refused",
			content:   []byte("danger\x1b[31mtext"),
			isTTY:     true,
			wantErrIs: ErrEscapeSequence,
		},
		{
			name:      "text with escape is refused when piped",
			content:   []byte("danger\x1b[31mtext"),
			isTTY:     false,
			wantErrIs: ErrEscapeSequence,
		},
		{
			name:      "binary to terminal is refused",
			content:   png,
			isTTY:     true,
			wantErrAs: &BinaryTerminalError{},
		},
		{
			name:    "binary when piped is written raw",
			content: png,
			isTTY:   false,
			wantOut: png,
		},
		{
			name:    "empty content writes nothing",
			content: []byte{},
			isTTY:   true,
			wantOut: nil,
		},
		{
			// Content past the sniff window is still inspected, so an escape
			// hiding beyond the first chunk is caught.
			name:      "text with escape past the sniff window is refused",