69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
|
package paginate
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestPaginationFromQuery(t *testing.T) {
|
||
|
|
req, err := http.NewRequest("GET", "/?page=2&pageSize=10", http.NoBody)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
pagination := PaginationFromQuery("page", "pageSize")(req, &Pagination{})
|
||
|
|
assert.Equal(t, &Pagination{Page: 2, PageSize: 10}, pagination)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestValidPagination(t *testing.T) {
|
||
|
|
opts := []Option{
|
||
|
|
SetPaginationDefaults(1, 10),
|
||
|
|
}
|
||
|
|
|
||
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
pagination := PaginationFromContext(r.Context())
|
||
|
|
assert.NotNil(t, pagination)
|
||
|
|
assert.Equal(t, 2, pagination.Page)
|
||
|
|
assert.Equal(t, 10, pagination.PageSize)
|
||
|
|
})
|
||
|
|
|
||
|
|
middleware := Middleware(opts...)(handler)
|
||
|
|
|
||
|
|
w := httptest.NewRecorder()
|
||
|
|
req := httptest.NewRequest("GET", "/?page=2&pageSize=10", http.NoBody)
|
||
|
|
middleware.ServeHTTP(w, req)
|
||
|
|
|
||
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestInvalidPagination(t *testing.T) {
|
||
|
|
opts := []Option{
|
||
|
|
SetPaginationDefaults(1, 10),
|
||
|
|
SetValidatePaginationFn(func(p *Pagination) error {
|
||
|
|
if p.Page < 1 || p.PageSize < 1 {
|
||
|
|
return errors.New("invalid pagination")
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}),
|
||
|
|
}
|
||
|
|
|
||
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
pagination := PaginationFromContext(r.Context())
|
||
|
|
assert.NotNil(t, pagination)
|
||
|
|
assert.Equal(t, -1, pagination.Page)
|
||
|
|
assert.Equal(t, 10, pagination.PageSize)
|
||
|
|
})
|
||
|
|
|
||
|
|
middleware := Middleware(opts...)(handler)
|
||
|
|
|
||
|
|
w := httptest.NewRecorder()
|
||
|
|
req := httptest.NewRequest("GET", "/?page=-1&pageSize=10", http.NoBody)
|
||
|
|
middleware.ServeHTTP(w, req)
|
||
|
|
|
||
|
|
assert.Equal(t, http.StatusBadRequest, w.Code)
|
||
|
|
}
|