diff options
Diffstat (limited to 'modules/starturl')
| -rw-r--r-- | modules/starturl/starturl_test.go | 23 |
1 files changed, 13 insertions, 10 deletions
diff --git a/modules/starturl/starturl_test.go b/modules/starturl/starturl_test.go index 86e4ad7..78efa6a 100644 --- a/modules/starturl/starturl_test.go +++ b/modules/starturl/starturl_test.go @@ -18,18 +18,21 @@ func TestStartURL(t *testing.T) { var url string var depth int - scraper := flyscrape.NewScraper() - scraper.LoadModule(&starturl.Module{URL: "http://www.example.com/foo/bar"}) - scraper.LoadModule(hook.Module{ - AdaptTransportFn: func(rt http.RoundTripper) http.RoundTripper { - return flyscrape.MockTransport(200, "") - }, - BuildRequestFn: func(r *flyscrape.Request) { - url = r.URL - depth = r.Depth + mods := []flyscrape.Module{ + &starturl.Module{URL: "http://www.example.com/foo/bar"}, + hook.Module{ + AdaptTransportFn: func(rt http.RoundTripper) http.RoundTripper { + return flyscrape.MockTransport(200, "") + }, + BuildRequestFn: func(r *flyscrape.Request) { + url = r.URL + depth = r.Depth + }, }, - }) + } + scraper := flyscrape.NewScraper() + scraper.Modules = mods scraper.Run() require.Equal(t, "http://www.example.com/foo/bar", url) |