Update continuation unit test

beevik · Jan 21, 2025 · f6baac9 · f6baac9
1 parent fd97f04
commit f6baac9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/etree_test.go b/etree_test.go
@@ -1634,6 +1634,7 @@ func TestContinuations(t *testing.T) {
 	doc := NewDocument()
 	root := doc.CreateElementContinue("root", func(e *Element) {
 		e.CreateElementContinue("child1", func(e *Element) {
+			e.CreateComment("Grandchildren of child #1")
 			e.CreateElementContinue("grandchild1", func(e *Element) {
 				e.CreateAttr("attr1", "1")
 				e.CreateAttr("attr2", "2")
@@ -1644,6 +1645,7 @@ func TestContinuations(t *testing.T) {
 			})
 		})
 		e.CreateElementContinue("child2", func(e *Element) {
+			e.CreateComment("Grandchildren of child #2")
 			e.CreateElementContinue("grandchild1", func(e *Element) {
 				e.CreateAttr("attr1", "5")
 				e.CreateAttr("attr2", "6")
@@ -1666,10 +1668,12 @@ func TestContinuations(t *testing.T) {
 	// Make sure the serialized XML matches expectation.
 	expected := `<root>
 	<child1>
+		<!--Grandchildren of child #1-->
 		<grandchild1 attr1="1" attr2="2"/>
 		<grandchild2 attr1="3" attr2="4"/>
 	</child1>
 	<child2>
+		<!--Grandchildren of child #2-->
 		<grandchild1 attr1="5" attr2="6"/>
 		<grandchild2 attr1="7" attr2="8"/>
 	</child2>