diff options
Diffstat (limited to 'internal/db/bundb/notification_test.go')
| -rw-r--r-- | internal/db/bundb/notification_test.go | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/internal/db/bundb/notification_test.go b/internal/db/bundb/notification_test.go index b14704dcc..d822c9a28 100644 --- a/internal/db/bundb/notification_test.go +++ b/internal/db/bundb/notification_test.go @@ -118,6 +118,35 @@ func (suite *NotificationTestSuite) TestGetNotificationsWithoutSpam() { } } +func (suite *NotificationTestSuite) TestClearNotificationsWithSpam() { + suite.spamNotifs() + testAccount := suite.testAccounts["local_account_1"] + err := suite.db.ClearNotifications(context.Background(), testAccount.ID) + suite.NoError(err) + + notifications, err := suite.db.GetNotifications(context.Background(), testAccount.ID, 20, "ZZZZZZZZZZZZZZZZZZZZZZZZZZ", "00000000000000000000000000") + suite.NoError(err) + suite.NotNil(notifications) + suite.Empty(notifications) +} + +func (suite *NotificationTestSuite) TestClearNotificationsWithTwoAccounts() { + suite.spamNotifs() + testAccount := suite.testAccounts["local_account_1"] + err := suite.db.ClearNotifications(context.Background(), testAccount.ID) + suite.NoError(err) + + notifications, err := suite.db.GetNotifications(context.Background(), testAccount.ID, 20, "ZZZZZZZZZZZZZZZZZZZZZZZZZZ", "00000000000000000000000000") + suite.NoError(err) + suite.NotNil(notifications) + suite.Empty(notifications) + + notif := []*gtsmodel.Notification{} + err = suite.db.GetAll(context.Background(), ¬if) + suite.NoError(err) + suite.NotEmpty(notif) +} + func TestNotificationTestSuite(t *testing.T) { suite.Run(t, new(NotificationTestSuite)) } |