diff options
Diffstat (limited to 'modules/ratelimit')
| -rw-r--r-- | modules/ratelimit/ratelimit_test.go | 27 |
1 files changed, 15 insertions, 12 deletions
diff --git a/modules/ratelimit/ratelimit_test.go b/modules/ratelimit/ratelimit_test.go index 1fe22b1..7be29a1 100644 --- a/modules/ratelimit/ratelimit_test.go +++ b/modules/ratelimit/ratelimit_test.go @@ -20,22 +20,25 @@ import ( func TestRatelimit(t *testing.T) { var times []time.Time - scraper := flyscrape.NewScraper() - scraper.LoadModule(&starturl.Module{URL: "http://www.example.com"}) - scraper.LoadModule(&followlinks.Module{}) - scraper.LoadModule(hook.Module{ - AdaptTransportFn: func(rt http.RoundTripper) http.RoundTripper { - return flyscrape.MockTransport(200, `<a href="foo">foo</a>`) + mods := []flyscrape.Module{ + &starturl.Module{URL: "http://www.example.com"}, + &followlinks.Module{}, + hook.Module{ + AdaptTransportFn: func(rt http.RoundTripper) http.RoundTripper { + return flyscrape.MockTransport(200, `<a href="foo">foo</a>`) + }, + ReceiveResponseFn: func(r *flyscrape.Response) { + times = append(times, time.Now()) + }, }, - ReceiveResponseFn: func(r *flyscrape.Response) { - times = append(times, time.Now()) + &ratelimit.Module{ + Rate: 100, }, - }) - scraper.LoadModule(&ratelimit.Module{ - Rate: 100, - }) + } start := time.Now() + scraper := flyscrape.NewScraper() + scraper.Modules = mods scraper.Run() first := times[0].Add(-10 * time.Millisecond) |