68 lines
1.1 KiB
Go
68 lines
1.1 KiB
Go
package scraper
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"golang.org/x/net/html"
|
|
)
|
|
|
|
const htmlStr = `<div><span id="test">Hello</span><span id="test">World</span></div>`
|
|
|
|
func TestSearchElem(t *testing.T) {
|
|
doc := Must(
|
|
html.Parse(strings.NewReader(htmlStr)),
|
|
)
|
|
|
|
ch := SearchElem(doc, "span")
|
|
count := 0
|
|
for range ch {
|
|
count++
|
|
}
|
|
|
|
if count != 2 {
|
|
t.Errorf("Expected: 2 span elements, got: %d", count)
|
|
}
|
|
}
|
|
|
|
func TestSearchAttr(t *testing.T) {
|
|
doc := Must(
|
|
html.Parse(strings.NewReader(htmlStr)),
|
|
)
|
|
|
|
ch := SearchAttr(doc, "id", "test")
|
|
count := 0
|
|
for range ch {
|
|
count++
|
|
}
|
|
|
|
if count != 2 {
|
|
t.Errorf("Expected: 2 span elements with id 'test', got: %d", count)
|
|
}
|
|
}
|
|
|
|
func TestSearchElemAttr(t *testing.T) {
|
|
doc := Must(
|
|
html.Parse(strings.NewReader(htmlStr)),
|
|
)
|
|
|
|
ch := SearchElemAttr(doc, "span", "id", "test")
|
|
count := 0
|
|
for range ch {
|
|
count++
|
|
}
|
|
|
|
if count != 2 {
|
|
t.Errorf("Expected: 2 span elements with id 'test', got: %d", count)
|
|
}
|
|
}
|
|
|
|
func TestCrawlTest(t *testing.T) {
|
|
fmt.Println(
|
|
crawlText(Must(
|
|
html.Parse(strings.NewReader(htmlStr)),
|
|
)),
|
|
)
|
|
}
|