diff options
Diffstat (limited to 'modules/proxy/proxy_test.go')
| -rw-r--r-- | modules/proxy/proxy_test.go | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/modules/proxy/proxy_test.go b/modules/proxy/proxy_test.go index e6058b8..62da23a 100644 --- a/modules/proxy/proxy_test.go +++ b/modules/proxy/proxy_test.go @@ -20,13 +20,17 @@ func TestProxy(t *testing.T) { p := newProxy(func() { called = true }) defer p.Close() - scraper := flyscrape.NewScraper() - scraper.LoadModule(&starturl.Module{URL: "http://www.example.com"}) - scraper.LoadModule(&proxy.Module{ - Proxies: []string{p.URL}, - }) + mods := []flyscrape.Module{ + &starturl.Module{URL: "http://www.example.com"}, + &proxy.Module{ + Proxies: []string{p.URL}, + }, + } + scraper := flyscrape.NewScraper() + scraper.Modules = mods scraper.Run() + require.True(t, called) } |