diff options
Diffstat (limited to 'modules/starturl/starturl_test.go')
| -rw-r--r-- | modules/starturl/starturl_test.go | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/starturl/starturl_test.go b/modules/starturl/starturl_test.go index 54f899a..4b6a554 100644 --- a/modules/starturl/starturl_test.go +++ b/modules/starturl/starturl_test.go @@ -6,6 +6,7 @@ package starturl_test import ( "net/http" + "sync" "testing" "github.com/philippta/flyscrape" @@ -100,6 +101,7 @@ func TestStartURL_MultipleStartingURLs(t *testing.T) { for _, tc := range testCases { t.Run(tc.name, func(t *testing.T) { urls := []string{} + mu := sync.Mutex{} mods := []flyscrape.Module{ tc.startURLModFn(), @@ -108,7 +110,9 @@ func TestStartURL_MultipleStartingURLs(t *testing.T) { return flyscrape.MockTransport(http.StatusOK, "") }, BuildRequestFn: func(r *flyscrape.Request) { + mu.Lock() urls = append(urls, r.URL) + mu.Unlock() }, }, } |