From 3e5b220810563716954fd4eb2e36b935f3455902 Mon Sep 17 00:00:00 2001
From: Joakim Olsson <joakim@unbound.se>
Date: Thu, 1 Jan 2026 15:37:19 +0100
Subject: [PATCH] feat: add TotalCount field to PageInfo test: add TotalCount
 assertions to existing tests

---
 pagination.go      | 2 ++
 pagination_test.go | 8 ++++++++
 2 files changed, 10 insertions(+)

diff --git a/pagination.go b/pagination.go
index ee19c44..d24cd55 100644
--- a/pagination.go
+++ b/pagination.go
@@ -99,6 +99,7 @@ func GetPage[T any](items []T, first *int, after *string, last *int, before *str
 		HasNextPage:     eIx < len(items),
 		HasPreviousPage: sIx > 0,
 		EndCursor:       ptr(EncodeCursor(fn(page[len(page)-1]))),
+		TotalCount:      len(items),
 	}
 }
 
@@ -111,4 +112,5 @@ type PageInfo struct {
 	HasNextPage     bool
 	HasPreviousPage bool
 	EndCursor       *string
+	TotalCount      int
 }
diff --git a/pagination_test.go b/pagination_test.go
index 4c42b81..26a322d 100644
--- a/pagination_test.go
+++ b/pagination_test.go
@@ -43,6 +43,7 @@ func TestGetPage(t *testing.T) {
 				StartCursor: ptr("MQ=="),
 				HasNextPage: true,
 				EndCursor:   ptr("Mg=="),
+				TotalCount:  3,
 			},
 		},
 		{
@@ -59,6 +60,7 @@ func TestGetPage(t *testing.T) {
 				StartCursor: ptr("MQ=="),
 				HasNextPage: false,
 				EndCursor:   ptr("Mw=="),
+				TotalCount:  3,
 			},
 		},
 		{
@@ -70,6 +72,7 @@ func TestGetPage(t *testing.T) {
 				HasNextPage:     false,
 				HasPreviousPage: true,
 				EndCursor:       ptr("NA=="),
+				TotalCount:      4,
 			},
 		},
 		{
@@ -81,6 +84,7 @@ func TestGetPage(t *testing.T) {
 				HasNextPage:     false,
 				HasPreviousPage: true,
 				EndCursor:       ptr("Mw=="),
+				TotalCount:      3,
 			},
 		},
 		{
@@ -92,6 +96,7 @@ func TestGetPage(t *testing.T) {
 				HasNextPage:     false,
 				HasPreviousPage: true,
 				EndCursor:       ptr("Mw=="),
+				TotalCount:      3,
 			},
 		},
 		{
@@ -103,6 +108,7 @@ func TestGetPage(t *testing.T) {
 				HasNextPage:     true,
 				HasPreviousPage: false,
 				EndCursor:       ptr("Mg=="),
+				TotalCount:      3,
 			},
 		},
 		{
@@ -114,6 +120,7 @@ func TestGetPage(t *testing.T) {
 				HasNextPage:     true,
 				HasPreviousPage: false,
 				EndCursor:       ptr("MQ=="),
+				TotalCount:      3,
 			},
 		},
 		{
@@ -125,6 +132,7 @@ func TestGetPage(t *testing.T) {
 				HasNextPage:     false,
 				HasPreviousPage: false,
 				EndCursor:       ptr("Mw=="),
+				TotalCount:      3,
 			},
 		},
 	}