diff options
Diffstat (limited to 'js_lib_test.go')
| -rw-r--r-- | js_lib_test.go | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/js_lib_test.go b/js_lib_test.go index aca8ce9..ad19380 100644 --- a/js_lib_test.go +++ b/js_lib_test.go @@ -8,6 +8,7 @@ import ( "encoding/json" "net/http" "os" + "sync/atomic" "testing" "github.com/philippta/flyscrape" @@ -203,10 +204,10 @@ func TestJSLibHTTPDownload(t *testing.T) { http.download("https://example.com/404.txt"); ` - nreqs := 0 + var nreqs atomic.Int32 client := &http.Client{ Transport: flyscrape.RoundTripFunc(func(r *http.Request) (*http.Response, error) { - nreqs++ + nreqs.Add(1) if r.URL.Path == "/content-disposition" { resp, err := flyscrape.MockResponse(200, "hello world") @@ -233,7 +234,7 @@ func TestJSLibHTTPDownload(t *testing.T) { wait() - require.Equal(t, nreqs, 8) + require.Equal(t, nreqs.Load(), int32(8)) require.FileExists(t, "foo.txt") require.FileExists(t, "dir/my-foo.txt") require.FileExists(t, "dir/bar.txt") |