gf/third/github.com/clbanning/mxj/misc_test.go

// misc_test.go

package mxj

import (
	"fmt"
	"testing"
)

var miscdata = []byte(`
<doc>
	<elem1 name="elem1" seq="1">
		<sub1 name="sub1" seq="1">sub_value_1</sub1>
		<sub2 name="sub2" seq="2">sub_value_2</sub2>
	</elem1>
	<elem2 name="elem2" seq="2">element_2</elem2>
</doc>
`)

func TestMisc(t *testing.T) {
	PrependAttrWithHyphen(true) // be safe
	fmt.Println("\n------------------ misc_test.go ...")
}

func TestRoot(t *testing.T) {
	m, err := NewMapXml(miscdata)
	if err != nil {
		t.Fatal(err)
	}
	r, err := m.Root()
	if err != nil {
		t.Fatal(err)
	}
	if r != "doc" {
		t.Fatal("Root not doc:", r)
	}
}

func TestElements(t *testing.T) {
	m, err := NewMapXml(miscdata)
	if err != nil {
		t.Fatal(err)
	}
	e, err := m.Elements("doc")
	if err != nil {
		t.Fatal(err)
	}
	elist := []string{"elem1", "elem2"}
	for i, ee := range e {
		if ee != elist[i] {
			t.Fatal("error in list, elem#:", i, "-", ee, ":", elist[i])
		}
	}

	e, err = m.Elements("doc.elem1")
	if err != nil {
		t.Fatal(err)
	}
	elist = []string{"sub1", "sub2"}
	for i, ee := range e {
		if ee != elist[i] {
			t.Fatal("error in list, elem#:", i, "-", ee, ":", elist[i])
		}
	}
}

func TestAttributes(t *testing.T) {
	m, err := NewMapXml(miscdata)
	if err != nil {
		t.Fatal(err)
	}
	a, err := m.Attributes("doc.elem2")
	if err != nil {
		t.Fatal(err)
	}
	alist := []string{"name", "seq"}
	for i, aa := range a {
		if aa != alist[i] {
			t.Fatal("error in list, elem#:", i, "-", aa, ":", alist[i])
		}
	}

	a, err = m.Attributes("doc.elem1.sub2")
	if err != nil {
		t.Fatal(err)
	}
	alist = []string{"name", "seq"}
	for i, aa := range a {
		if aa != alist[i] {
			t.Fatal("error in list, elem#:", i, "-", aa, ":", alist[i])
		}
	}
}

func TestElementsAttrPrefix(t *testing.T) {
	SetAttrPrefix("__")
	m, err := NewMapXml(miscdata)
	if err != nil {
		t.Fatal(err)
	}
	e, err := m.Elements("doc")
	if err != nil {
		t.Fatal(err)
	}
	elist := []string{"elem1", "elem2"}
	for i, ee := range e {
		if ee != elist[i] {
			t.Fatal("error in list, elem#:", i, "-", ee, ":", elist[i])
		}
	}

	e, err = m.Elements("doc.elem1")
	if err != nil {
		t.Fatal(err)
	}
	elist = []string{"sub1", "sub2"}
	for i, ee := range e {
		if ee != elist[i] {
			t.Fatal("error in list, elem#:", i, "-", ee, ":", elist[i])
		}
	}
}

func TestAttributesAttrPrefix(t *testing.T) {
	SetAttrPrefix("__")
	m, err := NewMapXml(miscdata)
	if err != nil {
		t.Fatal(err)
	}
	a, err := m.Attributes("doc.elem2")
	if err != nil {
		t.Fatal(err)
	}
	alist := []string{"name", "seq"}
	for i, aa := range a {
		if aa != alist[i] {
			t.Fatal("error in list, elem#:", i, "-", aa, ":", alist[i])
		}
	}

	a, err = m.Attributes("doc.elem1.sub2")
	if err != nil {
		t.Fatal(err)
	}
	alist = []string{"name", "seq"}
	for i, aa := range a {
		if aa != alist[i] {
			t.Fatal("error in list, elem#:", i, "-", aa, ":", alist[i])
		}
	}
}

func TestElementsNoAttrPrefix(t *testing.T) {
	PrependAttrWithHyphen(false)
	m, err := NewMapXml(miscdata)
	if err != nil {
		t.Fatal(err)
	}
	e, err := m.Elements("doc")
	if err != nil {
		t.Fatal(err)
	}
	if len(e) != 2 {
		t.Fatal("didn't get 2 elements:", e)
	}

	e, err = m.Elements("doc.elem1")
	if err != nil {
		t.Fatal(err)
	}
	if len(e) != 4 {
		t.Fatal("didn't get 4 elements:", e)
	}
	PrependAttrWithHyphen(true)
}

func TestAttributesNoAttrPrefix(t *testing.T) {
	PrependAttrWithHyphen(false)
	m, err := NewMapXml(miscdata)
	if err != nil {
		t.Fatal(err)
	}
	a, err := m.Attributes("doc.elem2")
	if err != nil {
		t.Fatal(err)
	}
	if len(a) > 0 {
		t.Fatal("found attributes where there are none:", a)
	}

	a, err = m.Attributes("doc.elem1.sub2")
	if err != nil {
		t.Fatal(err)
	}
	if len(a) > 0 {
		t.Fatal("found attributes where there are none:", a)
	}
	PrependAttrWithHyphen(true)
}