Refactor to prepare for builtin JS functions

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)
 }