From bd9e7f7acfd855d4685aa4544169c0e29cdbf205 Mon Sep 17 00:00:00 2001 From: Philipp Tanlak Date: Sun, 24 Sep 2023 23:36:00 +0200 Subject: clean up modules --- modules/depth/depth_test.go | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'modules/depth/depth_test.go') diff --git a/modules/depth/depth_test.go b/modules/depth/depth_test.go index 309e628..c9afd6f 100644 --- a/modules/depth/depth_test.go +++ b/modules/depth/depth_test.go @@ -17,17 +17,17 @@ import ( func TestDepth(t *testing.T) { scraper := flyscrape.NewScraper() - scraper.LoadModule(&starturl.Module{URL: "http://www.example.com/"}) + scraper.LoadModule(&starturl.Module{URL: "http://www.example.com"}) scraper.LoadModule(&followlinks.Module{}) scraper.LoadModule(&depth.Module{Depth: 2}) scraper.SetTransport(func(r *http.Request) (*http.Response, error) { switch r.URL.String() { - case "http://www.example.com/": + case "http://www.example.com": return flyscrape.MockResponse(200, `Google`) - case "http://www.google.com/": + case "http://www.google.com": return flyscrape.MockResponse(200, `DuckDuckGo`) - case "http://www.duckduckgo.com/": + case "http://www.duckduckgo.com": return flyscrape.MockResponse(200, `Example`) } return flyscrape.MockResponse(200, "") @@ -41,7 +41,7 @@ func TestDepth(t *testing.T) { scraper.Run() require.Len(t, urls, 3) - require.Contains(t, urls, "http://www.example.com/") - require.Contains(t, urls, "http://www.google.com/") - require.Contains(t, urls, "http://www.duckduckgo.com/") + require.Contains(t, urls, "http://www.example.com") + require.Contains(t, urls, "http://www.google.com") + require.Contains(t, urls, "http://www.duckduckgo.com") } -- cgit v1.2.3