129 lines
2.6 KiB
Go
129 lines
2.6 KiB
Go
|
package goquery
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reflect"
|
||
|
|
"sort"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"golang.org/x/net/html"
|
||
|
|
)
|
||
|
|
|
||
|
|
var allNodes = `<!doctype html>
|
||
|
|
<html>
|
||
|
|
<head>
|
||
|
|
<meta a="b">
|
||
|
|
</head>
|
||
|
|
<body>
|
||
|
|
<p><!-- this is a comment -->
|
||
|
|
This is some text.
|
||
|
|
</p>
|
||
|
|
<div></div>
|
||
|
|
<h1 class="header"></h1>
|
||
|
|
<h2 class="header"></h2>
|
||
|
|
</body>
|
||
|
|
</html>`
|
||
|
|
|
||
|
|
func TestNodeName(t *testing.T) {
|
||
|
|
doc, err := NewDocumentFromReader(strings.NewReader(allNodes))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
n0 := doc.Nodes[0]
|
||
|
|
nDT := n0.FirstChild
|
||
|
|
sMeta := doc.Find("meta")
|
||
|
|
nMeta := sMeta.Get(0)
|
||
|
|
sP := doc.Find("p")
|
||
|
|
nP := sP.Get(0)
|
||
|
|
nComment := nP.FirstChild
|
||
|
|
nText := nComment.NextSibling
|
||
|
|
|
||
|
|
cases := []struct {
|
||
|
|
node *html.Node
|
||
|
|
typ html.NodeType
|
||
|
|
want string
|
||
|
|
}{
|
||
|
|
{n0, html.DocumentNode, nodeNames[html.DocumentNode]},
|
||
|
|
{nDT, html.DoctypeNode, "html"},
|
||
|
|
{nMeta, html.ElementNode, "meta"},
|
||
|
|
{nP, html.ElementNode, "p"},
|
||
|
|
{nComment, html.CommentNode, nodeNames[html.CommentNode]},
|
||
|
|
{nText, html.TextNode, nodeNames[html.TextNode]},
|
||
|
|
}
|
||
|
|
for i, c := range cases {
|
||
|
|
got := NodeName(newSingleSelection(c.node, doc))
|
||
|
|
if c.node.Type != c.typ {
|
||
|
|
t.Errorf("%d: want type %v, got %v", i, c.typ, c.node.Type)
|
||
|
|
}
|
||
|
|
if got != c.want {
|
||
|
|
t.Errorf("%d: want %q, got %q", i, c.want, got)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNodeNameMultiSel(t *testing.T) {
|
||
|
|
doc, err := NewDocumentFromReader(strings.NewReader(allNodes))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
in := []string{"p", "h1", "div"}
|
||
|
|
var out []string
|
||
|
|
doc.Find(strings.Join(in, ", ")).Each(func(i int, s *Selection) {
|
||
|
|
got := NodeName(s)
|
||
|
|
out = append(out, got)
|
||
|
|
})
|
||
|
|
sort.Strings(in)
|
||
|
|
sort.Strings(out)
|
||
|
|
if !reflect.DeepEqual(in, out) {
|
||
|
|
t.Errorf("want %v, got %v", in, out)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestOuterHtml(t *testing.T) {
|
||
|
|
doc, err := NewDocumentFromReader(strings.NewReader(allNodes))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
n0 := doc.Nodes[0]
|
||
|
|
nDT := n0.FirstChild
|
||
|
|
sMeta := doc.Find("meta")
|
||
|
|
sP := doc.Find("p")
|
||
|
|
nP := sP.Get(0)
|
||
|
|
nComment := nP.FirstChild
|
||
|
|
nText := nComment.NextSibling
|
||
|
|
sHeaders := doc.Find(".header")
|
||
|
|
|
||
|
|
cases := []struct {
|
||
|
|
node *html.Node
|
||
|
|
sel *Selection
|
||
|
|
want string
|
||
|
|
}{
|
||
|
|
{nDT, nil, "<!DOCTYPE html>"}, // render makes DOCTYPE all caps
|
||
|
|
{nil, sMeta, `<meta a="b"/>`}, // and auto-closes the meta
|
||
|
|
{nil, sP, `<p><!-- this is a comment -->
|
||
|
|
This is some text.
|
||
|
|
</p>`},
|
||
|
|
{nComment, nil, "<!-- this is a comment -->"},
|
||
|
|
{nText, nil, `
|
||
|
|
This is some text.
|
||
|
|
`},
|
||
|
|
{nil, sHeaders, `<h1 class="header"></h1>`},
|
||
|
|
}
|
||
|
|
for i, c := range cases {
|
||
|
|
if c.sel == nil {
|
||
|
|
c.sel = newSingleSelection(c.node, doc)
|
||
|
|
}
|
||
|
|
got, err := OuterHtml(c.sel)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if got != c.want {
|
||
|
|
t.Errorf("%d: want %q, got %q", i, c.want, got)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|