Add TestXpath

Create sample.html
Jisin0 · Jun 1, 2024 · 7e75ab4 · 7e75ab4
1 parent 9c64a35
commit 7e75ab4
Show file tree

Hide file tree

Showing 2 changed files with 62 additions and 0 deletions.
diff --git a/encode/sample.html b/encode/sample.html
@@ -0,0 +1,25 @@
+<html>
+    <body>
+        <main>
+            <ul class="sample">
+                <li>Node 1</li>
+                <li>Node 2</li>
+                <li>Node 3</li>
+            </ul>
+
+            <span class="sample">
+                <a href="example.com">Data 1</a>
+                <a href="example.com">Data 2</a>
+                <span class="subnode">
+                    <a href="example.com">Data 3</a>
+                </span>
+            </span>
+
+            <p class="find-substring-here">
+                Lorem ipsum dolor sit amet, consectetur adipiscing elit. Donec non tortor in orci sagittis vehicula accumsan vel magna. Fusce et nulla nec eros egestas ullamcorper. Phasellus consequat euismod efficitur. Fusce nec bibendum diam, eu mollis turpis. Sed nisi tellus, euismod ut viverra vitae, efficitur et magna. Proin dictum tellus ac augue semper, quis pharetra mauris ullamcorper. Duis quis ullamcorper metus. Morbi eu leo et augue aliquam faucibus.
+            </p>
+
+            <p my-attr="sample"></p>
+        </main>
+    </body>
+</html>
diff --git a/encode/xpath_test.go b/encode/xpath_test.go
@@ -0,0 +1,37 @@
+package encode_test
+
+import (
+	"testing"
+
+	"github.com/Jisin0/filmigo/encode"
+	"github.com/Jisin0/filmigo/types"
+	"github.com/antchfx/htmlquery"
+)
+
+func TestXpath(t *testing.T) {
+	doc, err := htmlquery.LoadDoc("sample.html")
+	if err != nil {
+		t.Errorf("failed to open sample file : %v", err)
+		t.FailNow()
+	}
+
+	if doc != nil {
+		type sampleData struct {
+			InnerText  string      `xpath:"//p[contains(@class, 'substring')]"`
+			Attribute  string      `xpath:"//p[last()]|attr_my-attr"`
+			LinkList   types.Links `xpath:"//span[@class='sample']"`
+			StringList []string    `xpath:"//ul"`
+		}
+
+		res, ok := encode.Xpath(doc, sampleData{}).(sampleData)
+		if !ok {
+			t.Errorf("unknown type returned")
+			t.FailNow()
+		}
+
+		if res.Attribute == "" || res.InnerText == "" || len(res.LinkList) < 3 || len(res.StringList) < 3 {
+			t.Errorf("xpath failed with output : %+v", res)
+			t.FailNow()
+		}
+	}
+}