mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 19:40:37 +00:00
104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
package pagination_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/api/pagination"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
)
|
|
|
|
func TestStartAndEndPage(t *testing.T) {
|
|
tests := []struct {
|
|
token string
|
|
pageSize int
|
|
totalSize int
|
|
nextToken string
|
|
start int
|
|
end int
|
|
}{
|
|
{
|
|
token: "0",
|
|
pageSize: 9,
|
|
totalSize: 100,
|
|
nextToken: "1",
|
|
start: 0,
|
|
end: 9,
|
|
},
|
|
{
|
|
token: "10",
|
|
pageSize: 4,
|
|
totalSize: 100,
|
|
nextToken: "11",
|
|
start: 40,
|
|
end: 44,
|
|
},
|
|
{
|
|
token: "100",
|
|
pageSize: 5,
|
|
totalSize: 1000,
|
|
nextToken: "101",
|
|
start: 500,
|
|
end: 505,
|
|
},
|
|
{
|
|
token: "3",
|
|
pageSize: 33,
|
|
totalSize: 100,
|
|
nextToken: "",
|
|
start: 99,
|
|
end: 100,
|
|
},
|
|
{
|
|
token: "34",
|
|
pageSize: 500,
|
|
totalSize: 17500,
|
|
nextToken: "",
|
|
start: 17000,
|
|
end: 17500,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
start, end, next, err := pagination.StartAndEndPage(test.token, test.pageSize, test.totalSize)
|
|
require.NoError(t, err)
|
|
if test.start != start {
|
|
t.Errorf("expected start and computed start are not equal %d, %d", test.start, start)
|
|
}
|
|
if test.end != end {
|
|
t.Errorf("expected end and computed end are not equal %d, %d", test.end, end)
|
|
}
|
|
if test.nextToken != next {
|
|
t.Errorf("expected next token and computed next token are not equal %v, %v", test.nextToken, next)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestStartAndEndPage_CannotConvertPage(t *testing.T) {
|
|
wanted := "could not convert page token: strconv.Atoi: parsing"
|
|
_, _, _, err := pagination.StartAndEndPage("bad", 0, 0)
|
|
assert.ErrorContains(t, wanted, err)
|
|
}
|
|
|
|
func TestStartAndEndPage_ExceedsMaxPage(t *testing.T) {
|
|
wanted := "page start 0 >= list 0"
|
|
_, _, _, err := pagination.StartAndEndPage("", 0, 0)
|
|
assert.ErrorContains(t, wanted, err)
|
|
}
|
|
|
|
func TestStartAndEndPage_InvalidPageValues(t *testing.T) {
|
|
_, _, _, err := pagination.StartAndEndPage("10", -1, 10)
|
|
assert.ErrorContains(t, "invalid page and total sizes provided", err)
|
|
|
|
_, _, _, err = pagination.StartAndEndPage("12", 10, -10)
|
|
assert.ErrorContains(t, "invalid page and total sizes provided", err)
|
|
|
|
_, _, _, err = pagination.StartAndEndPage("12", -50, -60)
|
|
assert.ErrorContains(t, "invalid page and total sizes provided", err)
|
|
}
|
|
|
|
func TestStartAndEndPage_InvalidTokenValue(t *testing.T) {
|
|
_, _, _, err := pagination.StartAndEndPage("-12", 50, 60)
|
|
assert.ErrorContains(t, "invalid token value provided", err)
|
|
}
|