From a444adee979375ed5d7af38346029a3d90bc77eb Mon Sep 17 00:00:00 2001
From: kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com>
Date: Mon, 25 Nov 2024 15:33:21 +0000
Subject: [bugfix] notification types missing from link header (#3571)

* ensure notification types get included in link header query for notifications

* fix type query keys
---
 internal/db/bundb/notification_test.go | 46 +++++++++++++++++++---------------
 1 file changed, 26 insertions(+), 20 deletions(-)

(limited to 'internal/db/bundb/notification_test.go')

diff --git a/internal/db/bundb/notification_test.go b/internal/db/bundb/notification_test.go
index eb2c02066..8e2fb8031 100644
--- a/internal/db/bundb/notification_test.go
+++ b/internal/db/bundb/notification_test.go
@@ -28,6 +28,7 @@ import (
 	"github.com/superseriousbusiness/gotosocial/internal/gtscontext"
 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
 	"github.com/superseriousbusiness/gotosocial/internal/id"
+	"github.com/superseriousbusiness/gotosocial/internal/paging"
 	"github.com/superseriousbusiness/gotosocial/internal/util"
 )
 
@@ -92,10 +93,11 @@ func (suite *NotificationTestSuite) TestGetAccountNotificationsWithSpam() {
 	notifications, err := suite.db.GetAccountNotifications(
 		gtscontext.SetBarebones(context.Background()),
 		testAccount.ID,
-		id.Highest,
-		id.Lowest,
-		"",
-		20,
+		&paging.Page{
+			Min:   paging.EitherMinID("", id.Lowest),
+			Max:   paging.MaxID(id.Highest),
+			Limit: 20,
+		},
 		nil,
 		nil,
 	)
@@ -115,10 +117,11 @@ func (suite *NotificationTestSuite) TestGetAccountNotificationsWithoutSpam() {
 	notifications, err := suite.db.GetAccountNotifications(
 		gtscontext.SetBarebones(context.Background()),
 		testAccount.ID,
-		id.Highest,
-		id.Lowest,
-		"",
-		20,
+		&paging.Page{
+			Min:   paging.EitherMinID("", id.Lowest),
+			Max:   paging.MaxID(id.Highest),
+			Limit: 20,
+		},
 		nil,
 		nil,
 	)
@@ -140,10 +143,11 @@ func (suite *NotificationTestSuite) TestDeleteNotificationsWithSpam() {
 	notifications, err := suite.db.GetAccountNotifications(
 		gtscontext.SetBarebones(context.Background()),
 		testAccount.ID,
-		id.Highest,
-		id.Lowest,
-		"",
-		20,
+		&paging.Page{
+			Min:   paging.EitherMinID("", id.Lowest),
+			Max:   paging.MaxID(id.Highest),
+			Limit: 20,
+		},
 		nil,
 		nil,
 	)
@@ -161,10 +165,11 @@ func (suite *NotificationTestSuite) TestDeleteNotificationsWithSpam() {
 	notifications, err = suite.db.GetAccountNotifications(
 		gtscontext.SetBarebones(context.Background()),
 		testAccount.ID,
-		id.Highest,
-		id.Lowest,
-		"",
-		20,
+		&paging.Page{
+			Min:   paging.EitherMinID("", id.Lowest),
+			Max:   paging.MaxID(id.Highest),
+			Limit: 20,
+		},
 		nil,
 		nil,
 	)
@@ -183,10 +188,11 @@ func (suite *NotificationTestSuite) TestDeleteNotificationsWithTwoAccounts() {
 	notifications, err := suite.db.GetAccountNotifications(
 		gtscontext.SetBarebones(context.Background()),
 		testAccount.ID,
-		id.Highest,
-		id.Lowest,
-		"",
-		20,
+		&paging.Page{
+			Min:   paging.EitherMinID("", id.Lowest),
+			Max:   paging.MaxID(id.Highest),
+			Limit: 20,
+		},
 		nil,
 		nil,
 	)
-- 
cgit v1.2.3