mirror of https://github.com/gofiber/fiber.git
80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
package binder
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
func Test_RespHeaderBinder_Bind(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
b := &RespHeaderBinding{
|
|
EnableSplitting: true,
|
|
}
|
|
require.Equal(t, "respHeader", b.Name())
|
|
|
|
type User struct {
|
|
Name string `respHeader:"name"`
|
|
Posts []string `respHeader:"posts"`
|
|
Age int `respHeader:"age"`
|
|
}
|
|
var user User
|
|
|
|
resp := fasthttp.AcquireResponse()
|
|
resp.Header.Set("name", "john")
|
|
resp.Header.Set("age", "42")
|
|
resp.Header.Set("posts", "post1,post2,post3")
|
|
|
|
t.Cleanup(func() {
|
|
fasthttp.ReleaseResponse(resp)
|
|
})
|
|
|
|
err := b.Bind(resp, &user)
|
|
|
|
require.NoError(t, err)
|
|
require.Equal(t, "john", user.Name)
|
|
require.Equal(t, 42, user.Age)
|
|
require.Equal(t, []string{"post1", "post2", "post3"}, user.Posts)
|
|
|
|
b.Reset()
|
|
require.False(t, b.EnableSplitting)
|
|
}
|
|
|
|
func Benchmark_RespHeaderBinder_Bind(b *testing.B) {
|
|
b.ReportAllocs()
|
|
|
|
binder := &RespHeaderBinding{
|
|
EnableSplitting: true,
|
|
}
|
|
|
|
type User struct {
|
|
Name string `respHeader:"name"`
|
|
Posts []string `respHeader:"posts"`
|
|
Age int `respHeader:"age"`
|
|
}
|
|
var user User
|
|
|
|
resp := fasthttp.AcquireResponse()
|
|
resp.Header.Set("name", "john")
|
|
resp.Header.Set("age", "42")
|
|
resp.Header.Set("posts", "post1,post2,post3")
|
|
|
|
b.Cleanup(func() {
|
|
fasthttp.ReleaseResponse(resp)
|
|
})
|
|
|
|
b.ResetTimer()
|
|
|
|
var err error
|
|
for i := 0; i < b.N; i++ {
|
|
err = binder.Bind(resp, &user)
|
|
}
|
|
|
|
require.NoError(b, err)
|
|
require.Equal(b, "john", user.Name)
|
|
require.Equal(b, 42, user.Age)
|
|
require.Equal(b, []string{"post1", "post2", "post3"}, user.Posts)
|
|
}
|