497 lines
13 KiB
Go
497 lines
13 KiB
Go
package gxpath
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
"testing"
|
|
|
|
"goutil/xmlUtil/gxpath/xpath"
|
|
)
|
|
|
|
var html *TNode = example()
|
|
|
|
/*
|
|
testXPath2(t, html, "//title/node()", 1) still have some error,will fix in future.
|
|
testXPath(t, html, "//*[count(*)=3]", "ul")
|
|
*/
|
|
|
|
func TestSelf(t *testing.T) {
|
|
testXPath(t, html, ".", "html")
|
|
testXPath(t, html.FirstChild, ".", "head")
|
|
testXPath(t, html, "self::*", "html")
|
|
testXPath(t, html.LastChild, "self::body", "body")
|
|
testXPath2(t, html, "//body/./ul/li/a", 3)
|
|
}
|
|
|
|
func TestParent(t *testing.T) {
|
|
testXPath(t, html.LastChild, "..", "html")
|
|
testXPath(t, html.LastChild, "parent::*", "html")
|
|
a := selectNode(html, "//li/a")
|
|
testXPath(t, a, "parent::*", "li")
|
|
testXPath(t, html, "//title/parent::head", "head")
|
|
}
|
|
|
|
func TestAttribute(t *testing.T) {
|
|
testXPath(t, html, "@lang='en'", "html")
|
|
testXPath2(t, html, "@lang='zh'", 0)
|
|
testXPath2(t, html, "//@href", 3)
|
|
testXPath2(t, html, "//a[@*]", 3)
|
|
}
|
|
|
|
func TestRelativePath(t *testing.T) {
|
|
testXPath(t, html, "head", "head")
|
|
testXPath(t, html, "/head", "head")
|
|
testXPath(t, html, "body//li", "li")
|
|
testXPath(t, html, "/head/title", "title")
|
|
|
|
testXPath2(t, html, "/body/ul/li/a", 3)
|
|
testXPath(t, html, "//title", "title")
|
|
testXPath(t, html, "//title/..", "head")
|
|
testXPath(t, html, "//title/../..", "html")
|
|
testXPath2(t, html, "//a[@href]", 3)
|
|
testXPath(t, html, "//ul/../footer", "footer")
|
|
}
|
|
|
|
func TestChild(t *testing.T) {
|
|
testXPath(t, html, "/child::head", "head")
|
|
testXPath(t, html, "/child::head/child::title", "title")
|
|
testXPath(t, html, "//title/../child::title", "title")
|
|
testXPath(t, html.Parent, "//child::*", "html")
|
|
}
|
|
|
|
func TestDescendant(t *testing.T) {
|
|
testXPath2(t, html, "descendant::*", 15)
|
|
testXPath2(t, html, "/head/descendant::*", 2)
|
|
testXPath2(t, html, "//ul/descendant::*", 7) // <li> + <a>
|
|
testXPath2(t, html, "//ul/descendant::li", 4) // <li>
|
|
}
|
|
|
|
func TestAncestor(t *testing.T) {
|
|
testXPath2(t, html, "/body/footer/ancestor::*", 2) // body>html
|
|
testXPath2(t, html, "/body/ul/li/a/ancestor::li", 3)
|
|
}
|
|
|
|
func TestFollowingSibling(t *testing.T) {
|
|
var list []*TNode
|
|
list = selectNodes(html, "//li/following-sibling::*")
|
|
for _, n := range list {
|
|
if n.Data != "li" {
|
|
t.Fatalf("expected node is li,but got:%s", n.Data)
|
|
}
|
|
}
|
|
|
|
list = selectNodes(html, "//ul/following-sibling::*") // p,footer
|
|
for _, n := range list {
|
|
if n.Data != "p" && n.Data != "footer" {
|
|
t.Fatal("expected node is not one of the following nodes: [p,footer]")
|
|
}
|
|
}
|
|
testXPath(t, html, "//ul/following-sibling::footer", "footer")
|
|
list = selectNodes(html, "//h1/following::*") // ul>li>a,p,footer
|
|
if list[0].Data != "ul" {
|
|
t.Fatal("expected node is not ul")
|
|
}
|
|
if list[1].Data != "li" {
|
|
t.Fatal("expected node is not li")
|
|
}
|
|
if list[len(list)-1].Data != "footer" {
|
|
t.Fatal("expected node is not footer")
|
|
}
|
|
}
|
|
|
|
func TestPrecedingSibling(t *testing.T) {
|
|
testXPath(t, html, "/body/footer/preceding-sibling::*", "p")
|
|
testXPath2(t, html, "/body/footer/preceding-sibling::*", 3) // p,ul,h1
|
|
list := selectNodes(html, "//h1/preceding::*") // head>title>meta
|
|
if list[0].Data != "head" {
|
|
t.Fatal("expected is not head")
|
|
}
|
|
if list[1].Data != "title" {
|
|
t.Fatal("expected is not title")
|
|
}
|
|
if list[2].Data != "meta" {
|
|
t.Fatal("expected is not meta")
|
|
}
|
|
}
|
|
|
|
func TestStarWide(t *testing.T) {
|
|
testXPath(t, html, "/head/*", "title")
|
|
testXPath2(t, html, "//ul/*", 4)
|
|
testXPath(t, html, "@*", "html")
|
|
testXPath2(t, html, "/body/h1/*", 0)
|
|
testXPath2(t, html, `//ul/*/a`, 3)
|
|
}
|
|
|
|
func TestNodeTestType(t *testing.T) {
|
|
testXPath(t, html, "//title/text()", "Hello")
|
|
testXPath(t, html, "//a[@href='/']/text()", "Home")
|
|
testXPath2(t, html, "//head/node()", 2)
|
|
testXPath2(t, html, "//ul/node()", 4)
|
|
}
|
|
|
|
func TestPosition(t *testing.T) {
|
|
testXPath3(t, html, "/head[1]", html.FirstChild) // compare to 'head' element
|
|
ul := selectNode(html, "//ul")
|
|
testXPath3(t, html, "/head[last()]", html.FirstChild)
|
|
testXPath3(t, html, "//li[1]", ul.FirstChild)
|
|
testXPath3(t, html, "//li[4]", ul.LastChild)
|
|
testXPath3(t, html, "//li[last()]", ul.LastChild)
|
|
}
|
|
|
|
func TestPredicate(t *testing.T) {
|
|
testXPath(t, html.Parent, "html[@lang='en']", "html")
|
|
testXPath(t, html, "//a[@href='/']", "a")
|
|
testXPath(t, html, "//meta[@name]", "meta")
|
|
ul := selectNode(html, "//ul")
|
|
testXPath3(t, html, "//li[position()=4]", ul.LastChild)
|
|
testXPath3(t, html, "//li[position()=1]", ul.FirstChild)
|
|
testXPath2(t, html, "//li[position()>0]", 4)
|
|
testXPath3(t, html, "//a[text()='Home']", selectNode(html, "//a[1]"))
|
|
}
|
|
|
|
func TestOr_And(t *testing.T) {
|
|
list := selectNodes(html, "//h1|//footer")
|
|
if len(list) == 0 {
|
|
t.Fatal("//h1|//footer no any node found")
|
|
}
|
|
if list[0].Data != "h1" {
|
|
t.Fatalf("expected first node of node-set is h1,but got %s", list[0].Data)
|
|
}
|
|
if list[1].Data != "footer" {
|
|
t.Fatalf("expected first node of node-set is footer,but got %s", list[1].Data)
|
|
}
|
|
|
|
list = selectNodes(html, "//a[@id=1 or @id=2]")
|
|
if list[0] != selectNode(html, "//a[@id=1]") {
|
|
t.Fatal("node is not equal")
|
|
}
|
|
if list[1] != selectNode(html, "//a[@id=2]") {
|
|
t.Fatal("node is not equal")
|
|
}
|
|
testXPath3(t, html, "//a[@id=1 and @href='/']", selectNode(html, "//a[1]"))
|
|
testXPath3(t, html, "//a[text()='Home' and @id='1']", selectNode(html, "//a[1]"))
|
|
}
|
|
|
|
func TestFunction(t *testing.T) {
|
|
testXPath2(t, html, "//*[name()='a']", 3)
|
|
testXPath(t, html, "//*[starts-with(name(),'h1')]", "h1")
|
|
testXPath2(t, html, "//*[starts-with(@href,'/a')]", 2) // a links: `/account`,`/about`
|
|
testXPath3(t, html, "//h1[normalize-space(text())='This is a H1']", selectNode(html, "//h1"))
|
|
testXPath3(t, html, "//title[substring(.,0)='Hello']", selectNode(html, "//title"))
|
|
testXPath3(t, html, "//title[substring(text(),0,4)='Hell']", selectNode(html, "//title"))
|
|
}
|
|
|
|
func TestOperationOrLogical(t *testing.T) {
|
|
testXPath3(t, html, "//li[1+1]", selectNode(html, "//li[2]"))
|
|
testXPath3(t, html, "//li[5 div 2]", selectNode(html, "//li[2]"))
|
|
testXPath3(t, html, "//li[3 mod 2]", selectNode(html, "//li[1]"))
|
|
testXPath3(t, html, "//li[3 - 2]", selectNode(html, "//li[1]"))
|
|
testXPath2(t, html, "//li[position() mod 2 = 0 ]", 2) // //li[2],li[4]
|
|
testXPath2(t, html, "//a[@id>=1]", 3) // //a[@id>=1] == a[1],a[2],a[3]
|
|
testXPath2(t, html, "//a[@id<2]", 1) // //a[@id>=1] == a[1]
|
|
testXPath2(t, html, "//a[@id!=2]", 2) // //a[@id>=1] == a[1],a[3]
|
|
testXPath2(t, html, "//a[@id=1 or @id=3]", 2) // //a[@id>=1] == a[1],a[3]
|
|
testXPath3(t, html, "//a[@id=1 and @href='/']", selectNode(html, "//a[1]"))
|
|
}
|
|
|
|
func testXPath(t *testing.T, root *TNode, expr string, expected string) {
|
|
node := selectNode(root, expr)
|
|
if node == nil {
|
|
t.Fatalf("`%s` returns node is nil", expr)
|
|
}
|
|
if node.Data != expected {
|
|
t.Fatalf("`%s` expected node is %s,but got %s", expr, expected, node.Data)
|
|
}
|
|
}
|
|
|
|
func testXPath2(t *testing.T, root *TNode, expr string, expected int) {
|
|
list := selectNodes(root, expr)
|
|
if len(list) != expected {
|
|
t.Fatalf("`%s` expected node numbers is %d,but got %d", expr, expected, len(list))
|
|
}
|
|
}
|
|
|
|
func testXPath3(t *testing.T, root *TNode, expr string, expected *TNode) {
|
|
node := selectNode(root, expr)
|
|
if node == nil {
|
|
t.Fatalf("`%s` returns node is nil", expr)
|
|
}
|
|
if node != expected {
|
|
t.Fatalf("`%s` %s != %s", expr, node.Value(), expected.Value())
|
|
}
|
|
}
|
|
|
|
func selectNode(root *TNode, expr string) (n *TNode) {
|
|
t := Select(createNavigator(root), expr)
|
|
if t.MoveNext() {
|
|
n = (t.Current().(*TNodeNavigator)).curr
|
|
}
|
|
return n
|
|
}
|
|
|
|
func selectNodes(root *TNode, expr string) []*TNode {
|
|
t := Select(createNavigator(root), expr)
|
|
var nodes []*TNode
|
|
for t.MoveNext() {
|
|
node := (t.Current().(*TNodeNavigator)).curr
|
|
nodes = append(nodes, node)
|
|
}
|
|
return nodes
|
|
}
|
|
|
|
func createNavigator(n *TNode) *TNodeNavigator {
|
|
return &TNodeNavigator{curr: n, root: n, attr: -1}
|
|
}
|
|
|
|
type Attribute struct {
|
|
Key, Value string
|
|
}
|
|
|
|
type TNode struct {
|
|
Parent, FirstChild, LastChild, PrevSibling, NextSibling *TNode
|
|
|
|
Type xpath.NodeType
|
|
Data string
|
|
Attr []Attribute
|
|
}
|
|
|
|
func (n *TNode) Value() string {
|
|
if n.Type == xpath.TextNode {
|
|
return n.Data
|
|
}
|
|
|
|
var buff bytes.Buffer
|
|
var output func(*TNode)
|
|
output = func(node *TNode) {
|
|
if node.Type == xpath.TextNode {
|
|
buff.WriteString(node.Data)
|
|
}
|
|
for child := node.FirstChild; child != nil; child = child.NextSibling {
|
|
output(child)
|
|
}
|
|
}
|
|
output(n)
|
|
return buff.String()
|
|
}
|
|
|
|
// TNodeNavigator is for navigating TNode.
|
|
type TNodeNavigator struct {
|
|
curr, root *TNode
|
|
attr int
|
|
}
|
|
|
|
func (n *TNodeNavigator) NodeType() xpath.NodeType {
|
|
if n.curr.Type == xpath.ElementNode && n.attr != -1 {
|
|
return xpath.AttributeNode
|
|
}
|
|
return n.curr.Type
|
|
}
|
|
|
|
func (n *TNodeNavigator) LocalName() string {
|
|
if n.attr != -1 {
|
|
return n.curr.Attr[n.attr].Key
|
|
}
|
|
return n.curr.Data
|
|
}
|
|
|
|
func (n *TNodeNavigator) Prefix() string {
|
|
return ""
|
|
}
|
|
|
|
func (n *TNodeNavigator) Value() string {
|
|
switch n.curr.Type {
|
|
case xpath.CommentNode:
|
|
return n.curr.Data
|
|
case xpath.ElementNode:
|
|
if n.attr != -1 {
|
|
return n.curr.Attr[n.attr].Value
|
|
}
|
|
var buf bytes.Buffer
|
|
node := n.curr.FirstChild
|
|
for node != nil {
|
|
if node.Type == xpath.TextNode {
|
|
buf.WriteString(strings.TrimSpace(node.Data))
|
|
}
|
|
node = node.NextSibling
|
|
}
|
|
return buf.String()
|
|
case xpath.TextNode:
|
|
return n.curr.Data
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (n *TNodeNavigator) Copy() xpath.NodeNavigator {
|
|
n2 := *n
|
|
return &n2
|
|
}
|
|
|
|
func (n *TNodeNavigator) MoveToRoot() {
|
|
n.curr = n.root
|
|
}
|
|
|
|
func (n *TNodeNavigator) MoveToParent() bool {
|
|
if node := n.curr.Parent; node != nil {
|
|
n.curr = node
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (n *TNodeNavigator) MoveToNextAttribute() bool {
|
|
if n.attr >= len(n.curr.Attr)-1 {
|
|
return false
|
|
}
|
|
n.attr++
|
|
return true
|
|
}
|
|
|
|
func (n *TNodeNavigator) MoveToChild() bool {
|
|
if node := n.curr.FirstChild; node != nil {
|
|
n.curr = node
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (n *TNodeNavigator) MoveToFirst() bool {
|
|
if n.curr.PrevSibling == nil {
|
|
return false
|
|
}
|
|
for {
|
|
node := n.curr.PrevSibling
|
|
if node == nil {
|
|
break
|
|
}
|
|
n.curr = node
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (n *TNodeNavigator) String() string {
|
|
return n.Value()
|
|
}
|
|
|
|
func (n *TNodeNavigator) MoveToNext() bool {
|
|
if node := n.curr.NextSibling; node != nil {
|
|
n.curr = node
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (n *TNodeNavigator) MoveToPrevious() bool {
|
|
if node := n.curr.PrevSibling; node != nil {
|
|
n.curr = node
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (n *TNodeNavigator) MoveTo(other xpath.NodeNavigator) bool {
|
|
node, ok := other.(*TNodeNavigator)
|
|
if !ok || node.root != n.root {
|
|
return false
|
|
}
|
|
|
|
n.curr = node.curr
|
|
n.attr = node.attr
|
|
return true
|
|
}
|
|
|
|
func createNode(data string, typ xpath.NodeType) *TNode {
|
|
return &TNode{Data: data, Type: typ, Attr: make([]Attribute, 0)}
|
|
}
|
|
|
|
func (t *TNode) createChildNode(data string, typ xpath.NodeType) *TNode {
|
|
n := createNode(data, typ)
|
|
n.Parent = t
|
|
if t.FirstChild == nil {
|
|
t.FirstChild = n
|
|
} else {
|
|
t.LastChild.NextSibling = n
|
|
n.PrevSibling = t.LastChild
|
|
}
|
|
t.LastChild = n
|
|
return n
|
|
}
|
|
|
|
func (t *TNode) appendNode(data string, typ xpath.NodeType) *TNode {
|
|
n := createNode(data, typ)
|
|
n.Parent = t.Parent
|
|
t.NextSibling = n
|
|
n.PrevSibling = t
|
|
if t.Parent != nil {
|
|
t.Parent.LastChild = n
|
|
}
|
|
return n
|
|
}
|
|
|
|
func (t *TNode) addAttribute(k, v string) {
|
|
t.Attr = append(t.Attr, Attribute{k, v})
|
|
}
|
|
|
|
func example() *TNode {
|
|
/*
|
|
<html lang="en">
|
|
<head>
|
|
<title>Hello</title>
|
|
<meta name="language" content="en"/>
|
|
</head>
|
|
<body>
|
|
<h1> This is a H1 </h1>
|
|
<ul>
|
|
<li><a id="1" href="/">Home</a></li>
|
|
<li><a id="2" href="/about">about</a></li>
|
|
<li><a id="3" href="/account">login</a></li>
|
|
<li></li>
|
|
</ul>
|
|
<p>
|
|
Hello,This is an example for gxpath.
|
|
</p>
|
|
<footer>footer script</footer>
|
|
</body>
|
|
</html>
|
|
*/
|
|
doc := createNode("", xpath.RootNode)
|
|
xhtml := doc.createChildNode("html", xpath.ElementNode)
|
|
xhtml.addAttribute("lang", "en")
|
|
|
|
// The HTML head section.
|
|
head := xhtml.createChildNode("head", xpath.ElementNode)
|
|
n := head.createChildNode("title", xpath.ElementNode)
|
|
n = n.createChildNode("Hello", xpath.TextNode)
|
|
n = head.createChildNode("meta", xpath.ElementNode)
|
|
n.addAttribute("name", "language")
|
|
n.addAttribute("content", "en")
|
|
// The HTML body section.
|
|
body := xhtml.createChildNode("body", xpath.ElementNode)
|
|
n = body.createChildNode("h1", xpath.ElementNode)
|
|
n = n.createChildNode(" This is a H1 ", xpath.TextNode)
|
|
ul := body.createChildNode("ul", xpath.ElementNode)
|
|
n = ul.createChildNode("li", xpath.ElementNode)
|
|
n = n.createChildNode("a", xpath.ElementNode)
|
|
n.addAttribute("id", "1")
|
|
n.addAttribute("href", "/")
|
|
n = n.createChildNode("Home", xpath.TextNode)
|
|
n = ul.createChildNode("li", xpath.ElementNode)
|
|
n = n.createChildNode("a", xpath.ElementNode)
|
|
n.addAttribute("id", "2")
|
|
n.addAttribute("href", "/about")
|
|
n = n.createChildNode("about", xpath.TextNode)
|
|
n = ul.createChildNode("li", xpath.ElementNode)
|
|
n = n.createChildNode("a", xpath.ElementNode)
|
|
n.addAttribute("id", "3")
|
|
n.addAttribute("href", "/account")
|
|
n = n.createChildNode("login", xpath.TextNode)
|
|
n = ul.createChildNode("li", xpath.ElementNode)
|
|
|
|
n = body.createChildNode("p", xpath.ElementNode)
|
|
n = n.createChildNode("Hello,This is an example for gxpath.", xpath.TextNode)
|
|
|
|
n = body.createChildNode("footer", xpath.ElementNode)
|
|
n = n.createChildNode("footer script", xpath.TextNode)
|
|
|
|
return xhtml
|
|
}
|