mirror of https://github.com/gofiber/fiber.git
91 lines
1.7 KiB
Go
91 lines
1.7 KiB
Go
package binder
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
func Test_CookieBinder_Bind(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
b := &CookieBinding{
|
|
EnableSplitting: true,
|
|
}
|
|
require.Equal(t, "cookie", b.Name())
|
|
|
|
type Post struct {
|
|
Title string `form:"title"`
|
|
}
|
|
|
|
type User struct {
|
|
Name string `form:"name"`
|
|
Names []string `form:"names"`
|
|
Posts []Post `form:"posts"`
|
|
Age int `form:"age"`
|
|
}
|
|
var user User
|
|
|
|
req := fasthttp.AcquireRequest()
|
|
|
|
req.Header.SetCookie("name", "john")
|
|
req.Header.SetCookie("names", "john,doe")
|
|
req.Header.SetCookie("age", "42")
|
|
|
|
t.Cleanup(func() {
|
|
fasthttp.ReleaseRequest(req)
|
|
})
|
|
|
|
err := b.Bind(req, &user)
|
|
|
|
require.NoError(t, err)
|
|
require.Equal(t, "john", user.Name)
|
|
require.Equal(t, 42, user.Age)
|
|
require.Contains(t, user.Names, "john")
|
|
require.Contains(t, user.Names, "doe")
|
|
|
|
b.Reset()
|
|
require.False(t, b.EnableSplitting)
|
|
}
|
|
|
|
func Benchmark_CookieBinder_Bind(b *testing.B) {
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
|
|
binder := &CookieBinding{
|
|
EnableSplitting: true,
|
|
}
|
|
|
|
type User struct {
|
|
Name string `query:"name"`
|
|
Posts []string `query:"posts"`
|
|
Age int `query:"age"`
|
|
}
|
|
var user User
|
|
|
|
req := fasthttp.AcquireRequest()
|
|
b.Cleanup(func() {
|
|
fasthttp.ReleaseRequest(req)
|
|
})
|
|
|
|
req.Header.SetCookie("name", "john")
|
|
req.Header.SetCookie("age", "42")
|
|
req.Header.SetCookie("posts", "post1,post2,post3")
|
|
|
|
b.ResetTimer()
|
|
|
|
var err error
|
|
for i := 0; i < b.N; i++ {
|
|
err = binder.Bind(req, &user)
|
|
}
|
|
|
|
require.NoError(b, err)
|
|
require.Equal(b, "john", user.Name)
|
|
require.Equal(b, 42, user.Age)
|
|
require.Len(b, user.Posts, 3)
|
|
require.Contains(b, user.Posts, "post1")
|
|
require.Contains(b, user.Posts, "post2")
|
|
require.Contains(b, user.Posts, "post3")
|
|
}
|