// Copyright 2023 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package htmlcheck

import (
	"bytes"
	"errors"
	"reflect"
	"strings"
	"testing"

	"golang.org/x/net/html"
)

func TestParse(t *testing.T) {
	var testCases = []struct {
		text         string
		wantSelector *selector
		wantErr      error
	}{
		{
			"a#id",
			&selector{
				atoms: []selectorAtom{&elementSelector{"a"}, &idSelector{"id"}},
			},
			nil,
		},
		{
			"a",
			&selector{atoms: []selectorAtom{&elementSelector{"a"}}},
			nil,
		},
		{
			"#id",
			&selector{atoms: []selectorAtom{&idSelector{"id"}}},
			nil,
		},
		{
			".class a",
			&selector{
				atoms: []selectorAtom{&classSelector{"class"}},
				next:  &selector{atoms: []selectorAtom{&elementSelector{"a"}}},
			},
			nil,
		},
		{
			`[attribute-name="value"] a`,
			&selector{
				atoms: []selectorAtom{&attributeSelector{attribute: "attribute-name", value: "value"}},
				next:  &selector{atoms: []selectorAtom{&elementSelector{"a"}}},
			},
			nil,
		},
		{
			`a[attribute-name="value"] a`,
			&selector{
				atoms: []selectorAtom{&elementSelector{"a"}, &attributeSelector{attribute: "attribute-name", value: "value"}},
				next:  &selector{atoms: []selectorAtom{&elementSelector{"a"}}},
			},
			nil,
		},
		{
			`.class1.class2`,
			&selector{
				atoms: []selectorAtom{&classSelector{"class1"}, &classSelector{"class2"}},
			},
			nil,
		},
		{
			`a.class1.class2`,
			&selector{
				atoms: []selectorAtom{&elementSelector{"a"}, &classSelector{"class1"}, &classSelector{"class2"}},
			},
			nil,
		},
		{
			".",
			nil,
			errors.New("no class name after '.'"),
		},
		{
			"#.",
			nil,
			errors.New("no id name after '#'"),
		},
		{
			"[]",
			nil,
			errors.New("expected attribute name after '[' in attribute selector"),
		},
		{
			"[attribute-name]",
			nil,
			errors.New("expected '=' after attribute name in attribute selector"),
		},
		{
			"[attribute-name=]",
			nil,
			errors.New("expected '\"' after = in attribute selector"),
		},
		{
			`[attribute-name=""]`,
			&selector{atoms: []selectorAtom{&attributeSelector{attribute: "attribute-name", value: ""}}},
			nil,
		},
		{
			`[attribute-name="]`,
			nil,
			errors.New("expected '\"' after attribute value"),
		},
		{
			`[attribute-name="value`,
			nil,
			errors.New("expected '\"' after attribute value"),
		},
		{
			`[attribute-name="value"`,
			nil,
			errors.New("expected ']' at end of attribute selector"),
		},
	}
	for _, tc := range testCases {
		sel, err := parse(tc.text)
		if tc.wantErr != nil {
			if err == nil {
				t.Fatalf("parse(%q): got nil err, want err %q", tc.text, tc.wantErr)
			}
			if tc.wantErr.Error() != err.Error() {
				t.Fatalf("parse(%q): got err %q, want err %q", tc.text, err, tc.wantErr)
			}
		} else if err != nil {
			t.Fatalf("parse(%q): got err %q, want nil error", tc.text, err)
		}
		if !reflect.DeepEqual(sel, tc.wantSelector) {
			t.Fatalf("parse(%q): got %v; want %v", tc.text, sel, tc.wantSelector)
		}
	}
}

func TestQuery(t *testing.T) {
	var testCases = []struct {
		queriedText string
		selector    string
		want        string
	}{
		{"<a></a>", "a", "<a></a>"},
		{`<a></a><a id="id">text</a>`, "a#id", `<a id="id">text</a>`},
		{`<a></a><a class="class1"></a><a class="class2"></a><a class="class1 class2">text</a>`, ".class1.class2", `<a class="class1 class2">text</a>`},
		{`<a></a><a class="class1">first</a><a class="class2"></a><a class="class1 class2">text</a>`, ".class1", `<a class="class1">first</a>`},
		{`<div><div></div><div my-attr="my-val">text</div></div>`, `[my-attr="my-val"]`, `<div my-attr="my-val">text</div>`},
		{`<div><div></div><div my-attr="my-val">text</div></div>`, `[myattr="my-val"]`, ""},
		{`<html><head></head><body><div><div></div><div my-attr="my-val">text</div></div></body></html>`, ``, `<html><head></head><body><div><div></div><div my-attr="my-val">text</div></div></body></html>`},
		{`<div></div><div><div>match me</div></div>`, "div div", `<div>match me</div>`},
		{`<div></div><div><div>wrong</div></div><div id="wrong-id"><div class="my-class">also wrong</div></div><div id="my-id"><div class="wrong-class">still wrong</div></div><div id="my-id"><div class="my-class">match</div></div>`, "div#my-id div.my-class", `<div class="my-class">match</div>`},
		{`<a></a><div class="UnitMeta-repo"><a href="foo" title="">link body</a></div>`, ".UnitMeta-repo a", `<a href="foo" title="">link body</a>`},
		{`<ul class="UnitFiles-fileList"><li><a href="foo">a.go</a></li></ul>`, ".UnitFiles-fileList a", `<a href="foo">a.go</a>`},
	}
	for _, tc := range testCases {
		n, err := html.Parse(strings.NewReader(tc.queriedText))
		if err != nil {
			t.Fatalf("parsing queried text %q: %v", tc.queriedText, err)
		}
		sel, err := parse(tc.selector)
		if err != nil {
			t.Fatalf("parsing selector %q: %v", tc.selector, err)
		}
		got := query(n, sel)
		if got == nil {
			if tc.want == "" {
				continue
			}
			t.Fatalf("query(%q, %q): got nil; want %q", tc.queriedText, tc.selector, tc.want)
		}
		var buf bytes.Buffer
		err = html.Render(&buf, got)
		if err != nil {
			t.Fatalf("rendering result of query: %v", err)
		}
		if buf.String() != tc.want {
			t.Fatalf("query(%q, %q): got %q; want %q", tc.queriedText, tc.selector, buf.String(), tc.want)
		}
	}
}