Added Node.SetValue(string) method.
SetValue replaces all children of the current node with a single NT_TEXT node and sets its Value to the given parameter. This lets us replace node values before writing out the document.
This commit is contained in:
parent
745ca85455
commit
17e1b69620
7
node.go
7
node.go
|
@ -66,6 +66,13 @@ func (this *Node) GetValue() string {
|
|||
return res
|
||||
}
|
||||
|
||||
func (this *Node) SetValue(val string) {
|
||||
t := NewNode(NT_TEXT)
|
||||
t.Value = string([]byte(val))
|
||||
t.Parent = this
|
||||
this.Children = []*Node{t} // brutally replace all other children
|
||||
}
|
||||
|
||||
// Get node value as string
|
||||
func (this *Node) S(namespace, name string) string {
|
||||
foundNode := rec_SelectNode(this, namespace, name)
|
||||
|
|
75
xmlx_test.go
75
xmlx_test.go
|
@ -251,3 +251,78 @@ func TestElementNodeValueFetch(t *testing.T) {
|
|||
t.Errorf("Failed to get availability using B, got: %v, wanted: true", v)
|
||||
}
|
||||
}
|
||||
|
||||
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)
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue