package main import ( "fmt" "strings" "testing" ) const page0 = `
link0
` func TestLinksOnPage(t *testing.T) { links, err := linksOnPage( strings.NewReader(page0), func(s string) (string, error) { return s, nil }, ) if err != nil { t.Fatal(err) } if l := len(links); l != 4 { t.Fatalf("Expected 4 links, go %d\n", l) } for i, link := range links { href := fmt.Sprintf("link%d", i) if href != link { t.Fatalf("Expected link '%s', got '%s'\n", href, link) } } }