Import source: parser/html

Import the existing local project snapshot. This commit organizes existing files; it does not represent newly developed functionality.
2026-09-18 23:18:59 +03:30
parent ca61b2fcba
commit aa9dcd639c
2 changed files with 110 additions and 0 deletions
@@ -0,0 +1,62 @@
+//go:build unit
+// +build unit
+
+package parser_test
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"reactor-crw/parser"
+)
+
+func TestHtml_FindOne(t *testing.T) {
+	p := parser.Html{}
+
+	r := strings.NewReader(`
+		<body>
+			<h1>heading</h1>
+			<a class="test-class" href="href-test" target="_blank">First Link</a>
+			<a class="class" href="href-test" target="_blank">Second Link</a>
+			<a href="href-test" target="_blank">Third Link</a>
+		</body>
+	`)
+
+	res, err := p.FindContent(r, "body .test-class")
+	require.NoErrorf(t, err, "Wasn't expected an error on html parse")
+
+	require.Equal(t, "First Link", res)
+}
+
+func TestHtml_FindAttrMap(t *testing.T) {
+	p := parser.Html{}
+
+	r := strings.NewReader(`
+		<body>
+			<img src="image-src">
+			<a class="test-class" href="href-test" target="_blank">First Link</a>
+			<div class="div-class"></div>
+		</body>
+	`)
+
+	queryAttrMap := parser.QueryAttrMap{
+		".test-class": "href",
+		"div":         "class",
+		"img":         "src",
+		"body":        "src",
+	}
+	res := make(parser.QueryResult)
+
+	err := p.FindAttrMap(r, queryAttrMap, res)
+	require.NoErrorf(t, err, "Wasn't expected an error on html parse")
+	require.Equal(
+		t,
+		parser.QueryResult{
+			"image-src": struct{}{},
+			"href-test": struct{}{},
+			"div-class": struct{}{},
+		},
+		res,
+	)
+}