-
Notifications
You must be signed in to change notification settings - Fork 36
/
Copy pathparser_test.go
64 lines (60 loc) · 1.63 KB
/
parser_test.go
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
package gohtml
import (
"strings"
"testing"
)
func TestParse(t *testing.T) {
s := `<!DOCTYPE html><html><head><title>This is a title.</title></head><body><p>Line1<br>Line2</p><br/></body></html><!-- aaa --><a>`
htmlDoc := parse(strings.NewReader(s))
actual := htmlDoc.html()
expected := `<!DOCTYPE html>
<html>
<head>
<title>
This is a title.
</title>
</head>
<body>
<p>
Line1
<br>
Line2
</p>
<br/>
</body>
</html>
<!-- aaa -->
<a>`
if actual != expected {
t.Errorf("Invalid result. [expected: %s][actual: %s]", expected, actual)
}
}
func TestAppendElement(t *testing.T) {
htmlDoc := &htmlDocument{}
tagElem := &tagElement{}
textElem := &textElement{text: "test text"}
appendElement(htmlDoc, tagElem, textElem)
if len(tagElem.children) != 1 || tagElem.children[0] != textElem {
t.Errorf("tagElement.children is invalid. [expected: %+v][actual: %+v]", []element{textElem}, tagElem.children)
}
appendElement(htmlDoc, nil, textElem)
if len(htmlDoc.elements) != 1 || htmlDoc.elements[0] != textElem {
t.Errorf("htmlDocument.elements is invalid. [expected: %+v][actual: %+v]", []element{textElem}, htmlDoc.elements)
}
}
func TestHtmlEscape(t *testing.T) {
s := `<!DOCTYPE html><html><body><div>0 < 1. great insight! </sarcasm> over&out.&</div></body></html>`
expected := `<!DOCTYPE html>
<html>
<body>
<div>
0 < 1. great insight! </sarcasm> over&out.&
</div>
</body>
</html>`
htmlDoc := parse(strings.NewReader(s))
actual := htmlDoc.html()
if actual != expected {
t.Errorf("Invalid result. [expected: %s][actual: %s]", expected, actual)
}
}