401 lines
9.7 KiB
Go
401 lines
9.7 KiB
Go
// This work is subject to the CC0 1.0 Universal (CC0 1.0) Public Domain Dedication
|
||
// license. Its contents can be found at:
|
||
// http://creativecommons.org/publicdomain/zero/1.0/
|
||
|
||
package xmlx
|
||
|
||
import (
|
||
"encoding/xml"
|
||
"testing"
|
||
)
|
||
|
||
func TestLoadLocal(t *testing.T) {
|
||
doc := New()
|
||
|
||
if err := doc.LoadFile("test.xml", nil); err != nil {
|
||
t.Error(err.Error())
|
||
return
|
||
}
|
||
|
||
if len(doc.Root.Children) == 0 {
|
||
t.Errorf("Root node has no children.")
|
||
return
|
||
}
|
||
}
|
||
|
||
func TestWildcard(t *testing.T) {
|
||
doc := New()
|
||
|
||
if err := doc.LoadFile("test2.xml", nil); err != nil {
|
||
t.Error(err.Error())
|
||
return
|
||
}
|
||
|
||
list := doc.SelectNodes("ns", "*")
|
||
|
||
if len(list) != 1 {
|
||
t.Errorf("Wrong number of child elements. Expected 1, got %d.", len(list))
|
||
return
|
||
}
|
||
}
|
||
|
||
func TestWildcardRecursive(t *testing.T) {
|
||
doc := New()
|
||
|
||
if err := doc.LoadFile("test2.xml", nil); err != nil {
|
||
t.Error(err.Error())
|
||
return
|
||
}
|
||
|
||
list := doc.SelectNodesRecursive("ns", "*")
|
||
|
||
if len(list) != 7 {
|
||
t.Errorf("Wrong number of child elements. Expected 7, got %d.", len(list))
|
||
return
|
||
}
|
||
}
|
||
|
||
func _TestLoadRemote(t *testing.T) {
|
||
doc := New()
|
||
|
||
if err := doc.LoadUri("http://blog.golang.org/feeds/posts/default", nil); err != nil {
|
||
t.Error(err.Error())
|
||
return
|
||
}
|
||
|
||
if len(doc.Root.Children) == 0 {
|
||
t.Errorf("Root node has no children.")
|
||
return
|
||
}
|
||
}
|
||
|
||
func TestSave(t *testing.T) {
|
||
doc := New()
|
||
|
||
if err := doc.LoadFile("test.xml", nil); err != nil {
|
||
t.Errorf("LoadFile(): %s", err)
|
||
return
|
||
}
|
||
|
||
IndentPrefix = "\t"
|
||
if err := doc.SaveFile("test1.xml"); err != nil {
|
||
t.Errorf("SaveFile(): %s", err)
|
||
return
|
||
}
|
||
}
|
||
|
||
func TestNodeSearch(t *testing.T) {
|
||
doc := New()
|
||
|
||
if err := doc.LoadFile("test1.xml", nil); err != nil {
|
||
t.Errorf("LoadFile(): %s", err)
|
||
return
|
||
}
|
||
|
||
if node := doc.SelectNode("", "item"); node == nil {
|
||
t.Errorf("SelectNode(): No node found.")
|
||
return
|
||
}
|
||
|
||
nodes := doc.SelectNodes("", "item")
|
||
if len(nodes) == 0 {
|
||
t.Errorf("SelectNodes(): no nodes found.")
|
||
return
|
||
}
|
||
|
||
ch := doc.SelectNode("", "channel")
|
||
// Test that SelectNodes properly selects multiple nodes
|
||
links := ch.SelectNodes("", "link")
|
||
if len(links) != 8 {
|
||
t.Errorf("SelectNodes(): Expected 8, Got %d", len(links))
|
||
return
|
||
}
|
||
}
|
||
|
||
type Image struct {
|
||
Title string `xml:"title"`
|
||
Url string `xml:"url"`
|
||
Link string `xml:"link"`
|
||
Description string `xml:"description"`
|
||
Width int `xml:"width"`
|
||
Height int `xml:"height"`
|
||
}
|
||
|
||
func TestUnmarshal(t *testing.T) {
|
||
doc := New()
|
||
err := doc.LoadFile("test1.xml", nil)
|
||
|
||
if err != nil {
|
||
t.Errorf("LoadFile(): %s", err)
|
||
return
|
||
}
|
||
|
||
node := doc.SelectNode("", "image")
|
||
if node == nil {
|
||
t.Errorf("SelectNode(): No node found.")
|
||
return
|
||
}
|
||
|
||
var img Image
|
||
if err = node.Unmarshal(&img); err != nil {
|
||
t.Errorf("Unmarshal(): %s", err)
|
||
return
|
||
}
|
||
|
||
if img.Title != "WriteTheWeb" {
|
||
t.Errorf("Image.Title has incorrect value. Got '%s', expected 'WriteTheWeb'.", img.Title)
|
||
return
|
||
}
|
||
}
|
||
|
||
func TestStringNamespaces(t *testing.T) {
|
||
doc := New()
|
||
err := doc.LoadFile("test3.xml", nil)
|
||
|
||
if err != nil {
|
||
t.Errorf("LoadFile(): %s", err)
|
||
return
|
||
}
|
||
|
||
expected := `<root xmlns:foo="http:/example.org/foo">
|
||
<child foo:bar="1">
|
||
<grandchild xmlns:foo="">
|
||
<great-grandchild bar="2">
 </great-grandchild>
|
||
</grandchild>
|
||
</child>
|
||
</root>
|
||
`
|
||
|
||
if got := doc.Root.String(); got != expected {
|
||
t.Fatalf("expected: %s\ngot: %s\n", expected, got)
|
||
}
|
||
}
|
||
|
||
func TestStringEscaping(t *testing.T) {
|
||
doc := New()
|
||
err := doc.LoadFile("test4.xml", nil)
|
||
|
||
if err != nil {
|
||
t.Errorf("LoadFile(): %s", err)
|
||
return
|
||
}
|
||
|
||
expected := `<body>  <https://example.com/file/fm/SU0vRk0xLzIwMTMwOTEwLzA1MDA0MS5ybXdhdGVzdEByZXV0ZXJzLmNvbTEzNzg4NDU1OTk4OTA/Screen%20Shot%202013-09-10%20at%2021.33.54.png> File Attachment:-Screen Shot 2013-09-10 at 21.33.54.png </body>
|
||
`
|
||
|
||
if got := doc.Root.String(); got != expected {
|
||
t.Fatalf("expected: %s\ngot: %s\n", expected, got)
|
||
}
|
||
}
|
||
|
||
func TestElementNodeValueFetch(t *testing.T) {
|
||
data := `<car><color>
|
||
r<cool />
|
||
ed</color><brand>BMW</brand><price>50
|
||
<cheap />.25</price><count>6
|
||
<small />2
|
||
</count><available>
|
||
Tr
|
||
<found />
|
||
ue</available></car>`
|
||
doc := New()
|
||
|
||
if err := doc.LoadString(data, nil); nil != err {
|
||
t.Fatalf("LoadString(): %s", err)
|
||
}
|
||
|
||
carN := doc.SelectNode("", "car")
|
||
if v := carN.S("", "brand"); v != "BMW" {
|
||
t.Errorf("Failed to get brand as string, got: '%s', wanted: 'BMW'", v)
|
||
}
|
||
if v := carN.S("", "color"); v != "red" {
|
||
t.Errorf("Failed to get color as string, got: '%s', wanted: 'red'", v)
|
||
}
|
||
|
||
if v := carN.I("", "count"); v != 62 {
|
||
t.Errorf("Failed to get count using I, got: %v, wanted: 62", v)
|
||
}
|
||
if v := carN.I8("", "count"); v != 62 {
|
||
t.Errorf("Failed to get count using I8, got: %v, wanted: 62", v)
|
||
}
|
||
if v := carN.I16("", "count"); v != 62 {
|
||
t.Errorf("Failed to get count using I16, got: %v, wanted: 62", v)
|
||
}
|
||
if v := carN.I32("", "count"); v != 62 {
|
||
t.Errorf("Failed to get count using I32, got: %v, wanted: 62", v)
|
||
}
|
||
if v := carN.I64("", "count"); v != 62 {
|
||
t.Errorf("Failed to get count using I64, got: %v, wanted: 62", v)
|
||
}
|
||
if v := carN.U("", "count"); v != 62 {
|
||
t.Errorf("Failed to get count using U, got: %v, wanted: 62", v)
|
||
}
|
||
if v := carN.U8("", "count"); v != 62 {
|
||
t.Errorf("Failed to get count using U8, got: %v, wanted: 62", v)
|
||
}
|
||
if v := carN.U16("", "count"); v != 62 {
|
||
t.Errorf("Failed to get count using U16, got: %v, wanted: 62", v)
|
||
}
|
||
if v := carN.U32("", "count"); v != 62 {
|
||
t.Errorf("Failed to get count using U32, got: %v, wanted: 62", v)
|
||
}
|
||
if v := carN.U64("", "count"); v != 62 {
|
||
t.Errorf("Failed to get count using U64, got: %v, wanted: 62", v)
|
||
}
|
||
|
||
if v := carN.F32("", "price"); v != 50.25 {
|
||
t.Errorf("Failed to get price using F32, got: %v, wanted: 50.25", v)
|
||
}
|
||
if v := carN.F64("", "price"); v != 50.25 {
|
||
t.Errorf("Failed to get price using F64, got: %v, wanted: 50.25", v)
|
||
}
|
||
|
||
if v := carN.B("", "available"); v != true {
|
||
t.Errorf("Failed to get availability using B, got: %v, wanted: true", v)
|
||
}
|
||
}
|
||
|
||
// node.SetValue(x); x == node.GetValue
|
||
func TestElementNodeValueFetchAndSetIdentity(t *testing.T) {
|
||
// Setup: <root><text>xyzzy</text></root>
|
||
// The xmlx parser creates a nameless NT_TEXT node containing the value 'xyzzy'
|
||
rootN := NewNode(NT_ROOT)
|
||
rootN.Name = xml.Name{Space: "", Local: "root"}
|
||
textN := NewNode(NT_ELEMENT)
|
||
textN.Name = xml.Name{Space: "", Local: "text"}
|
||
namelessN := NewNode(NT_TEXT)
|
||
namelessN.Value = "xyzzy"
|
||
rootN.AddChild(textN)
|
||
textN.AddChild(namelessN)
|
||
|
||
targetN := rootN.SelectNode("", "text") // selects textN
|
||
if targetN != textN {
|
||
t.Errorf("Failed to get the correct textN, got %#v", targetN)
|
||
}
|
||
|
||
// targetN.Value is empty (as the value lives in the childNode)
|
||
if targetN.Value != "" {
|
||
t.Errorf("Failed to prepare correctly, TargetN.Value is not empty, it contains %#v", targetN.Value)
|
||
}
|
||
|
||
// Test correct retrieval
|
||
if v := rootN.S("", "text"); v != "xyzzy" {
|
||
t.Errorf("Failed to get value as string, got: '%s', wanted: 'xyzzy'", v)
|
||
}
|
||
|
||
// Set the value of the nameless child
|
||
targetN.SetValue("plugh")
|
||
|
||
// Test correct retrieval
|
||
if v := rootN.S("", "text"); v != "plugh" {
|
||
t.Errorf("Failed to get value as string, got: '%s', wanted: 'plugh'", v)
|
||
}
|
||
}
|
||
|
||
// Test as it could be used to read in a XML file, change some values and write it out again.
|
||
// For example, a HTML/XML proxy service.
|
||
func TestElementNodeValueFetchAndSet(t *testing.T) {
|
||
IndentPrefix = ""
|
||
|
||
data := `<car><color>
|
||
r<cool />
|
||
ed</color><brand>BM<awesome />W</brand><price>50
|
||
<cheap />.25</price><count>6
|
||
<small />2
|
||
</count><available>
|
||
Tr
|
||
<found />
|
||
ue</available></car>`
|
||
doc := New()
|
||
|
||
if err := doc.LoadString(data, nil); nil != err {
|
||
t.Fatalf("LoadString(): %s", err)
|
||
}
|
||
|
||
carN := doc.SelectNode("", "car")
|
||
if carN == nil {
|
||
t.Fatalf("Failed to get the car, got nil, wanted Node{car}")
|
||
}
|
||
|
||
colorNode := carN.SelectNode("", "color")
|
||
if colorNode == nil {
|
||
t.Fatalf("Failed to get the color, got nil, wanted Node{color}")
|
||
}
|
||
|
||
colorVal := colorNode.GetValue()
|
||
if colorVal != "red" {
|
||
t.Fatalf("Failed to get the color, got %v, wanted 'red'", colorVal)
|
||
}
|
||
|
||
colorNode.SetValue("blue")
|
||
|
||
expected := `<car><color>blue</color><brand>BM<awesome />W</brand><price>50
|
||
<cheap />.25</price><count>6
|
||
<small />2
|
||
</count><available>
|
||
Tr
|
||
<found />
|
||
ue</available></car>`
|
||
|
||
if got := doc.Root.String(); got != expected {
|
||
t.Fatalf("expected: \n%s\ngot: \n%s\n", expected, got)
|
||
}
|
||
|
||
// now set the brand
|
||
brandNode := carN.SelectNode("", "brand")
|
||
if brandNode == nil {
|
||
t.Fatalf("Failed to get the color, got nil, wanted Node{brand}")
|
||
}
|
||
|
||
brandVal := brandNode.GetValue()
|
||
if brandVal != "BMW" {
|
||
t.Fatalf("Failed to get the brand, got %v, wanted 'BMW'", brandVal)
|
||
}
|
||
|
||
brandNode.SetValue("Trabant")
|
||
|
||
// Notice, we lose the <awesome /> tag in BMW, that's intentional
|
||
expected = `<car><color>blue</color><brand>Trabant</brand><price>50
|
||
<cheap />.25</price><count>6
|
||
<small />2
|
||
</count><available>
|
||
Tr
|
||
<found />
|
||
ue</available></car>`
|
||
|
||
if got := doc.Root.String(); got != expected {
|
||
t.Fatalf("expected: \n%s\ngot: \n%s\n", expected, got)
|
||
}
|
||
}
|
||
|
||
func TestSelectNodesDirect(t *testing.T) {
|
||
data := `<root><wrapper><hidden></hidden>
|
||
<hidden></hidden></wrapper></root>`
|
||
doc := New()
|
||
|
||
if err := doc.LoadString(data, nil); nil != err {
|
||
t.Fatalf("LoadString(): %s", err)
|
||
}
|
||
|
||
root := doc.SelectNode("*", "root")
|
||
if root == nil {
|
||
t.Fatalf("Failed to get root, got nil, wanted Node{root}")
|
||
}
|
||
|
||
nodes := root.SelectNodesDirect("*", "hidden")
|
||
|
||
if len(nodes) != 0 {
|
||
t.Errorf("SelectDirectNodes should not select children of children.")
|
||
}
|
||
|
||
wrapper := root.SelectNode("*", "wrapper")
|
||
if wrapper == nil {
|
||
t.Fatalf("Failed to get wrapper, got nil, wanted Node{wrapper}")
|
||
}
|
||
|
||
nodes = wrapper.SelectNodesDirect("*", "hidden")
|
||
if len(nodes) != 2 {
|
||
t.Errorf("Unexcepted hidden nodes found. Expected: 2, Got: %d", len(nodes))
|
||
}
|
||
}
|