Add total results and pages to pager

cloudfoundry · Jan 20, 2023 · 1561eb6 · 1561eb6
1 parent d55b6aa
commit 1561eb6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/client/pager.go b/client/pager.go
@@ -13,6 +13,9 @@ type Pager struct {
 	NextPageURL     string
 	PreviousPageURL string
 
+	TotalResults int
+	TotalPages   int
+
 	nextPageQSReader     *path.QuerystringReader
 	previousPageQSReader *path.QuerystringReader
 }
@@ -21,6 +24,8 @@ func NewPager(pagination resource.Pagination) *Pager {
 	return &Pager{
 		NextPageURL:     pagination.Next.Href,
 		PreviousPageURL: pagination.Previous.Href,
+		TotalResults:    pagination.TotalResults,
+		TotalPages:      pagination.TotalPages,
 	}
 }
 

diff --git a/client/pager_test.go b/client/pager_test.go
@@ -61,6 +61,9 @@ func TestPager(t *testing.T) {
 
 	// First page
 	pager := NewPager(paginationPage1)
+	require.Equal(t, 120, pager.TotalResults)
+	require.Equal(t, 3, pager.TotalPages)
+
 	require.True(t, pager.HasNextPage())
 	require.False(t, pager.HasPreviousPage())
 	pager.NextPage(listOpts)