package chrome

import "testing"

func TestTranslateStringSelector_KnownKeys(t *testing.T) {
	cases := []struct {
		selector string
		want     string
		isXPath  bool
	}{
		{"id:email", `[id="email"]`, false},
		{"resource-id:account-name", `[id="account-name"]`, false},
		{"class:btn-primary", `[class~="btn-primary"]`, false},
		{"tag:button", `button`, false},
		{"text:Sign in", `//*[normalize-space(text())="Sign in"]`, true},
		{`text:Say "hi"`, `//*[normalize-space(text())='Say "hi"']`, true},
		{`text:it's`, `//*[normalize-space(text())="it's"]`, true},
		{`text:it's "fine"`, `//*[normalize-space(text())=concat("it's ", '"', "fine", '"', "")]`, true},
		{"desc:logout", `[aria-label="logout"]`, false},
		{"label:logout", `[aria-label="logout"]`, false},
		{"accessibilityLabel:logout", `[aria-label="logout"]`, false},
		{"aria-label:Sign in", `[aria-label="Sign in"]`, false},
		{"descPrefix:account:", `[aria-label^="account:"]`, false},
		{"testTag:submit", `[data-testid="submit"]`, false},
		{"testID:submit", `[data-testid="submit"]`, false},
		{"placeholder:Email", `[placeholder="Email"]`, false},
	}
	for _, testCase := range cases {
		got, isXPath, err := TranslateStringSelector(testCase.selector)
		if err != nil {
			t.Errorf("%q: unexpected error %v", testCase.selector, err)
			continue
		}
		if got != testCase.want || isXPath != testCase.isXPath {
			t.Errorf("%q: got (%q, xpath=%v), want (%q, xpath=%v)",
				testCase.selector, got, isXPath, testCase.want, testCase.isXPath)
		}
	}
}

func TestTranslateStringSelector_UnknownPrefixPassesThrough(t *testing.T) {
	got, _, err := TranslateStringSelector("role:button")
	if err != nil {
		t.Fatal(err)
	}
	if got != `[role="button"]` {
		t.Errorf("unknown prefix should map to attribute selector, got %q", got)
	}
}

func TestTranslateStringSelector_EscapesQuotes(t *testing.T) {
	got, _, err := TranslateStringSelector(`label:say "hi"`)
	if err != nil {
		t.Fatal(err)
	}
	if got != `[aria-label="say \"hi\""]` {
		t.Errorf("expected escaped quotes, got %q", got)
	}
}

func TestTranslateStringSelector_RejectsMissingPrefix(t *testing.T) {
	if _, _, err := TranslateStringSelector("foo"); err == nil {
		t.Error("expected error for missing prefix")
	}
	if _, _, err := TranslateStringSelector(""); err == nil {
		t.Error("expected error for empty selector")
	}
}

func TestTranslateStringSelector_RejectsUnsafePrefix(t *testing.T) {
	cases := []string{
		`foo]:has(*),body[x:value`,
		`x y:value`,
		`x"y:value`,
		`*:value`,
		`(:value`,
	}
	for _, selector := range cases {
		if _, _, err := TranslateStringSelector(selector); err == nil {
			t.Errorf("%q: expected error for unsafe prefix", selector)
		}
	}
}

func TestCSSEscape_ControlCharactersAndNUL(t *testing.T) {
	cases := []struct {
		name  string
		input string
		want  string
	}{
		{"plain ascii", "hello", "hello"},
		{"double quote", `say "hi"`, `say \"hi\"`},
		{"backslash", `a\b`, `a\\b`},
		{"newline", "a\nb", `a\A b`},
		{"carriage return", "a\rb", `a\D b`},
		{"form feed", "a\fb", `a\C b`},
		{"NUL replaced", "a\x00b", "a�b"},
		{"DEL", "a\x7Fb", `a\7F b`},
		{"non-ascii passes through", "café", "café"},
	}
	for _, testCase := range cases {
		got := cssEscape(testCase.input)
		if got != testCase.want {
			t.Errorf("%s: cssEscape(%q) = %q, want %q",
				testCase.name, testCase.input, got, testCase.want)
		}
	}
}