@@ -8,32 +8,35 @@ import (
88 "testing"
99 "time"
1010
11- "github.com/stretchr/testify/assert "
11+ "github.com/stretchr/testify/require "
1212)
1313
1414func TestErrRetryAfterIs (t * testing.T ) {
15+ t .Parallel ()
1516 var err error
1617
1718 err = NewErrorRetryAfter (errors .New ("test" ), 10 * time .Second )
18- assert .True (t , errors .Is (err , & ErrorRetryAfter {}), "pointer to error must be error" )
19+ require .True (t , errors .Is (err , & ErrorRetryAfter {}), "pointer to error must be error" )
1920
2021 err = fmt .Errorf ("wrapped: %w" , err )
21- assert .True (t , errors .Is (err , & ErrorRetryAfter {}), "wrapped pointer to error must be error" )
22+ require .True (t , errors .Is (err , & ErrorRetryAfter {}), "wrapped pointer to error must be error" )
2223}
2324
2425func TestErrRetryAfterAs (t * testing.T ) {
26+ t .Parallel ()
27+
2528 var (
2629 err error
2730 errRA * ErrorRetryAfter
2831 )
2932
3033 err = NewErrorRetryAfter (errors .New ("test" ), 25 * time .Second )
31- assert .True (t , errors .As (err , & errRA ), "pointer to error must be error" )
32- assert .EqualValues (t , errRA .RetryAfter , 25 * time .Second )
34+ require .True (t , errors .As (err , & errRA ), "pointer to error must be error" )
35+ require .EqualValues (t , errRA .RetryAfter , 25 * time .Second )
3336
3437 err = fmt .Errorf ("wrapped: %w" , err )
35- assert .True (t , errors .As (err , & errRA ), "wrapped pointer to error must be error" )
36- assert .EqualValues (t , errRA .RetryAfter , 25 * time .Second )
38+ require .True (t , errors .As (err , & errRA ), "wrapped pointer to error must be error" )
39+ require .EqualValues (t , errRA .RetryAfter , 25 * time .Second )
3740}
3841
3942func TestWebError (t * testing.T ) {
@@ -43,37 +46,45 @@ func TestWebError(t *testing.T) {
4346 config := & Config {Headers : map [string ][]string {}}
4447
4548 t .Run ("429 Too Many Requests" , func (t * testing.T ) {
49+ t .Parallel ()
50+
4651 err := fmt .Errorf ("wrapped for testing: %w" , NewErrorRetryAfter (ErrTooManyRequests , 0 ))
4752 w := httptest .NewRecorder ()
4853 r := httptest .NewRequest (http .MethodGet , "/blah" , nil )
4954 webError (w , r , config , err , http .StatusInternalServerError )
50- assert .Equal (t , http .StatusTooManyRequests , w .Result ().StatusCode )
51- assert .Zero (t , len (w .Result ().Header .Values ("Retry-After" )))
55+ require .Equal (t , http .StatusTooManyRequests , w .Result ().StatusCode )
56+ require .Zero (t , len (w .Result ().Header .Values ("Retry-After" )))
5257 })
5358
5459 t .Run ("429 Too Many Requests with Retry-After header" , func (t * testing.T ) {
60+ t .Parallel ()
61+
5562 err := NewErrorRetryAfter (ErrTooManyRequests , 25 * time .Second )
5663 w := httptest .NewRecorder ()
5764 r := httptest .NewRequest (http .MethodGet , "/blah" , nil )
5865 webError (w , r , config , err , http .StatusInternalServerError )
59- assert .Equal (t , http .StatusTooManyRequests , w .Result ().StatusCode )
60- assert .Equal (t , "25" , w .Result ().Header .Get ("Retry-After" ))
66+ require .Equal (t , http .StatusTooManyRequests , w .Result ().StatusCode )
67+ require .Equal (t , "25" , w .Result ().Header .Get ("Retry-After" ))
6168 })
6269
6370 t .Run ("503 Service Unavailable with Retry-After header" , func (t * testing.T ) {
71+ t .Parallel ()
72+
6473 err := NewErrorRetryAfter (ErrServiceUnavailable , 50 * time .Second )
6574 w := httptest .NewRecorder ()
6675 r := httptest .NewRequest (http .MethodGet , "/blah" , nil )
6776 webError (w , r , config , err , http .StatusInternalServerError )
68- assert .Equal (t , http .StatusServiceUnavailable , w .Result ().StatusCode )
69- assert .Equal (t , "50" , w .Result ().Header .Get ("Retry-After" ))
77+ require .Equal (t , http .StatusServiceUnavailable , w .Result ().StatusCode )
78+ require .Equal (t , "50" , w .Result ().Header .Get ("Retry-After" ))
7079 })
7180
7281 t .Run ("ErrorStatusCode propagates HTTP Status Code" , func (t * testing.T ) {
82+ t .Parallel ()
83+
7384 err := NewErrorStatusCodeFromStatus (http .StatusTeapot )
7485 w := httptest .NewRecorder ()
7586 r := httptest .NewRequest (http .MethodGet , "/blah" , nil )
7687 webError (w , r , config , err , http .StatusInternalServerError )
77- assert .Equal (t , http .StatusTeapot , w .Result ().StatusCode )
88+ require .Equal (t , http .StatusTeapot , w .Result ().StatusCode )
7889 })
7990}
0 commit comments