package main

import (
	"path/filepath"
	"strings"
	"testing"
)

func TestParseVendored(t *testing.T) {
	files, err := filepath.Glob("../../spec/xml/*.xml")
	if err != nil {
		t.Fatal(err)
	}
	if len(files) != 9 {
		t.Fatalf("got %d XML files, want 9", len(files))
	}
	for _, file := range files {
		t.Run(filepath.Base(file), func(t *testing.T) {
			n, err := parseFile(file)
			if err != nil {
				t.Fatal(err)
			}
			if n.Kind != "xcb" || len(n.Children) == 0 {
				t.Fatal("missing module AST")
			}
		})
	}
}
func TestParseRejects(t *testing.T) {
	cases := []struct{ name, body, want string }{
		{"unknown element", "<surprise/>", "unsupported element"},
		{"unknown attribute", `<xidtype name="WINDOW" surprise="yes"/>`, "unknown attribute"},
		{"misplaced known element", `<struct name="S"><reply/></struct>`, "unsupported <reply>"},
		{"sumof", `<request name="R" opcode="1"><list name="a" type="CARD8"><sumof ref="a"/></list></request>`, "sumof"},
		{"paramref", `<request name="R" opcode="1"><list name="a" type="CARD8"><paramref type="CARD8">a</paramref></list></request>`, "paramref"},
		{"case", `<request name="R" opcode="1"><switch name="s"><case/></switch></request>`, "case"},
		{"struct switch", `<struct name="S"><switch name="s"/></struct>`, "unsupported <switch>"},
		{"reply switch", `<request name="R" opcode="1"><reply><switch name="s"/></reply></request>`, "unsupported <switch>"},
		{"nested switch", `<request name="R" opcode="1"><switch name="s"><bitcase><switch name="s"/></bitcase></switch></request>`, "unsupported <switch>"},
		{"fd", `<request name="R" opcode="1"><fd name="fd"/></request>`, "fd"},
		{"eventstruct", `<eventstruct name="E"/>`, "eventstruct"},
		{"required alignment", `<struct name="S"><required_start_align align="4"/></struct>`, "required_start_align"},
		{"forward length", `<struct name="S"><list name="a" type="CARD8"><fieldref>n</fieldref></list><field name="n" type="CARD8"/></struct>`, "later field"},
		{"xge code", `<event name="E" number="128" xge="true"/>`, "evtype"},
		{"bad opcode", `<request name="R" opcode="256"/>`, "invalid opcode"},
		{"bad operator", `<request name="R" opcode="1"><exprfield name="x" type="CARD32"><op op="%"><value>1</value><value>2</value></op></exprfield></request>`, "unsupported operator"},
		{"wrong operands", `<request name="R" opcode="1"><exprfield name="x" type="CARD32"><op op="+"><value>1</value></op></exprfield></request>`, "two operands"},
		{"bad pad", `<struct name="S"><pad align="3"/></struct>`, "power of two"},
		{"namespace", `<xidtype xmlns="other" name="X"/>`, "unsupported element"},
		{"duplicate attr", `<xidtype name="X" name="Y"/>`, "duplicate attribute"},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			_, err := parseXML("bad.xml", strings.NewReader("<xcb header=\"test\">\n"+tc.body+"\n</xcb>"))
			if err == nil || !strings.Contains(err.Error(), tc.want) || !strings.Contains(err.Error(), "bad.xml:2:") {
				t.Fatalf("got %v, want file:line and %q", err, tc.want)
			}
		})
	}
}
func TestParseAST(t *testing.T) {
	n, err := parseXML("ok.xml", strings.NewReader(`<xcb header="a"><doc><arbitrary/></doc><request name="R" opcode="1"><field type="CARD32" name="n"/><list type="CARD8" name="data"><op op="*"><fieldref>n</fieldref><value>4</value></op></list></request></xcb>`))
	if err != nil {
		t.Fatal(err)
	}
	r := n.Children[1]
	if r.Attr["name"] != "R" || r.Children[1].Children[0].Children[0].Text != "n" {
		t.Fatal("AST lost order or expression")
	}
}
func FuzzParseXML(f *testing.F) {
	f.Add(`<xcb header="test"/>`)
	f.Fuzz(func(t *testing.T, raw string) { _, _ = parseXML("fuzz.xml", strings.NewReader(raw)) })
}