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/starturl/starturl.go | 9 +-------- modules/starturl/starturl_test.go | 2 +- 2 files changed, 2 insertions(+), 9 deletions(-) (limited to 'modules/starturl') diff --git a/modules/starturl/starturl.go b/modules/starturl/starturl.go index b2e6c47..109d28f 100644 --- a/modules/starturl/starturl.go +++ b/modules/starturl/starturl.go @@ -16,15 +16,8 @@ type Module struct { URL string `json:"url"` } -func (m *Module) ID() string { - return "starturl" -} - func (m *Module) OnLoad(v flyscrape.Visitor) { v.Visit(m.URL) } -var ( - _ flyscrape.Module = (*Module)(nil) - _ flyscrape.OnLoad = (*Module)(nil) -) +var _ flyscrape.OnLoad = (*Module)(nil) diff --git a/modules/starturl/starturl_test.go b/modules/starturl/starturl_test.go index 647e197..6fab776 100644 --- a/modules/starturl/starturl_test.go +++ b/modules/starturl/starturl_test.go @@ -12,7 +12,7 @@ import ( "github.com/stretchr/testify/require" ) -func TestFollowLinks(t *testing.T) { +func TestStartURL(t *testing.T) { scraper := flyscrape.NewScraper() scraper.LoadModule(&starturl.Module{URL: "http://www.example.com/foo/bar"}) scraper.SetTransport(flyscrape.MockTransport(200, "")) -- cgit v1.2.3