mirror of
https://github.com/gospider007/requests.git
synced 2025-12-24 13:57:52 +08:00
31 lines
642 B
Go
31 lines
642 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/gospider007/requests"
|
|
)
|
|
|
|
func TestRequestCallBack(t *testing.T) {
|
|
_, err := requests.Get(nil, "https://httpbin.org/anything", requests.RequestOption{
|
|
RequestCallBack: func(ctx context.Context, request *http.Request, response *http.Response) error {
|
|
if response != nil {
|
|
if response.ContentLength > 100 {
|
|
return errors.New("max length")
|
|
}
|
|
}
|
|
return nil
|
|
},
|
|
})
|
|
if err == nil {
|
|
t.Error("err is nil")
|
|
}
|
|
if !strings.Contains(err.Error(), "max length") {
|
|
t.Error("err is not max length")
|
|
}
|
|
}
|