package protocol import ( "context" "crypto/ecdsa" "testing" "github.com/stretchr/testify/suite" "go.uber.org/zap" gethbridge "github.com/status-im/status-go/eth-node/bridge/geth" "github.com/status-im/status-go/eth-node/crypto" "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" "github.com/status-im/status-go/protocol/tt" "github.com/status-im/status-go/waku" ) func TestMessengerDeleteMessageSuite(t *testing.T) { suite.Run(t, new(MessengerDeleteMessageSuite)) } type MessengerDeleteMessageSuite struct { suite.Suite m *Messenger // main instance of Messenger privateKey *ecdsa.PrivateKey // private key for the main instance of Messenger // If one wants to send messages between different instances of Messenger, // a single waku service should be shared. shh types.Waku logger *zap.Logger } func (s *MessengerDeleteMessageSuite) SetupTest() { s.logger = tt.MustCreateTestLogger() config := waku.DefaultConfig config.MinimumAcceptedPoW = 0 shh := waku.New(&config, s.logger) s.shh = gethbridge.NewGethWakuWrapper(shh) s.Require().NoError(shh.Start()) s.m = s.newMessenger() s.privateKey = s.m.identity _, err := s.m.Start() s.Require().NoError(err) } func (s *MessengerDeleteMessageSuite) TearDownTest() { s.Require().NoError(s.m.Shutdown()) } func (s *MessengerDeleteMessageSuite) newMessenger() *Messenger { privateKey, err := crypto.GenerateKey() s.Require().NoError(err) messenger, err := newMessengerWithKey(s.shh, privateKey, s.logger, nil) s.Require().NoError(err) return messenger } func (s *MessengerDeleteMessageSuite) TestDeleteMessage() { theirMessenger := s.newMessenger() _, err := theirMessenger.Start() s.Require().NoError(err) theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.transport) err = theirMessenger.SaveChat(theirChat) s.Require().NoError(err) ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.transport) err = s.m.SaveChat(ourChat) s.Require().NoError(err) inputMessage := buildTestMessage(*theirChat) sendResponse, err := theirMessenger.SendChatMessage(context.Background(), inputMessage) s.NoError(err) s.Require().Len(sendResponse.Messages(), 1) messageID := sendResponse.Messages()[0].ID response, err := WaitOnMessengerResponse( s.m, func(r *MessengerResponse) bool { return len(r.messages) > 0 }, "no messages", ) s.Require().NoError(err) s.Require().Len(response.Chats(), 1) s.Require().Len(response.Messages(), 1) ogMessage := sendResponse.Messages()[0] sendResponse, err = theirMessenger.DeleteMessageAndSend(context.Background(), ogMessage.ID) s.Require().NoError(err) s.Require().Len(sendResponse.Messages(), 0) s.Require().Len(sendResponse.RemovedMessages(), 1) s.Require().Equal(messageID, sendResponse.RemovedMessages()[0].MessageID) s.Require().Equal(sendResponse.RemovedMessages()[0].DeletedBy, "") s.Require().Len(sendResponse.Chats(), 1) // LastMessage is marked as deleted lastRemovedMessage := sendResponse.Chats()[0].LastMessage s.Require().Equal(lastRemovedMessage.ID, messageID) s.Require().Equal(lastRemovedMessage.Deleted, true) // Main instance user attempts to delete the message it received from theirMessenger _, err = s.m.DeleteMessageAndSend(context.Background(), ogMessage.ID) s.Require().ErrorContains(err, "can't find chat") } func (s *MessengerDeleteMessageSuite) TestDeleteMessagePreviousLastMessage() { theirMessenger := s.newMessenger() _, err := theirMessenger.Start() s.Require().NoError(err) theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.transport) err = theirMessenger.SaveChat(theirChat) s.Require().NoError(err) ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.transport) err = s.m.SaveChat(ourChat) s.Require().NoError(err) inputMessage1 := buildTestMessage(*theirChat) sendResponse, err := theirMessenger.SendChatMessage(context.Background(), inputMessage1) s.NoError(err) s.Require().Len(sendResponse.Messages(), 1) inputMessage2 := buildTestMessage(*theirChat) sendResponse, err = theirMessenger.SendChatMessage(context.Background(), inputMessage2) s.NoError(err) s.Require().Len(sendResponse.Messages(), 1) messageID := sendResponse.Messages()[0].ID response, err := WaitOnMessengerResponse( s.m, func(r *MessengerResponse) bool { return len(r.messages) > 0 }, "no messages", ) s.Require().NoError(err) s.Require().Len(response.Chats(), 1) s.Require().Len(response.Messages(), 2) ogMessage := sendResponse.Messages()[0] sendResponse, err = theirMessenger.DeleteMessageAndSend(context.Background(), ogMessage.ID) s.Require().NoError(err) s.Require().Len(sendResponse.Messages(), 0) s.Require().Len(sendResponse.RemovedMessages(), 1) s.Require().Equal(messageID, sendResponse.RemovedMessages()[0].MessageID) s.Require().Len(sendResponse.Chats(), 1) // LastMessage is updated and marked as deleted s.Require().NotNil(sendResponse.Chats()[0].LastMessage) s.Require().Equal(inputMessage2.ID, sendResponse.Chats()[0].LastMessage.ID) s.Require().Equal(sendResponse.Chats()[0].LastMessage.Deleted, true) } func (s *MessengerDeleteMessageSuite) TestDeleteWrongMessageType() { theirMessenger := s.newMessenger() _, err := theirMessenger.Start() s.Require().NoError(err) theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.transport) err = theirMessenger.SaveChat(theirChat) s.Require().NoError(err) ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.transport) err = s.m.SaveChat(ourChat) s.Require().NoError(err) inputMessage := buildTestGapMessage(*theirChat) sendResponse, err := theirMessenger.SendChatMessage(context.Background(), inputMessage) s.NoError(err) s.Require().Len(sendResponse.Messages(), 1) ogMessage := sendResponse.Messages()[0] // Delete should not work _, err = theirMessenger.DeleteMessageAndSend(context.Background(), ogMessage.ID) s.Require().Equal(ErrInvalidDeleteTypeAuthor, err) } // TODO fix activity center notifications not being deleted when a message is deleted func (s *MessengerDeleteMessageSuite) TestDeleteMessageFirstThenMessage() { theirMessenger := s.newMessenger() _, err := theirMessenger.Start() s.Require().NoError(err) theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.transport) err = theirMessenger.SaveChat(theirChat) s.Require().NoError(err) contact, err := BuildContactFromPublicKey(&theirMessenger.identity.PublicKey) s.Require().NoError(err) ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.transport) err = s.m.SaveChat(ourChat) s.Require().NoError(err) messageID := "message-id" inputMessage := buildTestMessage(*theirChat) inputMessage.Clock = 1 deleteMessage := DeleteMessage{ DeleteMessage: protobuf.DeleteMessage{ Clock: 2, MessageType: protobuf.MessageType_ONE_TO_ONE, MessageId: messageID, ChatId: theirChat.ID, }, From: common.PubkeyToHex(&theirMessenger.identity.PublicKey), } state := &ReceivedMessageState{ Response: &MessengerResponse{}, } // Handle Delete first err = s.m.HandleDeleteMessage(state, deleteMessage) s.Require().NoError(err) // // Handle chat message state = &ReceivedMessageState{ Response: &MessengerResponse{}, CurrentMessageState: &CurrentMessageState{ Message: inputMessage.ChatMessage, MessageID: messageID, WhisperTimestamp: s.m.getTimesource().GetCurrentTime(), Contact: contact, PublicKey: &theirMessenger.identity.PublicKey, }, } err = s.m.HandleChatMessage(state) s.Require().NoError(err) s.Require().Len(state.Response.Messages(), 0) // Message should not be added to response s.Require().Len(state.Response.RemovedMessages(), 0) s.Require().Nil(state.Response.Chats()[0].LastMessage) } func (s *MessengerDeleteMessageSuite) TestDeleteImageMessage() { theirMessenger := s.newMessenger() _, err := theirMessenger.Start() s.Require().NoError(err) theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.transport) err = theirMessenger.SaveChat(theirChat) s.Require().NoError(err) ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.transport) err = s.m.SaveChat(ourChat) s.Require().NoError(err) messageCount := 3 var album []*common.Message for i := 0; i < messageCount; i++ { image, err := buildImageWithoutAlbumIDMessage(*ourChat) s.NoError(err) album = append(album, image) } response, err := s.m.SendChatMessages(context.Background(), album) s.NoError(err) // Check that album count was the number of the images sent imagesCount := uint32(0) for _, message := range response.Messages() { if message.ContentType == protobuf.ChatMessage_IMAGE { imagesCount++ } } for _, message := range response.Messages() { s.Require().NotNil(message.GetImage()) s.Require().Equal(message.GetImage().AlbumImagesCount, imagesCount) } s.Require().Equal(messageCount, len(response.Messages()), "it returns the messages") s.Require().NoError(err) s.Require().Len(response.Messages(), messageCount) response, err = WaitOnMessengerResponse( theirMessenger, func(r *MessengerResponse) bool { return len(r.messages) == messageCount }, "no messages", ) s.Require().NoError(err) s.Require().Len(response.Chats(), 1) s.Require().Len(response.Messages(), messageCount) for _, message := range response.Messages() { image := message.GetImage() s.Require().NotNil(image, "Message.ID=%s", message.ID) s.Require().Equal(image.AlbumImagesCount, imagesCount) s.Require().NotEmpty(image.AlbumId, "Message.ID=%s", message.ID) } firstMessageID := response.Messages()[0].ID sendResponse, err := s.m.DeleteMessageAndSend(context.Background(), firstMessageID) s.Require().NoError(err) s.Require().Len(sendResponse.Messages(), 0) s.Require().Len(sendResponse.RemovedMessages(), 3) s.Require().Equal(sendResponse.RemovedMessages()[0].DeletedBy, "") s.Require().Len(sendResponse.Chats(), 1) // LastMessage marked as deleted s.Require().Equal(sendResponse.Chats()[0].LastMessage.ID, album[2].ID) s.Require().Equal(sendResponse.Chats()[0].LastMessage.Deleted, true) // Main instance user attempts to delete the message it received from theirMessenger _, err = theirMessenger.DeleteMessageAndSend(context.Background(), firstMessageID) s.Require().ErrorContains(err, "can't find chat") } func (s *MessengerDeleteMessageSuite) TestDeleteImageMessageFirstThenMessage() { theirMessenger := s.newMessenger() _, err := theirMessenger.Start() s.Require().NoError(err) theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.transport) err = theirMessenger.SaveChat(theirChat) s.Require().NoError(err) contact, err := BuildContactFromPublicKey(&theirMessenger.identity.PublicKey) s.Require().NoError(err) ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.transport) err = s.m.SaveChat(ourChat) s.Require().NoError(err) messageID1 := "message-id1" messageID2 := "message-id2" messageCount := 2 var album []*common.Message for i := 0; i < messageCount; i++ { image, err := buildImageWithoutAlbumIDMessage(*ourChat) image.Clock = 1 s.NoError(err) album = append(album, image) } deleteMessage := DeleteMessage{ DeleteMessage: protobuf.DeleteMessage{ Clock: 2, MessageType: protobuf.MessageType_ONE_TO_ONE, MessageId: messageID1, ChatId: theirChat.ID, }, From: common.PubkeyToHex(&theirMessenger.identity.PublicKey), } state := &ReceivedMessageState{ Response: &MessengerResponse{}, } // Handle Delete first err = s.m.HandleDeleteMessage(state, deleteMessage) s.Require().NoError(err) // Handle first image message state = &ReceivedMessageState{ Response: &MessengerResponse{}, CurrentMessageState: &CurrentMessageState{ Message: album[0].ChatMessage, MessageID: messageID1, WhisperTimestamp: s.m.getTimesource().GetCurrentTime(), Contact: contact, PublicKey: &theirMessenger.identity.PublicKey, }, } err = s.m.HandleChatMessage(state) s.Require().NoError(err) s.Require().Len(state.Response.Messages(), 0) // Message should not be added to response s.Require().Len(state.Response.RemovedMessages(), 0) s.Require().Nil(state.Response.Chats()[0].LastMessage) // Handle second image message state = &ReceivedMessageState{ Response: &MessengerResponse{}, CurrentMessageState: &CurrentMessageState{ Message: album[1].ChatMessage, MessageID: messageID2, WhisperTimestamp: s.m.getTimesource().GetCurrentTime(), Contact: contact, PublicKey: &theirMessenger.identity.PublicKey, }, } err = s.m.HandleChatMessage(state) s.Require().NoError(err) s.Require().Len(state.Response.Messages(), 0) // Message should not be added to response even if we didn't delete that ID s.Require().Len(state.Response.RemovedMessages(), 0) s.Require().Nil(state.Response.Chats()[0].LastMessage) } func (s *MessengerDeleteMessageSuite) TestDeleteMessageWithAMention() { theirMessenger := s.newMessenger() _, err := theirMessenger.Start() s.Require().NoError(err) theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.transport) err = theirMessenger.SaveChat(theirChat) s.Require().NoError(err) ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.transport) err = s.m.SaveChat(ourChat) s.Require().NoError(err) inputMessage := buildTestMessage(*theirChat) inputMessage.Text = "text with a mention @" + common.PubkeyToHex(&s.privateKey.PublicKey) sendResponse, err := theirMessenger.SendChatMessage(context.Background(), inputMessage) s.NoError(err) s.Require().Len(sendResponse.Messages(), 1) messageID := sendResponse.Messages()[0].ID response, err := WaitOnMessengerResponse( s.m, func(r *MessengerResponse) bool { return len(r.messages) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(response.Chats(), 1) s.Require().Len(response.Messages(), 1) // Receiver (us) is mentioned s.Require().Equal(int(response.Chats()[0].UnviewedMessagesCount), 1) s.Require().Equal(int(response.Chats()[0].UnviewedMentionsCount), 1) s.Require().True(response.Messages()[0].Mentioned) deleteMessage := DeleteMessage{ DeleteMessage: protobuf.DeleteMessage{ Clock: 2, MessageType: protobuf.MessageType_ONE_TO_ONE, MessageId: messageID, ChatId: theirChat.ID, }, From: common.PubkeyToHex(&theirMessenger.identity.PublicKey), } state := &ReceivedMessageState{ Response: &MessengerResponse{}, } // Handle Delete first err = s.m.HandleDeleteMessage(state, deleteMessage) s.Require().NoError(err) s.Require().Len(response.Chats(), 1) s.Require().Len(response.Messages(), 1) // Receiver (us) is no longer mentioned s.Require().Equal(int(response.Chats()[0].UnviewedMessagesCount), 0) s.Require().Equal(int(response.Chats()[0].UnviewedMentionsCount), 0) }