mirror of https://github.com/gofiber/fiber.git
🔥 Fiber Client poc
Kiyon
parent
d1d8edadf4
commit
51986b2e7c
|
|
@ -0,0 +1,569 @@
|
|||
package fiber
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/tls"
|
||||
"fmt"
|
||||
"io"
|
||||
"net"
|
||||
"os"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/gofiber/fiber/v2/internal/encoding/json"
|
||||
"github.com/valyala/fasthttp"
|
||||
)
|
||||
|
||||
// Request represents HTTP request.
|
||||
//
|
||||
// It is forbidden copying Request instances. Create new instances
|
||||
// and use CopyTo instead.
|
||||
//
|
||||
// Request instance MUST NOT be used from concurrently running goroutines.
|
||||
type Request = fasthttp.Request
|
||||
|
||||
// Response represents HTTP response.
|
||||
//
|
||||
// It is forbidden copying Response instances. Create new instances
|
||||
// and use CopyTo instead.
|
||||
//
|
||||
// Response instance MUST NOT be used from concurrently running goroutines.
|
||||
type Response = fasthttp.Response
|
||||
|
||||
// Args represents query arguments.
|
||||
//
|
||||
// It is forbidden copying Args instances. Create new instances instead
|
||||
// and use CopyTo().
|
||||
//
|
||||
// Args instance MUST NOT be used from concurrently running goroutines.
|
||||
type Args = fasthttp.Args
|
||||
|
||||
var defaultClient Client
|
||||
|
||||
// Client implements http client.
|
||||
//
|
||||
// It is safe calling Client methods from concurrently running goroutines.
|
||||
type Client struct {
|
||||
UserAgent string
|
||||
NoDefaultUserAgentHeader bool
|
||||
}
|
||||
|
||||
// Get returns a agent with http method GET.
|
||||
func Get(url string) *Agent { return defaultClient.Get(url) }
|
||||
|
||||
// Get returns a agent with http method GET.
|
||||
func (c *Client) Get(url string) *Agent {
|
||||
return c.createAgent(MethodGet, url)
|
||||
}
|
||||
|
||||
// Post sends POST request to the given url.
|
||||
func Post(url string) *Agent { return defaultClient.Post(url) }
|
||||
|
||||
// Post sends POST request to the given url.
|
||||
func (c *Client) Post(url string) *Agent {
|
||||
return c.createAgent(MethodPost, url)
|
||||
}
|
||||
|
||||
func (c *Client) createAgent(method, url string) *Agent {
|
||||
a := AcquireAgent()
|
||||
a.req.Header.SetMethod(method)
|
||||
a.req.SetRequestURI(url)
|
||||
|
||||
a.Name = c.UserAgent
|
||||
a.NoDefaultUserAgentHeader = c.NoDefaultUserAgentHeader
|
||||
|
||||
if err := a.Parse(); err != nil {
|
||||
a.errs = append(a.errs, err)
|
||||
}
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Agent is an object storing all request data for client.
|
||||
type Agent struct {
|
||||
*fasthttp.HostClient
|
||||
req *Request
|
||||
customReq *Request
|
||||
args *Args
|
||||
timeout time.Duration
|
||||
errs []error
|
||||
debugWriter io.Writer
|
||||
maxRedirectsCount int
|
||||
Name string
|
||||
NoDefaultUserAgentHeader bool
|
||||
reuse bool
|
||||
parsed bool
|
||||
}
|
||||
|
||||
var ErrorInvalidURI = fasthttp.ErrorInvalidURI
|
||||
|
||||
// Parse initializes URI and HostClient.
|
||||
func (a *Agent) Parse() error {
|
||||
if a.parsed {
|
||||
return nil
|
||||
}
|
||||
a.parsed = true
|
||||
|
||||
req := a.req
|
||||
if a.customReq != nil {
|
||||
req = a.customReq
|
||||
}
|
||||
|
||||
uri := req.URI()
|
||||
if uri == nil {
|
||||
return ErrorInvalidURI
|
||||
}
|
||||
|
||||
isTLS := false
|
||||
scheme := uri.Scheme()
|
||||
if bytes.Equal(scheme, strHTTPS) {
|
||||
isTLS = true
|
||||
} else if !bytes.Equal(scheme, strHTTP) {
|
||||
return fmt.Errorf("unsupported protocol %q. http and https are supported", scheme)
|
||||
}
|
||||
|
||||
name := a.Name
|
||||
if name == "" && !a.NoDefaultUserAgentHeader {
|
||||
name = defaultUserAgent
|
||||
}
|
||||
|
||||
a.HostClient = &fasthttp.HostClient{
|
||||
Addr: addMissingPort(string(uri.Host()), isTLS),
|
||||
Name: name,
|
||||
NoDefaultUserAgentHeader: a.NoDefaultUserAgentHeader,
|
||||
IsTLS: isTLS,
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func addMissingPort(addr string, isTLS bool) string {
|
||||
n := strings.Index(addr, ":")
|
||||
if n >= 0 {
|
||||
return addr
|
||||
}
|
||||
port := 80
|
||||
if isTLS {
|
||||
port = 443
|
||||
}
|
||||
return net.JoinHostPort(addr, strconv.Itoa(port))
|
||||
}
|
||||
|
||||
// Set sets the given 'key: value' header.
|
||||
//
|
||||
// Use Add for setting multiple header values under the same key.
|
||||
func (a *Agent) Set(k, v string) *Agent {
|
||||
a.req.Header.Set(k, v)
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Add adds the given 'key: value' header.
|
||||
//
|
||||
// Multiple headers with the same key may be added with this function.
|
||||
// Use Set for setting a single header for the given key.
|
||||
func (a *Agent) Add(k, v string) *Agent {
|
||||
a.req.Header.Add(k, v)
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Host sets host for the uri.
|
||||
func (a *Agent) Host(host string) *Agent {
|
||||
a.req.URI().SetHost(host)
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// ConnectionClose sets 'Connection: close' header.
|
||||
func (a *Agent) ConnectionClose() *Agent {
|
||||
a.req.Header.SetConnectionClose()
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// UserAgent sets User-Agent header value.
|
||||
func (a *Agent) UserAgent(userAgent string) *Agent {
|
||||
a.req.Header.SetUserAgent(userAgent)
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Debug mode enables logging request and response detail
|
||||
func (a *Agent) Debug(w ...io.Writer) *Agent {
|
||||
a.debugWriter = os.Stdout
|
||||
if len(w) > 0 {
|
||||
a.debugWriter = w[0]
|
||||
}
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Cookie sets one 'key: value' cookie.
|
||||
func (a *Agent) Cookie(key, value string) *Agent {
|
||||
a.req.Header.SetCookie(key, value)
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Cookies sets multiple 'key: value' cookies.
|
||||
func (a *Agent) Cookies(kv ...string) *Agent {
|
||||
for i := 1; i < len(kv); i += 2 {
|
||||
a.req.Header.SetCookie(kv[i-1], kv[i])
|
||||
}
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Timeout sets request timeout duration.
|
||||
func (a *Agent) Timeout(timeout time.Duration) *Agent {
|
||||
a.timeout = timeout
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Json sends a json request.
|
||||
func (a *Agent) Json(v interface{}) *Agent {
|
||||
a.req.Header.SetContentType(MIMEApplicationJSON)
|
||||
|
||||
if body, err := json.Marshal(v); err != nil {
|
||||
a.errs = append(a.errs, err)
|
||||
} else {
|
||||
a.req.SetBody(body)
|
||||
}
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Form sends request with body if args is non-nil.
|
||||
//
|
||||
// Note that this will force http method to post.
|
||||
func (a *Agent) Form(args *Args) *Agent {
|
||||
a.req.Header.SetContentType(MIMEApplicationForm)
|
||||
|
||||
if args != nil {
|
||||
if _, err := args.WriteTo(a.req.BodyWriter()); err != nil {
|
||||
a.errs = append(a.errs, err)
|
||||
}
|
||||
}
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// QueryString sets URI query string.
|
||||
func (a *Agent) QueryString(queryString string) *Agent {
|
||||
a.req.URI().SetQueryString(queryString)
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// BodyStream sets request body stream and, optionally body size.
|
||||
//
|
||||
// If bodySize is >= 0, then the bodyStream must provide exactly bodySize bytes
|
||||
// before returning io.EOF.
|
||||
//
|
||||
// If bodySize < 0, then bodyStream is read until io.EOF.
|
||||
//
|
||||
// bodyStream.Close() is called after finishing reading all body data
|
||||
// if it implements io.Closer.
|
||||
//
|
||||
// Note that GET and HEAD requests cannot have body.
|
||||
func (a *Agent) BodyStream(bodyStream io.Reader, bodySize int) *Agent {
|
||||
a.req.SetBodyStream(bodyStream, bodySize)
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Reuse indicates the createAgent can be used again after one request.
|
||||
func (a *Agent) Reuse() *Agent {
|
||||
a.reuse = true
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// InsecureSkipVerify controls whether the createAgent verifies the server's
|
||||
// certificate chain and host name.
|
||||
func (a *Agent) InsecureSkipVerify() *Agent {
|
||||
if a.HostClient.TLSConfig == nil {
|
||||
a.HostClient.TLSConfig = &tls.Config{InsecureSkipVerify: true}
|
||||
} else {
|
||||
a.HostClient.TLSConfig.InsecureSkipVerify = true
|
||||
}
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// TLSConfig sets tls config.
|
||||
func (a *Agent) TLSConfig(config *tls.Config) *Agent {
|
||||
a.HostClient.TLSConfig = config
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Request sets custom request for createAgent.
|
||||
func (a *Agent) Request(req *Request) *Agent {
|
||||
a.customReq = req
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Referer sets Referer header value.
|
||||
func (a *Agent) Referer(referer string) *Agent {
|
||||
a.req.Header.SetReferer(referer)
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// ContentType sets Content-Type header value.
|
||||
func (a *Agent) ContentType(contentType string) *Agent {
|
||||
a.req.Header.SetContentType(contentType)
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// MaxRedirectsCount sets max redirect count for GET and HEAD.
|
||||
func (a *Agent) MaxRedirectsCount(count int) *Agent {
|
||||
a.maxRedirectsCount = count
|
||||
|
||||
return a
|
||||
}
|
||||
|
||||
// Bytes returns the status code, bytes body and errors of url.
|
||||
func (a *Agent) Bytes(customResp ...*Response) (code int, body []byte, errs []error) {
|
||||
defer a.release()
|
||||
|
||||
if errs = append(errs, a.errs...); len(errs) > 0 {
|
||||
return
|
||||
}
|
||||
|
||||
req := a.req
|
||||
if a.customReq != nil {
|
||||
req = a.customReq
|
||||
}
|
||||
|
||||
var (
|
||||
resp *Response
|
||||
releaseResp bool
|
||||
)
|
||||
if len(customResp) > 0 {
|
||||
resp = customResp[0]
|
||||
} else {
|
||||
resp = AcquireResponse()
|
||||
releaseResp = true
|
||||
}
|
||||
defer func() {
|
||||
if a.debugWriter != nil {
|
||||
printDebugInfo(req, resp, a.debugWriter)
|
||||
}
|
||||
|
||||
if len(errs) == 0 {
|
||||
code = resp.StatusCode()
|
||||
}
|
||||
|
||||
if releaseResp {
|
||||
body = append(body, resp.Body()...)
|
||||
ReleaseResponse(resp)
|
||||
} else {
|
||||
body = resp.Body()
|
||||
}
|
||||
}()
|
||||
|
||||
if a.timeout > 0 {
|
||||
if err := a.HostClient.DoTimeout(req, resp, a.timeout); err != nil {
|
||||
errs = append(errs, err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
if a.maxRedirectsCount > 0 && (string(req.Header.Method()) == MethodGet || string(req.Header.Method()) == MethodHead) {
|
||||
if err := a.HostClient.DoRedirects(req, resp, a.maxRedirectsCount); err != nil {
|
||||
errs = append(errs, err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
if err := a.HostClient.Do(req, resp); err != nil {
|
||||
errs = append(errs, err)
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func printDebugInfo(req *Request, resp *Response, w io.Writer) {
|
||||
msg := fmt.Sprintf("Connected to %s(%s)\r\n\r\n", req.URI().Host(), resp.RemoteAddr())
|
||||
_, _ = w.Write(getBytes(msg))
|
||||
_, _ = req.WriteTo(w)
|
||||
_, _ = resp.WriteTo(w)
|
||||
}
|
||||
|
||||
// String returns the status code, string body and errors of url.
|
||||
func (a *Agent) String(resp ...*Response) (int, string, []error) {
|
||||
code, body, errs := a.Bytes(resp...)
|
||||
|
||||
return code, getString(body), errs
|
||||
}
|
||||
|
||||
// Struct returns the status code, bytes body and errors of url.
|
||||
// And bytes body will be unmarshalled to given v.
|
||||
func (a *Agent) Struct(v interface{}, resp ...*Response) (code int, body []byte, errs []error) {
|
||||
code, body, errs = a.Bytes(resp...)
|
||||
|
||||
if err := json.Unmarshal(body, v); err != nil {
|
||||
errs = append(errs, err)
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (a *Agent) release() {
|
||||
if !a.reuse {
|
||||
ReleaseAgent(a)
|
||||
} else {
|
||||
a.errs = a.errs[:0]
|
||||
}
|
||||
}
|
||||
|
||||
func (a *Agent) reset() {
|
||||
a.HostClient = nil
|
||||
a.req.Reset()
|
||||
a.customReq = nil
|
||||
a.timeout = 0
|
||||
a.args = nil
|
||||
a.errs = a.errs[:0]
|
||||
a.debugWriter = nil
|
||||
a.reuse = false
|
||||
a.parsed = false
|
||||
a.maxRedirectsCount = 0
|
||||
a.Name = ""
|
||||
a.NoDefaultUserAgentHeader = false
|
||||
}
|
||||
|
||||
var (
|
||||
clientPool sync.Pool
|
||||
agentPool sync.Pool
|
||||
requestPool sync.Pool
|
||||
responsePool sync.Pool
|
||||
argsPool sync.Pool
|
||||
)
|
||||
|
||||
// AcquireAgent returns an empty Agent instance from createAgent pool.
|
||||
//
|
||||
// The returned Agent instance may be passed to ReleaseAgent when it is
|
||||
// no longer needed. This allows Agent recycling, reduces GC pressure
|
||||
// and usually improves performance.
|
||||
func AcquireAgent() *Agent {
|
||||
v := agentPool.Get()
|
||||
if v == nil {
|
||||
return &Agent{req: fasthttp.AcquireRequest()}
|
||||
}
|
||||
return v.(*Agent)
|
||||
}
|
||||
|
||||
// ReleaseAgent returns a acquired via AcquireAgent to createAgent pool.
|
||||
//
|
||||
// It is forbidden accessing req and/or its' members after returning
|
||||
// it to createAgent pool.
|
||||
func ReleaseAgent(a *Agent) {
|
||||
a.reset()
|
||||
agentPool.Put(a)
|
||||
}
|
||||
|
||||
// AcquireClient returns an empty Client instance from client pool.
|
||||
//
|
||||
// The returned Client instance may be passed to ReleaseClient when it is
|
||||
// no longer needed. This allows Client recycling, reduces GC pressure
|
||||
// and usually improves performance.
|
||||
func AcquireClient() *Client {
|
||||
v := clientPool.Get()
|
||||
if v == nil {
|
||||
return &Client{}
|
||||
}
|
||||
return v.(*Client)
|
||||
}
|
||||
|
||||
// ReleaseClient returns c acquired via AcquireClient to client pool.
|
||||
//
|
||||
// It is forbidden accessing req and/or its' members after returning
|
||||
// it to client pool.
|
||||
func ReleaseClient(c *Client) {
|
||||
c.UserAgent = ""
|
||||
c.NoDefaultUserAgentHeader = false
|
||||
|
||||
clientPool.Put(c)
|
||||
}
|
||||
|
||||
// AcquireRequest returns an empty Request instance from request pool.
|
||||
//
|
||||
// The returned Request instance may be passed to ReleaseRequest when it is
|
||||
// no longer needed. This allows Request recycling, reduces GC pressure
|
||||
// and usually improves performance.
|
||||
func AcquireRequest() *Request {
|
||||
v := requestPool.Get()
|
||||
if v == nil {
|
||||
return &Request{}
|
||||
}
|
||||
return v.(*Request)
|
||||
}
|
||||
|
||||
// ReleaseRequest returns req acquired via AcquireRequest to request pool.
|
||||
//
|
||||
// It is forbidden accessing req and/or its' members after returning
|
||||
// it to request pool.
|
||||
func ReleaseRequest(req *Request) {
|
||||
req.Reset()
|
||||
requestPool.Put(req)
|
||||
}
|
||||
|
||||
// AcquireResponse returns an empty Response instance from response pool.
|
||||
//
|
||||
// The returned Response instance may be passed to ReleaseResponse when it is
|
||||
// no longer needed. This allows Response recycling, reduces GC pressure
|
||||
// and usually improves performance.
|
||||
// Copy from fasthttp
|
||||
func AcquireResponse() *Response {
|
||||
v := responsePool.Get()
|
||||
if v == nil {
|
||||
return &Response{}
|
||||
}
|
||||
return v.(*Response)
|
||||
}
|
||||
|
||||
// ReleaseResponse return resp acquired via AcquireResponse to response pool.
|
||||
//
|
||||
// It is forbidden accessing resp and/or its' members after returning
|
||||
// it to response pool.
|
||||
// Copy from fasthttp
|
||||
func ReleaseResponse(resp *Response) {
|
||||
resp.Reset()
|
||||
responsePool.Put(resp)
|
||||
}
|
||||
|
||||
// AcquireArgs returns an empty Args object from the pool.
|
||||
//
|
||||
// The returned Args may be returned to the pool with ReleaseArgs
|
||||
// when no longer needed. This allows reducing GC load.
|
||||
// Copy from fasthttp
|
||||
func AcquireArgs() *Args {
|
||||
v := argsPool.Get()
|
||||
if v == nil {
|
||||
return &Args{}
|
||||
}
|
||||
return v.(*Args)
|
||||
}
|
||||
|
||||
// ReleaseArgs returns the object acquired via AcquireArgs to the pool.
|
||||
//
|
||||
// String not access the released Args object, otherwise data races may occur.
|
||||
// Copy from fasthttp
|
||||
func ReleaseArgs(a *Args) {
|
||||
a.Reset()
|
||||
argsPool.Put(a)
|
||||
}
|
||||
|
||||
var (
|
||||
strHTTP = []byte("http")
|
||||
strHTTPS = []byte("https")
|
||||
defaultUserAgent = "fiber"
|
||||
)
|
||||
|
|
@ -0,0 +1,636 @@
|
|||
package fiber
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/tls"
|
||||
"net"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/gofiber/fiber/v2/utils"
|
||||
"github.com/valyala/fasthttp/fasthttputil"
|
||||
)
|
||||
|
||||
func Test_Client_Invalid_URL(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", func(c *Ctx) error {
|
||||
return c.SendString(c.Hostname())
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
a := Get("http://example.com\r\n\r\nGET /\r\n\r\n")
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
_, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, "", body)
|
||||
utils.AssertEqual(t, 1, len(errs))
|
||||
utils.AssertEqual(t, "missing required Host header in request", errs[0].Error())
|
||||
}
|
||||
|
||||
func Test_Client_Unsupported_Protocol(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
a := Get("ftp://example.com")
|
||||
|
||||
_, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, "", body)
|
||||
utils.AssertEqual(t, 1, len(errs))
|
||||
utils.AssertEqual(t, `unsupported protocol "ftp". http and https are supported`,
|
||||
errs[0].Error())
|
||||
}
|
||||
|
||||
func Test_Client_Get(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", func(c *Ctx) error {
|
||||
return c.SendString(c.Hostname())
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
for i := 0; i < 5; i++ {
|
||||
a := Get("http://example.com")
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
code, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, "example.com", body)
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
}
|
||||
}
|
||||
|
||||
func Test_Client_Post(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Post("/", func(c *Ctx) error {
|
||||
return c.SendString(c.Hostname())
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
for i := 0; i < 5; i++ {
|
||||
args := AcquireArgs()
|
||||
|
||||
args.Set("foo", "bar")
|
||||
|
||||
a := Post("http://example.com").
|
||||
Form(args)
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
code, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, "example.com", body)
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
|
||||
ReleaseArgs(args)
|
||||
}
|
||||
}
|
||||
|
||||
func Test_Client_UserAgent(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", func(c *Ctx) error {
|
||||
return c.Send(c.Request().Header.UserAgent())
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
t.Run("default", func(t *testing.T) {
|
||||
for i := 0; i < 5; i++ {
|
||||
a := Get("http://example.com")
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
code, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, defaultUserAgent, body)
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("custom", func(t *testing.T) {
|
||||
for i := 0; i < 5; i++ {
|
||||
c := AcquireClient()
|
||||
c.UserAgent = "ua"
|
||||
|
||||
a := c.Get("http://example.com")
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
code, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, "ua", body)
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
ReleaseClient(c)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Specific_Host(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", func(c *Ctx) error {
|
||||
return c.SendString(c.Hostname())
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
a := Get("http://1.1.1.1:8080").
|
||||
Host("example.com")
|
||||
|
||||
utils.AssertEqual(t, "1.1.1.1:8080", a.HostClient.Addr)
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
code, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, "example.com", body)
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Headers(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
c.Request().Header.VisitAll(func(key, value []byte) {
|
||||
if k := string(key); k == "K1" || k == "K2" {
|
||||
_, _ = c.Write(key)
|
||||
_, _ = c.Write(value)
|
||||
}
|
||||
})
|
||||
return nil
|
||||
}
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.Set("k1", "v1").
|
||||
Add("k1", "v11").
|
||||
Set("k2", "v2")
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, "K1v1K1v11K2v2")
|
||||
}
|
||||
|
||||
func Test_Client_Agent_UserAgent(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
return c.Send(c.Request().Header.UserAgent())
|
||||
}
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.UserAgent("ua")
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, "ua")
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Connection_Close(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
if c.Request().Header.ConnectionClose() {
|
||||
return c.SendString("close")
|
||||
}
|
||||
return c.SendString("not close")
|
||||
}
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.ConnectionClose()
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, "close")
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Referer(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
return c.Send(c.Request().Header.Referer())
|
||||
}
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.Referer("http://referer.com")
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, "http://referer.com")
|
||||
}
|
||||
|
||||
func Test_Client_Agent_QueryString(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
return c.Send(c.Request().URI().QueryString())
|
||||
}
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.QueryString("foo=bar&bar=baz")
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, "foo=bar&bar=baz")
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Cookie(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
return c.SendString(
|
||||
c.Cookies("k1") + c.Cookies("k2") + c.Cookies("k3") + c.Cookies("k4"))
|
||||
}
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.Cookie("k1", "v1").
|
||||
Cookie("k2", "v2").
|
||||
Cookies("k3", "v3", "k4", "v4")
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, "v1v2v3v4")
|
||||
}
|
||||
|
||||
func Test_Client_Agent_ContentType(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
return c.Send(c.Request().Header.ContentType())
|
||||
}
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.ContentType("custom-type")
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, "custom-type")
|
||||
}
|
||||
|
||||
func Test_Client_Agent_BodyStream(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
return c.Send(c.Request().Body())
|
||||
}
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.BodyStream(strings.NewReader("body stream"), -1)
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, "body stream")
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Form(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
utils.AssertEqual(t, MIMEApplicationForm, string(c.Request().Header.ContentType()))
|
||||
|
||||
return c.Send(c.Request().Body())
|
||||
}
|
||||
|
||||
args := AcquireArgs()
|
||||
|
||||
args.Set("a", "b")
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.Form(args)
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, "a=b")
|
||||
|
||||
ReleaseArgs(args)
|
||||
}
|
||||
|
||||
type jsonData struct {
|
||||
F string `json:"f"`
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Json(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
utils.AssertEqual(t, MIMEApplicationJSON, string(c.Request().Header.ContentType()))
|
||||
|
||||
return c.Send(c.Request().Body())
|
||||
}
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.Json(jsonData{F: "f"})
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, `{"f":"f"}`)
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Json_Error(t *testing.T) {
|
||||
a := Get("http://example.com").
|
||||
Json(complex(1, 1))
|
||||
|
||||
_, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, "", body)
|
||||
utils.AssertEqual(t, 1, len(errs))
|
||||
utils.AssertEqual(t, "json: unsupported type: complex128", errs[0].Error())
|
||||
}
|
||||
|
||||
func Test_Client_Debug(t *testing.T) {
|
||||
handler := func(c *Ctx) error {
|
||||
return c.SendString("debug")
|
||||
}
|
||||
|
||||
var output bytes.Buffer
|
||||
|
||||
wrapAgent := func(a *Agent) {
|
||||
a.Debug(&output)
|
||||
}
|
||||
|
||||
testAgent(t, handler, wrapAgent, "debug", 1)
|
||||
|
||||
str := output.String()
|
||||
|
||||
utils.AssertEqual(t, true, strings.Contains(str, "Connected to example.com(pipe)"))
|
||||
utils.AssertEqual(t, true, strings.Contains(str, "GET / HTTP/1.1"))
|
||||
utils.AssertEqual(t, true, strings.Contains(str, "User-Agent: fiber"))
|
||||
utils.AssertEqual(t, true, strings.Contains(str, "Host: example.com\r\n\r\n"))
|
||||
utils.AssertEqual(t, true, strings.Contains(str, "HTTP/1.1 200 OK"))
|
||||
utils.AssertEqual(t, true, strings.Contains(str, "Content-Type: text/plain; charset=utf-8\r\nContent-Length: 5\r\n\r\ndebug"))
|
||||
}
|
||||
|
||||
func testAgent(t *testing.T, handler Handler, wrapAgent func(agent *Agent), excepted string, count ...int) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", handler)
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
c := 1
|
||||
if len(count) > 0 {
|
||||
c = count[0]
|
||||
}
|
||||
|
||||
for i := 0; i < c; i++ {
|
||||
a := Get("http://example.com")
|
||||
|
||||
wrapAgent(a)
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
code, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, excepted, body)
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
}
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Timeout(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", func(c *Ctx) error {
|
||||
time.Sleep(time.Millisecond * 200)
|
||||
return c.SendString("timeout")
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
a := Get("http://example.com").
|
||||
Timeout(time.Millisecond * 100)
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
_, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, "", body)
|
||||
utils.AssertEqual(t, 1, len(errs))
|
||||
utils.AssertEqual(t, "timeout", errs[0].Error())
|
||||
}
|
||||
|
||||
func Test_Client_Agent_MaxRedirectsCount(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", func(c *Ctx) error {
|
||||
if c.Request().URI().QueryArgs().Has("foo") {
|
||||
return c.Redirect("/foo")
|
||||
}
|
||||
return c.Redirect("/")
|
||||
})
|
||||
app.Get("/foo", func(c *Ctx) error {
|
||||
return c.SendString("redirect")
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
t.Run("success", func(t *testing.T) {
|
||||
a := Get("http://example.com?foo").
|
||||
MaxRedirectsCount(1)
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
code, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, 200, code)
|
||||
utils.AssertEqual(t, "redirect", body)
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
})
|
||||
|
||||
t.Run("error", func(t *testing.T) {
|
||||
a := Get("http://example.com").
|
||||
MaxRedirectsCount(1)
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
_, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, "", body)
|
||||
utils.AssertEqual(t, 1, len(errs))
|
||||
utils.AssertEqual(t, "too many redirects detected when doing the request", errs[0].Error())
|
||||
})
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Custom(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", func(c *Ctx) error {
|
||||
return c.SendString("custom")
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
for i := 0; i < 5; i++ {
|
||||
a := AcquireAgent()
|
||||
req := AcquireRequest()
|
||||
resp := AcquireResponse()
|
||||
|
||||
req.Header.SetMethod(MethodGet)
|
||||
req.SetRequestURI("http://example.com")
|
||||
a.Request(req)
|
||||
|
||||
utils.AssertEqual(t, nil, a.Parse())
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
code, body, errs := a.String(resp)
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, "custom", body)
|
||||
utils.AssertEqual(t, "custom", string(resp.Body()))
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
|
||||
ReleaseRequest(req)
|
||||
ReleaseResponse(resp)
|
||||
}
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Reuse(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", func(c *Ctx) error {
|
||||
return c.SendString("reuse")
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
a := Get("http://example.com").
|
||||
Reuse()
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
code, body, errs := a.String()
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, "reuse", body)
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
|
||||
code, body, errs = a.String()
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, "reuse", body)
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Parse(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
a := Get("https://example.com:10443")
|
||||
|
||||
utils.AssertEqual(t, nil, a.Parse())
|
||||
}
|
||||
|
||||
func Test_Client_Agent_TLS(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
// Create tls certificate
|
||||
cer, err := tls.LoadX509KeyPair("./.github/testdata/ssl.pem", "./.github/testdata/ssl.key")
|
||||
utils.AssertEqual(t, nil, err)
|
||||
|
||||
config := &tls.Config{
|
||||
Certificates: []tls.Certificate{cer},
|
||||
}
|
||||
|
||||
ln, err := net.Listen(NetworkTCP4, "127.0.0.1:0")
|
||||
utils.AssertEqual(t, nil, err)
|
||||
|
||||
ln = tls.NewListener(ln, config)
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", func(c *Ctx) error {
|
||||
return c.SendString("tls")
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
code, body, errs := Get("https://" + ln.Addr().String()).
|
||||
InsecureSkipVerify().
|
||||
TLSConfig(config).
|
||||
InsecureSkipVerify().
|
||||
String()
|
||||
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, "tls", body)
|
||||
}
|
||||
|
||||
type data struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func Test_Client_Agent_Struct(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
ln := fasthttputil.NewInmemoryListener()
|
||||
|
||||
app := New(Config{DisableStartupMessage: true})
|
||||
|
||||
app.Get("/", func(c *Ctx) error {
|
||||
return c.JSON(data{true})
|
||||
})
|
||||
|
||||
app.Get("/error", func(c *Ctx) error {
|
||||
return c.SendString(`{"success"`)
|
||||
})
|
||||
|
||||
go app.Listener(ln) //nolint:errcheck
|
||||
|
||||
t.Run("success", func(t *testing.T) {
|
||||
a := Get("http://example.com")
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
var d data
|
||||
|
||||
code, body, errs := a.Struct(&d)
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, `{"success":true}`, string(body))
|
||||
utils.AssertEqual(t, 0, len(errs))
|
||||
utils.AssertEqual(t, true, d.Success)
|
||||
})
|
||||
|
||||
t.Run("error", func(t *testing.T) {
|
||||
a := Get("http://example.com/error")
|
||||
|
||||
a.HostClient.Dial = func(addr string) (net.Conn, error) { return ln.Dial() }
|
||||
|
||||
var d data
|
||||
|
||||
code, body, errs := a.Struct(&d)
|
||||
|
||||
utils.AssertEqual(t, StatusOK, code)
|
||||
utils.AssertEqual(t, `{"success"`, string(body))
|
||||
utils.AssertEqual(t, 1, len(errs))
|
||||
utils.AssertEqual(t, "json: unexpected end of JSON input after object field key: ", errs[0].Error())
|
||||
})
|
||||
}
|
||||
|
||||
func Test_AddMissingPort_TLS(t *testing.T) {
|
||||
addr := addMissingPort("example.com", true)
|
||||
utils.AssertEqual(t, "example.com:443", addr)
|
||||
}
|
||||
Loading…
Reference in New Issue