package cascadia import ( "strings" "testing" "golang.org/x/net/html" ) type selectorTest struct { HTML, selector string results []string } func nodeString(n *html.Node) string { switch n.Type { case html.TextNode: return n.Data case html.ElementNode: return html.Token{ Type: html.StartTagToken, Data: n.Data, Attr: n.Attr, }.String() } return "" } var selectorTests = []selectorTest{ { `

This address...

`, "address", []string{ "

", }, }, { ``, "*", []string{ "", "", "", "", }, }, { `

`, "#foo", []string{ `

`, }, }, { `

`, "li#t1", []string{ `
`, }, }, { `
2. `, "*#t4", []string{ `
3. `, }, }, { `
  - `, ".t1", []string{ `
  - `, }, }, { `
    `, "p.t1", []string{ `
    `, }, }, { `
    `, "div.teST", []string{}, }, { `
    `, ".t1.fail", []string{}, }, { `
    `, "p.t1.t2", []string{ `
    `, }, }, { `
    `, "p[title]", []string{ `
    `, }, }, { `
    `, `address[title="foo"]`, []string{ `
    `, }, }, { `
    `, `[ title ~= foo ]`, []string{ `
    `, }, }, { `
    `, `[title~="hello world"]`, []string{}, }, { `
    `, `[lang|="en"]`, []string{ `
    `, `
    `, }, }, { `
    `, `[title^="foo"]`, []string{ `
    `, }, }, { `
    `, `[title$="bar"]`, []string{ `
    `, }, }, { `
    `, `[title*="bar"]`, []string{ `
    `, }, }, { `
    `, ".t1:not(.t2)", []string{}, }, { `
    `, `div:not(.t1)`, []string{ `
    `, }, }, { `
    `, `li:nth-child(odd)`, []string{ `
  - `, `
  - `, }, }, { `
    `, `li:nth-child(even)`, []string{ `
  - `, }, }, { `
    `, `li:nth-child(-n+2)`, []string{ `
  - `, `
  - `, }, }, { `
    `, `li:nth-child(3n+1)`, []string{ `
  - `, }, }, { `
    `, `li:nth-last-child(odd)`, []string{ `
  - `, `
  - `, }, }, { `
    `, `li:nth-last-child(even)`, []string{ `
  - `, `
  - `, }, }, { `
    `, `li:nth-last-child(-n+2)`, []string{ `
  - `, `
  - `, }, }, { `
    `, `li:nth-last-child(3n+1)`, []string{ `
  - `, `
  - `, }, }, { `
    some text and a span and another
    `, `span:first-child`, []string{ `
    `, }, }, { `a span and some text`, `span:last-child`, []string{ `
    `, }, }, { `
    `, `p:nth-of-type(2)`, []string{ `
    `, }, }, { `
    `, `p:nth-last-of-type(2)`, []string{ `
    `, }, }, { `
    `, `p:last-of-type`, []string{ `
    `, }, }, { `
    `, `p:first-of-type`, []string{ `
    `, }, }, { `
    `, `p:only-child`, []string{ `
    `, }, }, { `
    `, `p:only-of-type`, []string{ `
    `, }, }, { `
    Hello
    `, `:empty`, []string{ ``, `
    `, ``, }, }, { `
    `, `div p`, []string{ `
    `, `
    `, }, }, { `
    `, `div table p`, []string{ `
    `, }, }, { `
    `, `div > p`, []string{ `
    `, `
    `, }, }, { `
    `, `p ~ p`, []string{ `
    `, `
    `, }, }, { `
    
    `, `p + p`, []string{ `
    `, }, }, { `
    `, `li, p`, []string{ "
  - ", "
  - ", "
    ", }, }, { `
    `, `p +/*This is a comment*/ p`, []string{ `
    `, }, }, { `
    Text block that wraps inner text and continues
    `, `p:contains("that wraps")`, []string{ `
    `, }, }, { `
    Text block that wraps inner text and continues
    `, `p:containsOwn("that wraps")`, []string{}, }, { `
    Text block that wraps inner text and continues
    `, `:containsOwn("inner")`, []string{ `
    `, }, }, { `
    Text block that wraps inner text and continues
    `, `p:containsOwn("block")`, []string{ `
    `, }, }, { `
    text content
    `, `div:has(#p1)`, []string{ `
    `, }, }, { `
    contents 1
    
    contents 2
    `, `div:has(:containsOwn("2"))`, []string{ `
    `, }, }, { `
    contents 1
    
    contents 2
    `, `body :has(:containsOwn("2"))`, []string{ `
    `, `
    `, }, }, { `
    contents 1
    
    contents 2
    `, `body :haschild(:containsOwn("2"))`, []string{ `
    `, }, }, { `
    0123456789
    abcdef
    0123ABCD
    `, `p:matches([\d])`, []string{ `
    `, `
    `, }, }, { `
    0123456789
    abcdef
    0123ABCD
    `, `p:matches([a-z])`, []string{ `
    `, }, }, { `
    0123456789
    abcdef
    0123ABCD
    `, `p:matches([a-zA-Z])`, []string{ `
    `, `
    `, }, }, { `
    0123456789
    abcdef
    0123ABCD
    `, `p:matches([^\d])`, []string{ `
    `, `
    `, }, }, { `
    0123456789
    abcdef
    0123ABCD
    `, `p:matches(^(0|a))`, []string{ `
    `, `
    `, `
    `, }, }, { `
    0123456789
    abcdef
    0123ABCD
    `, `p:matches(^\d+$)`, []string{ `
    `, }, }, { `
    0123456789
    abcdef
    0123ABCD
    `, `p:not(:matches(^\d+$))`, []string{ `
    `, `
    `, }, }, { `
    0123456789
    `, `div :matchesOwn(^\d+$)`, []string{ `
    `, ``, }, }, { `
    
    `, `[href#=(fina)]:not([href#=(\/\/[^\/]+untrusted)])`, []string{ ``, ``, }, }, { `
    
    `, `[href#=(^https:\/\/[^\/]*\/?news)]`, []string{ ``, }, }, { `
    Username Password Country Bio
    `, `:input`, []string{ ``, ``, ``, `<button>`, }, }, } func TestSelectors(t *testing.T) { for _, test := range selectorTests { s, err := Compile(test.selector) if err != nil { t.Errorf("error compiling %q: %s", test.selector, err) continue } doc, err := html.Parse(strings.NewReader(test.HTML)) if err != nil { t.Errorf("error parsing %q: %s", test.HTML, err) continue } matches := s.MatchAll(doc) if len(matches) != len(test.results) { t.Errorf("wanted %d elements, got %d instead", len(test.results), len(matches)) continue } for i, m := range matches { got := nodeString(m) if got != test.results[i] { t.Errorf("wanted %s, got %s instead", test.results[i], got) } } firstMatch := s.MatchFirst(doc) if len(test.results) == 0 { if firstMatch != nil { t.Errorf("MatchFirst: want nil, got %s", nodeString(firstMatch)) } } else { got := nodeString(firstMatch) if got != test.results[0] { t.Errorf("MatchFirst: want %s, got %s", test.results[0], got) } } } }