package protocol import ( "strconv" "testing" "time" "github.com/stretchr/testify/suite" "github.com/status-im/status-go/eth-node/types" "github.com/status-im/status-go/protocol/common" "github.com/status-im/status-go/protocol/protobuf" ) func TestActivityCenterPersistence(t *testing.T) { suite.Run(t, new(ActivityCenterPersistenceTestSuite)) } type ActivityCenterPersistenceTestSuite struct { suite.Suite idCounter int } func (s *ActivityCenterPersistenceTestSuite) SetupTest() { s.idCounter = 0 } func currentMilliseconds() uint64 { c := time.Now().UnixMilli() return uint64(c) } func (s *ActivityCenterPersistenceTestSuite) createNotifications(p *sqlitePersistence, notifications []*ActivityCenterNotification) []*ActivityCenterNotification { now := currentMilliseconds() for index, notif := range notifications { if notif.Timestamp == 0 { notif.Timestamp = now } if len(notif.ID) == 0 { s.idCounter++ notif.ID = types.HexBytes(strconv.Itoa(s.idCounter + index)) } if notif.UpdatedAt == 0 { notif.UpdatedAt = now } _, err := p.SaveActivityCenterNotification(notif, true) s.Require().NoError(err, notif.ID) } // Fetches notifications to get an up-to-date slice. var createdNotifications []*ActivityCenterNotification for _, notif := range notifications { n, err := p.GetActivityCenterNotificationByID(notif.ID) s.Require().NoError(err, notif.ID) createdNotifications = append(createdNotifications, n) } return createdNotifications } func (s *ActivityCenterPersistenceTestSuite) Test_DeleteActivityCenterNotificationsWhenEmpty() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) chat := CreatePublicChat("test-chat", &testTimeSource{}) message := common.NewMessage() message.Text = "sample text" chat.LastMessage = message err = p.SaveChat(*chat) s.Require().NoError(err) s.createNotifications(p, []*ActivityCenterNotification{ { Type: ActivityCenterNotificationTypeMention, }, }) var count uint64 count, _ = p.ActivityCenterNotificationsCount([]ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, false) s.Require().Equal(uint64(1), count) _, err = p.MarkActivityCenterNotificationsDeleted([]types.HexBytes{}, currentMilliseconds()) s.Require().NoError(err) count, _ = p.ActivityCenterNotificationsCount([]ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, false) s.Require().Equal(uint64(1), count) } func (s *ActivityCenterPersistenceTestSuite) Test_DeleteActivityCenterNotificationsWithMultipleIds() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) chat := CreatePublicChat("test-chat", &testTimeSource{}) message := common.NewMessage() message.Text = "sample text" chat.LastMessage = message err = p.SaveChat(*chat) s.Require().NoError(err) notifications := s.createNotifications(p, []*ActivityCenterNotification{ {Type: ActivityCenterNotificationTypeMention}, {Type: ActivityCenterNotificationTypeNewOneToOne}, {Type: ActivityCenterNotificationTypeNewOneToOne}, }) var count uint64 count, _ = p.ActivityCenterNotificationsCount([]ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, false) s.Require().Equal(uint64(3), count) _, err = p.MarkActivityCenterNotificationsDeleted([]types.HexBytes{notifications[1].ID, notifications[2].ID}, currentMilliseconds()) s.Require().NoError(err) count, _ = p.ActivityCenterNotificationsCount([]ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, false) s.Require().Equal(uint64(1), count) } func (s *ActivityCenterPersistenceTestSuite) Test_DeleteActivityCenterNotificationsForMessage() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) chat := CreatePublicChat("test-chat", &testTimeSource{}) err = p.SaveChat(*chat) s.Require().NoError(err) chat2 := CreatePublicChat("test-chat", &testTimeSource{}) err = p.SaveChat(*chat2) s.Require().NoError(err) messages := []*common.Message{ { ID: "0x1", ChatMessage: &protobuf.ChatMessage{}, LocalChatID: chat.ID, }, { ID: "0x2", ChatMessage: &protobuf.ChatMessage{}, LocalChatID: chat.ID, }, { ChatMessage: &protobuf.ChatMessage{}, ID: "0x3", }, } err = p.SaveMessages(messages) s.Require().NoError(err) chat.LastMessage = messages[1] err = p.SaveChat(*chat) s.Require().NoError(err) chatMessages, _, err := p.MessageByChatID(chat.ID, "", 2) s.Require().NoError(err) s.Require().Len(chatMessages, 2) nID1 := types.HexBytes("1") nID2 := types.HexBytes("2") nID3 := types.HexBytes("3") nID4 := types.HexBytes("4") s.createNotifications(p, []*ActivityCenterNotification{ { ID: nID1, ChatID: chat.ID, Type: ActivityCenterNotificationTypeMention, Message: messages[0], }, { ID: nID2, ChatID: chat.ID, Type: ActivityCenterNotificationTypeMention, Message: messages[1], }, { ID: nID3, ChatID: chat.ID, Type: ActivityCenterNotificationTypeMention, }, { ID: nID4, Type: ActivityCenterNotificationTypeMention, }, }) // Test: soft delete only the notifications that have Message.ID == messages[0].ID. _, err = p.DeleteActivityCenterNotificationForMessage(chat.ID, messages[0].ID, currentMilliseconds()) s.Require().NoError(err) notif, err := p.GetActivityCenterNotificationByID(nID1) s.Require().NoError(err) s.Require().True(notif.Deleted) s.Require().True(notif.Dismissed) s.Require().True(notif.Read) // Other notifications are not affected. for _, id := range []types.HexBytes{nID2, nID3, nID4} { notif, err = p.GetActivityCenterNotificationByID(id) s.Require().NoError(err) s.Require().False(notif.Deleted, notif.ID) s.Require().False(notif.Dismissed, notif.ID) s.Require().False(notif.Read, notif.ID) } // Test: soft delete the notifications that have Message.ID == messages[1].ID // or LastMessage.ID == chat.LastMessage. _, err = p.DeleteActivityCenterNotificationForMessage(chat.ID, messages[1].ID, currentMilliseconds()) s.Require().NoError(err) for _, id := range []types.HexBytes{nID2, nID3} { notif, err = p.GetActivityCenterNotificationByID(id) s.Require().NoError(err, notif.ID) s.Require().True(notif.Deleted, notif.ID) s.Require().True(notif.Dismissed, notif.ID) s.Require().True(notif.Read, notif.ID) } notif, err = p.GetActivityCenterNotificationByID(nID4) s.Require().NoError(err) s.Require().False(notif.Deleted) s.Require().False(notif.Dismissed) s.Require().False(notif.Read) // Test: don't do anything if passed a chat and message without notifications. _, err = p.DeleteActivityCenterNotificationForMessage(chat2.ID, messages[2].ID, currentMilliseconds()) s.Require().NoError(err) } func (s *ActivityCenterPersistenceTestSuite) Test_AcceptActivityCenterNotificationsForInvitesFromUser() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) nID1 := types.HexBytes("1") nID2 := types.HexBytes("2") nID3 := types.HexBytes("3") nID4 := types.HexBytes("4") userPublicKey := "zQ3sh" notifications := []*ActivityCenterNotification{ { ID: nID1, Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Timestamp: 1, }, { ID: nID2, Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Timestamp: 1, Author: userPublicKey, }, { ID: nID3, Type: ActivityCenterNotificationTypeMention, Timestamp: 1, Author: userPublicKey, }, { ID: nID4, Timestamp: 1, Type: ActivityCenterNotificationTypeMention, }, } var notif *ActivityCenterNotification for _, notif = range notifications { _, err = p.SaveActivityCenterNotification(notif, true) s.Require().NoError(err, notif.ID) } // Only notifications of type new private group chat and with Author equal to // userPublicKey should be marked as accepted & read. _, err = p.GetActivityCenterNotificationByID(nID2) s.Require().NoError(err) s.Require().False(notifications[0].Accepted) s.Require().False(notifications[0].Read) notifications, err = p.AcceptActivityCenterNotificationsForInvitesFromUser(userPublicKey, 1) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().Equal(nID2, notifications[0].ID) notif, err = p.GetActivityCenterNotificationByID(nID2) s.Require().NoError(err) s.Require().True(notif.Accepted) s.Require().True(notif.Read) // Deleted notifications are ignored. notif = &ActivityCenterNotification{ ID: types.HexBytes("99"), Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Timestamp: 1, Author: userPublicKey, Deleted: true, } _, err = p.SaveActivityCenterNotification(notif, true) s.Require().NoError(err) _, err = p.AcceptActivityCenterNotificationsForInvitesFromUser(userPublicKey, currentMilliseconds()) s.Require().NoError(err) notif, err = p.GetActivityCenterNotificationByID(notif.ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().False(notif.Read) s.Require().True(notif.Deleted) // Dismissed notifications are ignored. notif = &ActivityCenterNotification{ ID: types.HexBytes("100"), Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Timestamp: 1, Author: userPublicKey, Dismissed: true, } _, err = p.SaveActivityCenterNotification(notif, true) s.Require().NoError(err) _, err = p.AcceptActivityCenterNotificationsForInvitesFromUser(userPublicKey, currentMilliseconds()) s.Require().NoError(err) notif, err = p.GetActivityCenterNotificationByID(notif.ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().False(notif.Read) s.Require().True(notif.Dismissed) } func (s *ActivityCenterPersistenceTestSuite) Test_GetToProcessActivityCenterNotificationIds() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) notifications := s.createNotifications(p, []*ActivityCenterNotification{ { Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Deleted: true, }, { Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Dismissed: true, }, { Type: ActivityCenterNotificationTypeMention, Accepted: true, }, { Type: ActivityCenterNotificationTypeMention, }, }) ids, err := p.GetToProcessActivityCenterNotificationIds() s.Require().NoError(err) s.Require().Len(ids, 1) s.Require().Equal(notifications[3].ID, types.HexBytes(ids[0])) } func (s *ActivityCenterPersistenceTestSuite) Test_HasPendingNotificationsForChat() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) chat := CreatePublicChat("test-chat", &testTimeSource{}) err = p.SaveChat(*chat) s.Require().NoError(err) // Test: there are no notifications. result, err := p.HasPendingNotificationsForChat(chat.ID) s.Require().NoError(err) s.Require().False(result) // Test: there are only deleted, dismissed or accepted notifications, // therefore, no pending notifications. s.createNotifications(p, []*ActivityCenterNotification{ { ChatID: chat.ID, Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Deleted: true, }, { ChatID: chat.ID, Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Dismissed: true, }, { ChatID: chat.ID, Type: ActivityCenterNotificationTypeMention, Accepted: true, }, }) result, err = p.HasPendingNotificationsForChat(chat.ID) s.Require().NoError(err) s.Require().False(result) // Test: there's one pending notification. notif := &ActivityCenterNotification{ ID: types.HexBytes("99"), ChatID: chat.ID, Type: ActivityCenterNotificationTypeCommunityRequest, Timestamp: 1, } _, err = p.SaveActivityCenterNotification(notif, true) s.Require().NoError(err) result, err = p.HasPendingNotificationsForChat(chat.ID) s.Require().NoError(err) s.Require().True(result) } func (s *ActivityCenterPersistenceTestSuite) Test_DismissAllActivityCenterNotificationsFromUser() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) publicKey := "0x04" notifications := s.createNotifications(p, []*ActivityCenterNotification{ { Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Author: publicKey, Deleted: true, }, { Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Author: publicKey, Dismissed: true, }, { Type: ActivityCenterNotificationTypeMention, Author: publicKey, Accepted: true, }, { Type: ActivityCenterNotificationTypeMention, Author: "0x09", }, { Type: ActivityCenterNotificationTypeMention, Author: publicKey, }, }) _, err = p.DismissAllActivityCenterNotificationsFromUser(publicKey, 1) s.Require().NoError(err) // Ignores already soft deleted. notif, err := p.GetActivityCenterNotificationByID(notifications[0].ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().False(notif.Read) s.Require().False(notif.Dismissed) s.Require().True(notif.Deleted) // Ignores already dismissed. notif, err = p.GetActivityCenterNotificationByID(notifications[1].ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().False(notif.Read) s.Require().True(notif.Dismissed) s.Require().False(notif.Deleted) // Ignores already accepted. notif, err = p.GetActivityCenterNotificationByID(notifications[2].ID) s.Require().NoError(err) s.Require().True(notif.Accepted) s.Require().False(notif.Read) s.Require().False(notif.Dismissed) s.Require().False(notif.Deleted) // Ignores notification from different author. notif, err = p.GetActivityCenterNotificationByID(notifications[3].ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().False(notif.Read) s.Require().False(notif.Dismissed) s.Require().False(notif.Deleted) // Finally, dismiss and mark as read this one notification. notif, err = p.GetActivityCenterNotificationByID(notifications[4].ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().True(notif.Read) s.Require().True(notif.Dismissed) s.Require().False(notif.Deleted) } func (s *ActivityCenterPersistenceTestSuite) Test_DismissAllActivityCenterNotificationsFromChatID() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) chatID := "0x99" notifications := s.createNotifications(p, []*ActivityCenterNotification{ { ChatID: chatID, Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Deleted: true, }, { ChatID: chatID, Type: ActivityCenterNotificationTypeNewPrivateGroupChat, Dismissed: true, }, { ChatID: chatID, Type: ActivityCenterNotificationTypeMention, Accepted: true, }, { Type: ActivityCenterNotificationTypeMention, }, { ChatID: chatID, Type: ActivityCenterNotificationTypeContactRequest, }, { ChatID: chatID, Type: ActivityCenterNotificationTypeMention, }, }) _, err = p.DismissAllActivityCenterNotificationsFromChatID(chatID, currentMilliseconds()) s.Require().NoError(err) // Ignores already soft deleted. notif, err := p.GetActivityCenterNotificationByID(notifications[0].ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().False(notif.Read) s.Require().False(notif.Dismissed) s.Require().True(notif.Deleted) // Do not ignore already dismissed, because notifications can become // read/unread AND dismissed, and the method should still update the Read // column. notif, err = p.GetActivityCenterNotificationByID(notifications[1].ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().True(notif.Read) s.Require().True(notif.Dismissed) s.Require().False(notif.Deleted) // Ignores already accepted. notif, err = p.GetActivityCenterNotificationByID(notifications[2].ID) s.Require().NoError(err) s.Require().True(notif.Accepted) s.Require().False(notif.Read) s.Require().False(notif.Dismissed) s.Require().False(notif.Deleted) // Ignores notification from different chat. notif, err = p.GetActivityCenterNotificationByID(notifications[3].ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().False(notif.Read) s.Require().False(notif.Dismissed) s.Require().False(notif.Deleted) // Ignores contact request notifications. notif, err = p.GetActivityCenterNotificationByID(notifications[4].ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().False(notif.Read) s.Require().False(notif.Dismissed) s.Require().False(notif.Deleted) // Finally, dismiss and mark as read this one notification. notif, err = p.GetActivityCenterNotificationByID(notifications[5].ID) s.Require().NoError(err) s.Require().False(notif.Accepted) s.Require().True(notif.Read) s.Require().True(notif.Dismissed) s.Require().False(notif.Deleted) } func (s *ActivityCenterPersistenceTestSuite) Test_ActiveContactRequestNotification() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) chat := CreatePublicChat("test-chat", &testTimeSource{}) err = p.SaveChat(*chat) s.Require().NoError(err) contactID := "0x99" // Test: ignores deleted/dismissed/accepted notifications, as well as // notifications not associated to any chat. s.createNotifications(p, []*ActivityCenterNotification{ { ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeContactRequest, Deleted: true, }, { ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeContactRequest, Dismissed: true, }, { ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeContactRequest, Accepted: true, }, }) notif, err := p.ActiveContactRequestNotification(contactID) s.Require().NoError(err) s.Require().Nil(notif) // Test: Ignores notifications that are not contact requests. s.createNotifications(p, []*ActivityCenterNotification{ {ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeCommunityInvitation}, {ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeCommunityKicked}, {ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeCommunityMembershipRequest}, {ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeCommunityRequest}, {ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeContactVerification}, {ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeMention}, {ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeNewOneToOne}, {ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeNewPrivateGroupChat}, {ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeReply}, }) notif, err = p.ActiveContactRequestNotification(contactID) s.Require().NoError(err) s.Require().Nil(notif) // Test: Returns one, and only one contact request notification for the // contact under test. s.createNotifications(p, []*ActivityCenterNotification{ {ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeContactRequest}, }) notif, err = p.ActiveContactRequestNotification(contactID) s.Require().NoError(err) s.Require().NotNil(notif) s.Require().Equal(ActivityCenterNotificationTypeContactRequest, notif.Type) // Test: In case there's more than one notification, return the most recent // one according to the notification's timestamp. expectedID := types.HexBytes("667") t1 := currentMilliseconds() t2 := currentMilliseconds() s.createNotifications(p, []*ActivityCenterNotification{ { ID: expectedID, Timestamp: t2 + 1, ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeContactRequest, }, { ID: types.HexBytes("666"), Timestamp: t1, ChatID: chat.ID, Author: contactID, Type: ActivityCenterNotificationTypeContactRequest, }, }) notif, err = p.ActiveContactRequestNotification(contactID) s.Require().NoError(err) s.Require().Equal(expectedID, notif.ID) } func (s *ActivityCenterPersistenceTestSuite) Test_UnreadActivityCenterNotificationsCount() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) s.createNotifications(p, []*ActivityCenterNotification{ {Type: ActivityCenterNotificationTypeMention, Read: true}, {Type: ActivityCenterNotificationTypeNewOneToOne, Deleted: true}, {Type: ActivityCenterNotificationTypeMention, Dismissed: true}, {Type: ActivityCenterNotificationTypeCommunityRequest, Accepted: true}, {Type: ActivityCenterNotificationTypeContactRequest}, }) // Test: Ignore soft deleted and accepted. count, err := p.ActivityCenterNotificationsCount([]ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, true) s.Require().NoError(err) s.Require().Equal(uint64(3), count) } func (s *ActivityCenterPersistenceTestSuite) Test_UnreadAndAcceptedActivityCenterNotificationsCount() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) s.createNotifications(p, []*ActivityCenterNotification{ {Type: ActivityCenterNotificationTypeMention, Read: true}, {Type: ActivityCenterNotificationTypeNewOneToOne, Deleted: true}, {Type: ActivityCenterNotificationTypeMention, Dismissed: true}, {Type: ActivityCenterNotificationTypeCommunityRequest, Accepted: true}, {Type: ActivityCenterNotificationTypeContactRequest}, }) // Test: counts everything, except soft deleted notifications. count, err := p.ActivityCenterNotificationsCount([]ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, true) s.Require().NoError(err) s.Require().Equal(uint64(3), count) // Test: counts everything, except soft deleted ones and limit by type. count, err = p.ActivityCenterNotificationsCount([]ActivityCenterType{ ActivityCenterNotificationTypeContactRequest, }, ActivityCenterQueryParamsReadUnread, true) s.Require().NoError(err) s.Require().Equal(uint64(1), count) } func (s *ActivityCenterPersistenceTestSuite) Test_ActivityCenterPersistence() { nID1 := types.HexBytes([]byte("1")) nID2 := types.HexBytes([]byte("2")) nID3 := types.HexBytes([]byte("3")) nID4 := types.HexBytes([]byte("4")) db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) chat := CreatePublicChat("test-chat", &testTimeSource{}) message := common.NewMessage() message.Text = "sample text" chat.LastMessage = message err = p.SaveChat(*chat) s.Require().NoError(err) notification := &ActivityCenterNotification{ ID: nID1, Type: ActivityCenterNotificationTypeNewOneToOne, ChatID: chat.ID, Timestamp: 1, } _, err = p.SaveActivityCenterNotification(notification, true) s.Require().NoError(err) cursor, notifications, err := p.ActivityCenterNotifications("", 2, []ActivityCenterType{}, ActivityCenterQueryParamsReadAll, false) s.Require().NoError(err) s.Require().Empty(cursor) s.Require().Len(notifications, 1) s.Require().Equal(chat.ID, notifications[0].ChatID) s.Require().Equal(message, notifications[0].LastMessage) // Add another notification notification = &ActivityCenterNotification{ ID: nID2, Type: ActivityCenterNotificationTypeNewOneToOne, Timestamp: 2, } _, err = p.SaveActivityCenterNotification(notification, true) s.Require().NoError(err) cursor, notifications, err = p.ActivityCenterNotifications("", 1, []ActivityCenterType{}, ActivityCenterQueryParamsReadAll, false) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().NotEmpty(cursor) s.Require().Equal(nID2, notifications[0].ID) // fetch next pagination cursor, notifications, err = p.ActivityCenterNotifications(cursor, 1, []ActivityCenterType{}, ActivityCenterQueryParamsReadAll, false) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().Empty(cursor) s.Require().False(notifications[0].Read) s.Require().Equal(nID1, notifications[0].ID) // Check count count, err := p.ActivityCenterNotificationsCount([]ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, false) s.Require().NoError(err) s.Require().Equal(uint64(2), count) var updatedAt uint64 = 1 // Mark first one as read s.Require().NoError(p.MarkActivityCenterNotificationsRead([]types.HexBytes{nID1}, updatedAt)) count, err = p.ActivityCenterNotificationsCount([]ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, false) s.Require().NoError(err) s.Require().Equal(uint64(1), count) // Mark first one as unread updatedAt++ _, err = p.MarkActivityCenterNotificationsUnread([]types.HexBytes{nID1}, updatedAt) s.Require().NoError(err) count, err = p.ActivityCenterNotificationsCount([]ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, false) s.Require().NoError(err) s.Require().Equal(uint64(2), count) // Mark all read updatedAt++ s.Require().NoError(p.MarkAllActivityCenterNotificationsRead(updatedAt)) _, notifications, err = p.ActivityCenterNotifications(cursor, 2, []ActivityCenterType{}, ActivityCenterQueryParamsReadAll, false) s.Require().NoError(err) s.Require().Len(notifications, 2) s.Require().Empty(cursor) s.Require().True(notifications[0].Read) s.Require().True(notifications[1].Read) // Check count count, err = p.ActivityCenterNotificationsCount([]ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, false) s.Require().NoError(err) s.Require().Equal(uint64(0), count) // Mark first one as accepted updatedAt++ notifications, err = p.AcceptActivityCenterNotifications([]types.HexBytes{nID1}, updatedAt) s.Require().NoError(err) s.Require().Len(notifications, 1) _, notifications, err = p.ActivityCenterNotifications("", 2, []ActivityCenterType{}, ActivityCenterQueryParamsReadAll, false) s.Require().NoError(err) // It should not be returned anymore s.Require().Len(notifications, 1) // Mark last one as dismissed updatedAt++ s.Require().NoError(p.DismissActivityCenterNotifications([]types.HexBytes{nID2}, updatedAt)) _, notifications, err = p.ActivityCenterNotifications("", 2, []ActivityCenterType{}, ActivityCenterQueryParamsReadAll, false) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().True(notifications[0].Dismissed) // Insert new notification notification = &ActivityCenterNotification{ ID: nID3, Type: ActivityCenterNotificationTypeNewOneToOne, Timestamp: 3, } _, err = p.SaveActivityCenterNotification(notification, true) s.Require().NoError(err) // Mark all as accepted updatedAt++ notifications, err = p.AcceptAllActivityCenterNotifications(updatedAt) s.Require().NoError(err) s.Require().Len(notifications, 2) _, notifications, err = p.ActivityCenterNotifications("", 2, []ActivityCenterType{}, ActivityCenterQueryParamsReadAll, false) s.Require().NoError(err) s.Require().Len(notifications, 1) // Insert new notification notification = &ActivityCenterNotification{ ID: nID4, Type: ActivityCenterNotificationTypeNewOneToOne, Timestamp: 4, } _, err = p.SaveActivityCenterNotification(notification, true) s.Require().NoError(err) // Mark all as dismissed updatedAt++ s.Require().NoError(p.DismissAllActivityCenterNotifications(updatedAt)) _, notifications, err = p.ActivityCenterNotifications("", 2, []ActivityCenterType{}, ActivityCenterQueryParamsReadAll, false) s.Require().NoError(err) s.Require().Len(notifications, 2) s.Require().True(notifications[0].Dismissed) s.Require().True(notifications[1].Dismissed) } func (s *ActivityCenterPersistenceTestSuite) Test_ActivityCenterReadUnreadPagination() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) initialOrFinalCursor := "" chat := CreatePublicChat("test-chat", &testTimeSource{}) message := common.NewMessage() message.Text = "sample text" chat.LastMessage = message err = p.SaveChat(*chat) s.Require().NoError(err) nID1 := types.HexBytes("1") nID2 := types.HexBytes("2") nID3 := types.HexBytes("3") nID4 := types.HexBytes("4") nID5 := types.HexBytes("5") allNotifications := []*ActivityCenterNotification{ { ID: nID1, Type: ActivityCenterNotificationTypeNewOneToOne, ChatID: chat.ID, Timestamp: 1, }, { ID: nID2, Type: ActivityCenterNotificationTypeNewOneToOne, ChatID: chat.ID, Timestamp: 1, }, { ID: nID3, Type: ActivityCenterNotificationTypeNewOneToOne, ChatID: chat.ID, Timestamp: 1, }, { ID: nID4, Type: ActivityCenterNotificationTypeNewOneToOne, ChatID: chat.ID, Timestamp: 1, }, { ID: nID5, Type: ActivityCenterNotificationTypeNewOneToOne, ChatID: chat.ID, Timestamp: 1, }, } for _, notification := range allNotifications { _, err = p.SaveActivityCenterNotification(notification, true) s.Require().NoError(err) } // Mark the notification as read err = p.MarkActivityCenterNotificationsRead([]types.HexBytes{nID2}, currentMilliseconds()) s.Require().NoError(err) err = p.MarkActivityCenterNotificationsRead([]types.HexBytes{nID4}, currentMilliseconds()) s.Require().NoError(err) // Fetch UNREAD notifications, first page. cursor, notifications, err := p.ActivityCenterNotifications( initialOrFinalCursor, 1, []ActivityCenterType{ActivityCenterNotificationTypeNewOneToOne}, ActivityCenterQueryParamsReadUnread, false, ) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().Equal(nID5, notifications[0].ID) s.Require().NotEmpty(cursor) // Fetch next pages. cursor, notifications, err = p.ActivityCenterNotifications( cursor, 1, []ActivityCenterType{ActivityCenterNotificationTypeNewOneToOne}, ActivityCenterQueryParamsReadUnread, false, ) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().Equal(nID3, notifications[0].ID) s.Require().NotEmpty(cursor) cursor, notifications, err = p.ActivityCenterNotifications( cursor, 1, []ActivityCenterType{ActivityCenterNotificationTypeNewOneToOne}, ActivityCenterQueryParamsReadUnread, false, ) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().Equal(nID1, notifications[0].ID) s.Require().Empty(cursor) // Fetch READ notifications, first page. cursor, notifications, err = p.ActivityCenterNotifications( initialOrFinalCursor, 1, []ActivityCenterType{ActivityCenterNotificationTypeNewOneToOne}, ActivityCenterQueryParamsReadRead, false, ) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().Equal(nID4, notifications[0].ID) s.Require().NotEmpty(cursor) // Fetch next page. cursor, notifications, err = p.ActivityCenterNotifications( cursor, 1, []ActivityCenterType{ActivityCenterNotificationTypeNewOneToOne}, ActivityCenterQueryParamsReadRead, false, ) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().Equal(nID2, notifications[0].ID) s.Require().Empty(cursor) } func (s *ActivityCenterPersistenceTestSuite) Test_ActivityCenterReadUnreadFilterByTypes() { db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) chat := CreatePublicChat("test-chat", &testTimeSource{}) message := common.NewMessage() message.Text = "sample text" chat.LastMessage = message err = p.SaveChat(*chat) s.Require().NoError(err) initialCursor := "" limit := uint64(3) nID1 := types.HexBytes("1") nID2 := types.HexBytes("2") nID3 := types.HexBytes("3") allNotifications := []*ActivityCenterNotification{ { ID: nID1, Type: ActivityCenterNotificationTypeMention, ChatID: chat.ID, Timestamp: 1, }, { ID: nID2, Type: ActivityCenterNotificationTypeNewOneToOne, ChatID: chat.ID, Timestamp: 1, }, { ID: nID3, Type: ActivityCenterNotificationTypeMention, ChatID: chat.ID, Timestamp: 1, }, } for _, notification := range allNotifications { _, err = p.SaveActivityCenterNotification(notification, true) s.Require().NoError(err) } // Don't filter by type if the array of types is empty. _, notifications, err := p.ActivityCenterNotifications( initialCursor, limit, []ActivityCenterType{}, ActivityCenterQueryParamsReadUnread, false, ) s.Require().NoError(err) s.Require().Len(notifications, 3) s.Require().Equal(nID3, notifications[0].ID) s.Require().Equal(nID2, notifications[1].ID) s.Require().Equal(nID1, notifications[2].ID) _, notifications, err = p.ActivityCenterNotifications( initialCursor, limit, []ActivityCenterType{ActivityCenterNotificationTypeNewOneToOne}, ActivityCenterQueryParamsReadUnread, false, ) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().Equal(nID2, notifications[0].ID) _, notifications, err = p.ActivityCenterNotifications( initialCursor, limit, []ActivityCenterType{ActivityCenterNotificationTypeMention}, ActivityCenterQueryParamsReadUnread, false, ) s.Require().NoError(err) s.Require().Len(notifications, 2) s.Require().Equal(nID3, notifications[0].ID) s.Require().Equal(nID1, notifications[1].ID) _, notifications, err = p.ActivityCenterNotifications( initialCursor, limit, []ActivityCenterType{ActivityCenterNotificationTypeMention, ActivityCenterNotificationTypeNewOneToOne}, ActivityCenterQueryParamsReadUnread, false, ) s.Require().NoError(err) s.Require().Len(notifications, 3) s.Require().Equal(nID3, notifications[0].ID) s.Require().Equal(nID2, notifications[1].ID) s.Require().Equal(nID1, notifications[2].ID) // Mark all notifications as read. for _, notification := range allNotifications { err = p.MarkActivityCenterNotificationsRead([]types.HexBytes{notification.ID}, currentMilliseconds()) s.Require().NoError(err) } _, notifications, err = p.ActivityCenterNotifications( initialCursor, limit, []ActivityCenterType{ActivityCenterNotificationTypeNewOneToOne}, ActivityCenterQueryParamsReadRead, false, ) s.Require().NoError(err) s.Require().Len(notifications, 1) s.Require().Equal(nID2, notifications[0].ID) _, notifications, err = p.ActivityCenterNotifications( initialCursor, limit, []ActivityCenterType{ActivityCenterNotificationTypeMention}, ActivityCenterQueryParamsReadRead, false, ) s.Require().NoError(err) s.Require().Len(notifications, 2) s.Require().Equal(nID3, notifications[0].ID) s.Require().Equal(nID1, notifications[1].ID) } func (s *ActivityCenterPersistenceTestSuite) Test_ActivityCenterReadUnread() { nID1 := types.HexBytes("1") nID2 := types.HexBytes("2") db, err := openTestDB() s.Require().NoError(err) p := newSQLitePersistence(db) chat := CreatePublicChat("test-chat", &testTimeSource{}) message := common.NewMessage() message.Text = "sample text" chat.LastMessage = message err = p.SaveChat(*chat) s.Require().NoError(err) notification := &ActivityCenterNotification{ ID: nID1, Type: ActivityCenterNotificationTypeNewOneToOne, ChatID: chat.ID, Timestamp: 1, } _, err = p.SaveActivityCenterNotification(notification, true) s.Require().NoError(err) notification = &ActivityCenterNotification{ ID: nID2, Type: ActivityCenterNotificationTypeNewOneToOne, ChatID: chat.ID, Timestamp: 1, } _, err = p.SaveActivityCenterNotification(notification, true) s.Require().NoError(err) // Mark the notification as read err = p.MarkActivityCenterNotificationsRead([]types.HexBytes{nID2}, currentMilliseconds()) s.Require().NoError(err) cursor, notifications, err := p.ActivityCenterNotifications( "", 2, []ActivityCenterType{ActivityCenterNotificationTypeNewOneToOne}, ActivityCenterQueryParamsReadUnread, false, ) s.Require().NoError(err) s.Require().Empty(cursor) s.Require().Len(notifications, 1) s.Require().Equal(nID1, notifications[0].ID) cursor, notifications, err = p.ActivityCenterNotifications( "", 2, []ActivityCenterType{ActivityCenterNotificationTypeNewOneToOne}, ActivityCenterQueryParamsReadRead, false, ) s.Require().NoError(err) s.Require().Empty(cursor) s.Require().Len(notifications, 1) s.Require().Equal(nID2, notifications[0].ID) }