Element.Create returns the created element

Author: beevik

etree.go

@@ -773,15 +773,17 @@ type CreateContinuation func(e *Element)
 // This method of creating elements is useful when building nested XML
 // document from code. For example:
 //
-//	doc.Create("organization", func(e *Element) {
+//	org := doc.Create("organization", func(e *Element) {
 //		e.Create("person", func(e *Element) {
 //			e.CreateAttr("name", "Mary")
 //			e.CreateAttr("age", "30")
 //			e.CreateAttr("hair", "brown")
 //		})
 //	})
-func (e *Element) Create(tag string, f CreateContinuation) {
-	f(e.CreateElement(tag))
+func (e *Element) Create(tag string, f CreateContinuation) *Element {
+	child := e.CreateElement(tag)
+	f(child)
+	return child
 }
 
 // AddChild adds the token 't' as the last child of the element. If token 't'

etree_test.go

@@ -1632,7 +1632,7 @@ func TestSiblingElement(t *testing.T) {
 
 func TestContinuations(t *testing.T) {
 	doc := NewDocument()
-	doc.Create("root", func(e *Element) {
+	root := doc.Create("root", func(e *Element) {
 		e.Create("child1", func(e *Element) {
 			e.Create("grandchild1", func(e *Element) {
 				e.CreateAttr("attr1", "1")
@@ -1654,9 +1654,10 @@ func TestContinuations(t *testing.T) {
 			})
 		})
 	})
-	doc.IndentTabs()
+	checkStrEq(t, root.Tag, "root")
 
 	// Serialize the document to a string
+	doc.IndentTabs()
 	s, err := doc.WriteToString()
 	if err != nil {
 		t.Error("etree: failed to serialize document")