package main import ( "fmt" "strings" "testing" ) const page0 = ` Not a JSON link0
  1. link1
  2. link1

  • link1

    • ` func TestLinksOnPage(t *testing.T) { var links []string err := linksOnPage( strings.NewReader(page0), func(s string) error { if strings.HasSuffix(s, ".json") { links = append(links, s) } return 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.json", i) if href != link { t.Fatalf("Expected link '%s', got '%s'\n", href, link) } } }