package main
import (
"net/url"
"strings"
"testing"
"golang.org/x/net/html"
)
func parseDoc(t *testing.T, s string) *html.Node {
t.Helper()
doc, err := html.Parse(strings.NewReader(s))
if err != nil {
t.Fatal(err)
}
return doc
}
func TestFaviconExtractionPriorityAndFallback(t *testing.T) {
base, _ := url.Parse("https://example.com/path/page")
cases := []struct{ name, html, want string }{
{"icon", ``, "https://example.com/icon.png"},
{"shortcut", ``, "https://example.com/path/shortcut.ico"},
{"apple", ``, "https://cdn.test/apple.png"},
{"priority", ``, "https://example.com/icon.png"},
{"fallback", ``, "https://example.com/favicon.ico"},
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
if got := extractFavicon(parseDoc(t, tc.html), base); got != tc.want {
t.Fatalf("got %q want %q", got, tc.want)
}
})
}
}
func TestExtractLinkTags(t *testing.T) {
base, _ := url.Parse("https://example.com/dir/page")
doc := parseDoc(t, ``)
links := extractLinkTags(doc, base)
if len(links) != 2 {
t.Fatalf("links %#v", links)
}
if links[0].Rel != "preload" || links[0].Href != "https://example.com/style.css" || links[0].Extra != `as="style"` {
t.Fatalf("first link %#v", links[0])
}
if links[1].Href != "https://example.com/dir/canonical" {
t.Fatalf("relative href %q", links[1].Href)
}
if got := extractLinkTags(parseDoc(t, ``), base); got != nil {
t.Fatalf("expected nil without head, got %#v", got)
}
}