1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
package html
import (
"strings"
"testing"
base "src.userspace.com.au/query"
)
func TestParse(t *testing.T) {
src := `<html><body><p>One</p><p>Two</p></body></html>`
doc, err := Parse(strings.NewReader(src))
if err != nil {
t.Fatalf("Expected no error but got %s", err)
}
if doc == nil {
t.Fatal("Expected node but got nil")
}
//doc.PrintTree(0)
// document
nt := doc.Type()
nd := doc.Data()
if nt != base.DocumentNode {
t.Fatalf("Expected %q but got %q", "DocumentNode", nt)
}
if nd != "" {
t.Fatalf("Expected %q but got %q", "", nd)
}
// get <html>
n := doc.FirstChild()
if n == nil {
t.Fatal("Expected node but got nil")
}
nt = n.Type()
nd = n.Data()
if nt != base.ElementNode {
t.Fatalf("Expected %q but got %q", "ElementNode", nd)
}
if nd != "html" {
t.Fatalf("Expected %q but got %q", "html", nd)
}
// get <body>
//n = n.FirstChild()
// TODO why?
n = n.LastChild()
if n == nil {
t.Fatal("Expected node but got nil")
}
nt = n.Type()
nd = n.Data()
if nt != base.ElementNode {
t.Fatalf("Expected %q but got %q", "ElementNode", nd)
}
if nd != "body" {
t.Fatalf("Expected %q but got %q", "body", nd)
}
// get first <p>
n = n.LastChild()
if n == nil {
t.Fatal("Expected node but got nil")
}
nt = n.Type()
nd = n.Data()
if nt != base.ElementNode {
t.Fatalf("Expected %q but got %q", "ElementNode", nd)
}
if nd != "p" {
t.Fatalf("Expected %q but got %q", "p", nd)
}
}
|
