fix ContainsNode test

JohannesKaufmann · Jun 30, 2024 · af7a9be · af7a9be
1 parent de4174d
commit af7a9be
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/find_test.go b/find_test.go
@@ -8,7 +8,15 @@ import (
 )
 
 func TestContainsNode(t *testing.T) {
-	node := parse(t, `<root><start><span><target></target></span><span><target></target></span></start></root><next></next>`, "root")
+	input := `<root><start><span><target></target></span><span><target></target></span></start></root><next></next>`
+	doc, err := html.Parse(strings.NewReader(input))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	node := FindFirstNode(doc, func(node *html.Node) bool {
+		return node.Data == "root"
+	})
 
 	var called1 int
 	res1 := ContainsNode(node.FirstChild, func(n *html.Node) bool {