From 4fdbef04b4ae41e6193d4f4416c83edf91e1bfb7 Mon Sep 17 00:00:00 2001 From: Artémis Date: Mon, 1 Aug 2022 11:13:49 +0200 Subject: [feature] Implemented notification clear (#720) * Implemented notification clear * Added the cache clear mechanism * added multi user check test --- internal/db/bundb/notification_test.go | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) (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 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)) } -- cgit v1.2.3