// Copyright (C) 2021  Simon Ruderich
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package frontend

import (
	"testing"

	"ruderich.org/simon/safcm/testutil"
)

func TestEscapeControlCharacters(t *testing.T) {
	tests := []struct {
		name  string
		isTTY bool
		x     string
		exp   string
	}{
		{
			"UTF-8",
			false,
			"Hello, 世界",
			"Hello, 世界",
		},
		{
			"UTF-8 (TTY)",
			true,
			"Hello, 世界",
			"Hello, 世界",
		},

		{
			"color",
			false,
			ColorString(true, ColorRed, "red"),
			`\x1B[31mred\x1B[0m`,
		},
		{
			"color (TTY)",
			true,
			ColorString(true, ColorRed, "red"),
			"\x1B[35m\\x1B\x1B[0m[31mred\x1B[35m\\x1B\x1B[0m[0m",
		},

		{
			"\\r\\n",
			false,
			"\r\n",
			"\\r\n",
		},
		{
			"\\r\\n (TTY)",
			true,
			"\r\n",
			"\x1B[35m\\r\x1B[0m\n",
		},

		{
			"binary",
			false,
			"\xD6\x24\xA4\x45\xA2\x68\xD3\x0E\xD4\xC7\xC3\x1F",
			"\xD6$\xA4E\xA2h\xD3\\x0E\xD4\xC7\xC3\\x1F",
		},
		{
			"binary (TTY)",
			true,
			"\xD6\x24\xA4\x45\xA2\x68\xD3\x0E\xD4\xC7\xC3\x1F",
			"\xD6$\xA4E\xA2h\xD3\x1B[35m\\x0E\x1B[0m\xD4\xC7\xc3\x1B[35m\\x1F\x1B[0m",
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			res := EscapeControlCharacters(tc.isTTY, tc.x)
			testutil.AssertEqual(t, "res", res, tc.exp)
		})
	}
}