broodjeaap/go-watch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added test for XPath filter

Browse source
This commit is contained in:
BroodjeAap 2022-08-11 18:28:11 +00:00
parent ab3f02778f
commit ab7a7fd534
1 changed files with 52 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
scraping_test.go
View file

@ -2,9 +2,61 @@ package main
import (
"fmt"
"reflect"
"testing"
)
const HTML_STRING = `<html>
<head>
<title>title</title>
</head>
<body>
<table class="product-table" id="product-table">
<caption>product-table-caption</caption>
<thead>
<tr>
<th>Name</th>
<th>Stock</th>
<th>Price</th>
</tr>
</thead>
<tbody>
<tr><td>product1</td><td>10</td><td>100</td></tr>
<tr><td>product2</td><td>20</td><td>200</td></tr>
<tr><td>product3</td><td>30</td><td>300</td></tr>
<tr><td>product4</td><td>40</td><td>400</td></tr>
</tbody>
</table>
</body>
</html>`
func TestFilterXPath(t *testing.T) {
var tests = []struct {
Query string
Want []string
}{
{"//title", []string{"<title>title</title>"}},
{"//table[@id='product-table']//tr//td[last()]", []string{"<td>100</td>", "<td>200</td>", "<td>300</td>", "<td>400</td>"}},
}
for _, test := range tests {
testname := fmt.Sprintf("%s", test.Query)
t.Run(testname, func(t *testing.T) {
want := []string{}
getFilterResultXPath(
HTML_STRING,
&Filter{
From: test.Query,
},
&want,
)
if !reflect.DeepEqual(test.Want, want) {
t.Errorf("Got %s, want %s", want, test.Want)
}
})
}
}
func TestFilterSubstring(t *testing.T) {
var tests = []struct {
Input string