mirror of
https://github.com/status-im/status-go.git
synced 2026-08-31 00:51:12 +00:00
Igor Sirotin
f3e4363808
Part of the Go project layout migration, item 27. Pure move plus import-path rewrite across 502 files. No API or behaviour change. `internal/` keeps the messaging application logic unimportable from outside the module, which is what the issue asks for -- status-go is consumed through the C-bindings in mobile/, not as a Go library. Things that had to follow the move, beyond the Go imports: - tools/generate-handlers/template.txt. messenger_handlers.go is generated, and the template hard-codes the imports it emits, so the generated file kept importing protocol/common and failed typecheck. - .gitignore. The ignore rule for that generated file was pinned to the old path; without moving it, a 1486-line generated file starts being tracked. - Makefile: the logosstorage and torrent test targets (both the archive packages and ./protocol itself), the archive README, migration-protocol. - scripts/run_unit_tests.sh, which names the protocol package explicitly to shard its tests. - scripts/cleanup_generated_files.sh and .golangci.yml. scripts/migration_check.sh also needed a fix that is not specific to this move: it validated every file the branch touched under a migration dir against the timestamp naming rule, and a directory rename makes every migration in it look newly added. It now excludes renames, so moving a migration is not mistaken for adding one. refs #7067
682 lines
21 KiB
Go
682 lines
21 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/status-im/status-go/internal/crypto"
|
|
"github.com/status-im/status-go/internal/crypto/types"
|
|
"github.com/status-im/status-go/internal/protocol/common"
|
|
"github.com/status-im/status-go/internal/protocol/contacts"
|
|
"github.com/status-im/status-go/internal/protocol/protobuf"
|
|
"github.com/status-im/status-go/internal/protocol/requests"
|
|
)
|
|
|
|
func TestMessengerEditMessageSuite(t *testing.T) {
|
|
suite.Run(t, new(MessengerEditMessageSuite))
|
|
}
|
|
|
|
type MessengerEditMessageSuite struct {
|
|
MessengerBaseTestSuite
|
|
}
|
|
|
|
func (s *MessengerEditMessageSuite) TestEditMessage() {
|
|
theirMessenger := s.newMessenger()
|
|
|
|
theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.getTimesource())
|
|
err := theirMessenger.SaveChat(theirChat)
|
|
s.Require().NoError(err)
|
|
|
|
ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.getTimesource())
|
|
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)
|
|
|
|
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]
|
|
|
|
messageID, err := types.DecodeHex(ogMessage.ID)
|
|
s.Require().NoError(err)
|
|
|
|
editedText := "edited text"
|
|
editedMessage := &requests.EditMessage{
|
|
ID: messageID,
|
|
Text: editedText,
|
|
}
|
|
|
|
sendResponse, err = theirMessenger.EditMessage(context.Background(), editedMessage)
|
|
|
|
s.Require().NoError(err)
|
|
s.Require().Len(sendResponse.Messages(), 1)
|
|
s.Require().NotEmpty(sendResponse.Messages()[0].EditedAt)
|
|
s.Require().Equal(sendResponse.Messages()[0].Text, editedText)
|
|
s.Require().Len(sendResponse.Chats(), 1)
|
|
s.Require().NotNil(sendResponse.Chats()[0].LastMessage)
|
|
s.Require().NotEmpty(sendResponse.Chats()[0].LastMessage.EditedAt)
|
|
|
|
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)
|
|
s.Require().NotEmpty(response.Messages()[0].EditedAt)
|
|
s.Require().False(response.Messages()[0].New)
|
|
|
|
// Main instance user attempts to edit the message it received from theirMessenger
|
|
editedMessage = &requests.EditMessage{
|
|
ID: messageID,
|
|
Text: "edited-again text",
|
|
}
|
|
_, err = s.m.EditMessage(context.Background(), editedMessage)
|
|
|
|
s.Require().Equal(ErrInvalidEditOrDeleteAuthor, err)
|
|
}
|
|
|
|
func (s *MessengerEditMessageSuite) TestEditImageMessage() {
|
|
theirMessenger := s.newMessenger()
|
|
|
|
theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.getTimesource())
|
|
err := theirMessenger.SaveChat(theirChat)
|
|
s.Require().NoError(err)
|
|
|
|
ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.getTimesource())
|
|
err = s.m.SaveChat(ourChat)
|
|
s.Require().NoError(err)
|
|
|
|
const messageCount = 1
|
|
var album []*common.Message
|
|
|
|
for i := 0; i < messageCount; i++ {
|
|
image, err := buildImageWithoutAlbumIDMessage(*ourChat)
|
|
s.NoError(err)
|
|
image.Text = "my message"
|
|
album = append(album, image)
|
|
}
|
|
|
|
sendResponse, err := s.m.SendChatMessages(context.Background(), album)
|
|
s.NoError(err)
|
|
s.Require().Len(sendResponse.Messages(), messageCount)
|
|
|
|
ogMessage := sendResponse.Messages()[0]
|
|
|
|
response, err := WaitOnMessengerResponse(
|
|
theirMessenger,
|
|
func(r *MessengerResponse) bool {
|
|
if len(r.messages) == 0 {
|
|
return false
|
|
}
|
|
_, ok := r.messages[ogMessage.ID]
|
|
return ok
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(response.Chats(), 1)
|
|
s.Require().Len(response.Messages(), 1)
|
|
|
|
messageID, err := types.DecodeHex(ogMessage.ID)
|
|
s.Require().NoError(err)
|
|
|
|
editedText := "edited text"
|
|
editedMessage := &requests.EditMessage{
|
|
ID: messageID,
|
|
Text: editedText,
|
|
}
|
|
|
|
sendResponse, err = s.m.EditMessage(context.Background(), editedMessage)
|
|
|
|
s.Require().NoError(err)
|
|
s.Require().Len(sendResponse.Messages(), 1)
|
|
s.Require().Equal(editedText, sendResponse.Messages()[0].Text)
|
|
s.Require().Equal(protobuf.ChatMessage_IMAGE, sendResponse.Messages()[0].ContentType)
|
|
|
|
response, err = WaitOnMessengerResponse(
|
|
theirMessenger,
|
|
func(r *MessengerResponse) bool {
|
|
if len(r.messages) == 0 {
|
|
return false
|
|
}
|
|
_, ok := r.messages[sendResponse.Messages()[0].ID]
|
|
return ok
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(response.Messages(), 1)
|
|
s.Require().NotEmpty(response.Messages()[0].EditedAt)
|
|
s.Require().False(response.Messages()[0].New)
|
|
s.Require().Equal(protobuf.ChatMessage_IMAGE, response.Messages()[0].ContentType)
|
|
|
|
// Check DB to make sure the message is still an image
|
|
dbMessage, err := s.m.persistence.MessageByID(ogMessage.ID)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(protobuf.ChatMessage_IMAGE, dbMessage.ContentType)
|
|
}
|
|
|
|
func (s *MessengerEditMessageSuite) TestEditBridgeMessage() {
|
|
theirMessenger := s.newMessenger()
|
|
|
|
theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.getTimesource())
|
|
err := theirMessenger.SaveChat(theirChat)
|
|
s.Require().NoError(err)
|
|
|
|
ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.getTimesource())
|
|
err = s.m.SaveChat(ourChat)
|
|
s.Require().NoError(err)
|
|
|
|
bridgeMessage := buildTestMessage(*theirChat)
|
|
bridgeMessage.ContentType = protobuf.ChatMessage_BRIDGE_MESSAGE
|
|
bridgeMessage.Payload = &protobuf.ChatMessage_BridgeMessage{
|
|
BridgeMessage: &protobuf.BridgeMessage{
|
|
BridgeName: "discord",
|
|
UserName: "user1",
|
|
UserAvatar: "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAADIAAAAyCAIAAACRXR/mAAAAjklEQVR4nOzXwQmFMBAAUZXUYh32ZB32ZB02sxYQQSZGsod55/91WFgSS0RM+SyjA56ZRZhFmEWYRRT6h+M6G16zrxv6fdJpmUWYRbxsYr13dKfanpN0WmYRZhGzXz6AWYRZRIfbaX26fT9Jk07LLMIsosPt9I/dTDotswizCG+nhFmEWYRZhFnEHQAA///z1CFkYamgfQAAAABJRU5ErkJggg==",
|
|
UserID: "123",
|
|
Content: "text1",
|
|
MessageID: "456",
|
|
ParentMessageID: "789",
|
|
},
|
|
}
|
|
|
|
sendResponse, err := theirMessenger.SendChatMessage(context.Background(), bridgeMessage)
|
|
s.NoError(err)
|
|
s.Require().Len(sendResponse.Messages(), 1)
|
|
|
|
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)
|
|
|
|
messageToEdit := sendResponse.Messages()[0]
|
|
|
|
messageID, err := types.DecodeHex(messageToEdit.ID)
|
|
s.Require().NoError(err)
|
|
|
|
editedText := "edited text"
|
|
editedMessage := &requests.EditMessage{
|
|
ID: messageID,
|
|
Text: editedText,
|
|
}
|
|
|
|
sendResponse, err = theirMessenger.EditMessage(context.Background(), editedMessage)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(sendResponse.Messages(), 1)
|
|
s.Require().NotEmpty(sendResponse.Messages()[0].EditedAt)
|
|
s.Require().Equal(sendResponse.Messages()[0].Text, "text-input-message")
|
|
s.Require().Equal(sendResponse.Messages()[0].GetBridgeMessage().Content, editedText)
|
|
s.Require().Len(sendResponse.Chats(), 1)
|
|
s.Require().NotNil(sendResponse.Chats()[0].LastMessage)
|
|
s.Require().NotEmpty(sendResponse.Chats()[0].LastMessage.EditedAt)
|
|
|
|
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)
|
|
|
|
s.Require().NotEmpty(response.Chats()[0].LastMessage.EditedAt)
|
|
s.Require().Equal(response.Messages()[0].GetBridgeMessage().Content, "edited text")
|
|
}
|
|
|
|
func (s *MessengerEditMessageSuite) TestEditMessageEdgeCases() {
|
|
theirMessenger := s.newMessenger()
|
|
|
|
theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.getTimesource())
|
|
err := theirMessenger.SaveChat(theirChat)
|
|
s.Require().NoError(err)
|
|
|
|
ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.getTimesource())
|
|
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)
|
|
|
|
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)
|
|
|
|
chat := response.Chats()[0]
|
|
editedMessage := sendResponse.Messages()[0]
|
|
|
|
newContactKey, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
wrongContact, err := contacts.BuildContactFromPublicKey(&newContactKey.PublicKey)
|
|
s.Require().NoError(err)
|
|
|
|
editMessage := EditMessage{
|
|
EditMessage: &protobuf.EditMessage{
|
|
Clock: editedMessage.Clock + 1,
|
|
Text: "some text",
|
|
MessageId: editedMessage.ID,
|
|
ChatId: chat.ID,
|
|
},
|
|
From: wrongContact.ID,
|
|
}
|
|
|
|
state := &ReceivedMessageState{
|
|
Response: &MessengerResponse{},
|
|
AllChats: &chatMap{},
|
|
}
|
|
state.AllChats.Store(ourChat.ID, ourChat)
|
|
|
|
err = s.m.handleEditMessage(state, editMessage)
|
|
// It should error as the user can't edit this message
|
|
s.Require().Error(err)
|
|
|
|
// Edit with a newer clock value
|
|
|
|
contact, err := contacts.BuildContactFromPublicKey(&theirMessenger.identity.PublicKey)
|
|
s.Require().NoError(err)
|
|
|
|
editMessage = EditMessage{
|
|
EditMessage: &protobuf.EditMessage{
|
|
Clock: editedMessage.Clock + 2,
|
|
Text: "some text",
|
|
MessageType: protobuf.MessageType_ONE_TO_ONE,
|
|
MessageId: editedMessage.ID,
|
|
ChatId: chat.ID,
|
|
},
|
|
From: contact.ID,
|
|
}
|
|
|
|
err = s.m.handleEditMessage(state, editMessage)
|
|
s.Require().NoError(err)
|
|
// It save the edit
|
|
s.Require().Len(state.Response.Messages(), 1)
|
|
s.Require().Len(state.Response.Chats(), 1)
|
|
s.Require().NotNil(state.Response.Chats()[0].LastMessage)
|
|
s.Require().NotEmpty(state.Response.Chats()[0].LastMessage.EditedAt)
|
|
|
|
editedMessage = state.Response.Messages()[0]
|
|
|
|
// In-between edit
|
|
editMessage = EditMessage{
|
|
EditMessage: &protobuf.EditMessage{
|
|
Clock: editedMessage.Clock + 1,
|
|
Text: "some other text",
|
|
MessageType: protobuf.MessageType_ONE_TO_ONE,
|
|
MessageId: editedMessage.ID,
|
|
ChatId: chat.ID,
|
|
},
|
|
From: contact.ID,
|
|
}
|
|
|
|
state.Response = &MessengerResponse{}
|
|
|
|
err = s.m.handleEditMessage(state, editMessage)
|
|
// It should error as the user can't edit this message
|
|
s.Require().NoError(err)
|
|
// It discards the edit
|
|
s.Require().Len(state.Response.Messages(), 0)
|
|
}
|
|
|
|
func (s *MessengerEditMessageSuite) TestEditMessageFirstEditsThenMessage() {
|
|
theirMessenger := s.newMessenger()
|
|
|
|
theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.getTimesource())
|
|
err := theirMessenger.SaveChat(theirChat)
|
|
s.Require().NoError(err)
|
|
|
|
contact, err := contacts.BuildContactFromPublicKey(&theirMessenger.identity.PublicKey)
|
|
s.Require().NoError(err)
|
|
|
|
ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.getTimesource())
|
|
err = s.m.SaveChat(ourChat)
|
|
s.Require().NoError(err)
|
|
messageID := "message-id"
|
|
|
|
inputMessage := buildTestMessage(*theirChat)
|
|
inputMessage.Clock = 1
|
|
editMessage := EditMessage{
|
|
EditMessage: &protobuf.EditMessage{
|
|
Clock: 2,
|
|
Text: "some text",
|
|
MessageType: protobuf.MessageType_ONE_TO_ONE,
|
|
MessageId: messageID,
|
|
ChatId: theirChat.ID,
|
|
},
|
|
From: crypto.PubkeyToHex(&theirMessenger.identity.PublicKey),
|
|
}
|
|
state := &ReceivedMessageState{
|
|
Response: &MessengerResponse{},
|
|
}
|
|
|
|
// Handle edit first
|
|
err = s.m.handleEditMessage(state, editMessage)
|
|
s.Require().NoError(err)
|
|
|
|
// Handle chat message
|
|
response := &MessengerResponse{}
|
|
state = &ReceivedMessageState{
|
|
Response: response,
|
|
CurrentMessageState: &CurrentMessageState{
|
|
MessageID: messageID,
|
|
WhisperTimestamp: s.m.getTimesource().GetCurrentTime(),
|
|
Contact: contact,
|
|
PublicKey: &theirMessenger.identity.PublicKey,
|
|
},
|
|
}
|
|
err = s.m.HandleChatMessage(context.Background(), state, inputMessage.ChatMessage, nil, false)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(response.Messages(), 1)
|
|
|
|
editedMessage := response.Messages()[0]
|
|
|
|
s.Require().Equal(uint64(2), editedMessage.EditedAt)
|
|
}
|
|
|
|
// Test editing a message on an existing private group chat
|
|
func (s *MessengerEditMessageSuite) TestEditGroupChatMessage() {
|
|
theirMessenger := s.newMessenger()
|
|
|
|
response, err := s.m.CreateGroupChatWithMembers(context.Background(), "id", []string{})
|
|
s.NoError(err)
|
|
s.Require().Len(response.Chats(), 1)
|
|
|
|
ourChat := response.Chats()[0]
|
|
|
|
err = s.m.SaveChat(ourChat)
|
|
s.NoError(err)
|
|
|
|
s.Require().NoError(makeMutualContact(s.m, &theirMessenger.identity.PublicKey))
|
|
|
|
members := []string{crypto.PubkeyToHex(&theirMessenger.identity.PublicKey)}
|
|
_, err = s.m.AddMembersToGroupChat(context.Background(), ourChat.ID, members)
|
|
s.NoError(err)
|
|
|
|
// Retrieve their messages so that the chat is created
|
|
response, err = WaitOnMessengerResponse(
|
|
theirMessenger,
|
|
func(r *MessengerResponse) bool { return len(r.Chats()) > 0 },
|
|
"chat invitation not received",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(response.Chats(), 1)
|
|
s.Require().Len(response.ActivityCenterNotifications(), 1)
|
|
s.Require().False(response.Chats()[0].Active)
|
|
|
|
_, err = theirMessenger.ConfirmJoiningGroup(context.Background(), ourChat.ID)
|
|
s.NoError(err)
|
|
|
|
// Wait for the message to reach its destination
|
|
_, err = WaitOnMessengerResponse(
|
|
s.m,
|
|
func(r *MessengerResponse) bool { return len(r.Chats()) > 0 },
|
|
"no joining group event received",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
inputMessage := buildTestMessage(*ourChat)
|
|
|
|
sendResponse, err := theirMessenger.SendChatMessage(context.Background(), inputMessage)
|
|
s.NoError(err)
|
|
s.Require().Len(sendResponse.Messages(), 1)
|
|
|
|
sentMessage := sendResponse.Messages()[0]
|
|
|
|
_, err = WaitOnMessengerResponse(
|
|
s.m,
|
|
func(r *MessengerResponse) bool { return len(r.Messages()) > 0 },
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
// Edit message
|
|
|
|
messageID, err := types.DecodeHex(sentMessage.ID)
|
|
s.Require().NoError(err)
|
|
|
|
editedText := "edited text"
|
|
editedMessage := &requests.EditMessage{
|
|
ID: messageID,
|
|
Text: editedText,
|
|
}
|
|
|
|
_, err = theirMessenger.EditMessage(context.Background(), editedMessage)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
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)
|
|
s.Require().NotEmpty(response.Messages()[0].EditedAt)
|
|
s.Require().False(response.Messages()[0].New)
|
|
}
|
|
|
|
func (s *MessengerEditMessageSuite) TestEditMessageWithMention() {
|
|
theirMessenger := s.newMessenger()
|
|
|
|
theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.getTimesource())
|
|
err := theirMessenger.SaveChat(theirChat)
|
|
s.Require().NoError(err)
|
|
|
|
ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.getTimesource())
|
|
err = s.m.SaveChat(ourChat)
|
|
s.Require().NoError(err)
|
|
|
|
inputMessage := buildTestMessage(*theirChat)
|
|
// Send first message with no mention
|
|
sendResponse, err := theirMessenger.SendChatMessage(context.Background(), inputMessage)
|
|
s.NoError(err)
|
|
s.Require().Len(sendResponse.Messages(), 1)
|
|
|
|
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)
|
|
// Make sure the message is not marked as Mentioned (chat still counts it because it's 1-1)
|
|
s.Require().False(response.Messages()[0].Mentioned)
|
|
s.Require().Equal(int(response.Chats()[0].UnviewedMessagesCount), 1)
|
|
s.Require().Equal(int(response.Chats()[0].UnviewedMentionsCount), 1)
|
|
|
|
ogMessage := sendResponse.Messages()[0]
|
|
|
|
messageID, err := types.DecodeHex(ogMessage.ID)
|
|
s.Require().NoError(err)
|
|
|
|
// Edit the message and add a mention
|
|
editedText := "edited text @" + crypto.PubkeyToHex(&s.privateKey.PublicKey)
|
|
editedMessage := &requests.EditMessage{
|
|
ID: messageID,
|
|
Text: editedText,
|
|
}
|
|
|
|
sendResponse, err = theirMessenger.EditMessage(context.Background(), editedMessage)
|
|
|
|
s.Require().NoError(err)
|
|
s.Require().Len(sendResponse.Messages(), 1)
|
|
s.Require().NotEmpty(sendResponse.Messages()[0].EditedAt)
|
|
s.Require().Equal(sendResponse.Messages()[0].Text, editedText)
|
|
s.Require().Len(sendResponse.Chats(), 1)
|
|
s.Require().NotNil(sendResponse.Chats()[0].LastMessage)
|
|
s.Require().NotEmpty(sendResponse.Chats()[0].LastMessage.EditedAt)
|
|
s.Require().False(sendResponse.Messages()[0].Mentioned) // Sender is still not mentioned
|
|
|
|
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)
|
|
s.Require().NotEmpty(response.Messages()[0].EditedAt)
|
|
s.Require().False(response.Messages()[0].New)
|
|
// Receiver (us) is now mentioned
|
|
s.Require().True(response.Messages()[0].Mentioned)
|
|
s.Require().Equal(int(response.Chats()[0].UnviewedMessagesCount), 1)
|
|
s.Require().Equal(int(response.Chats()[0].UnviewedMentionsCount), 1)
|
|
|
|
// Edit the message again but remove the mention
|
|
editedText = "edited text no mention"
|
|
editedMessage = &requests.EditMessage{
|
|
ID: messageID,
|
|
Text: editedText,
|
|
}
|
|
|
|
sendResponse, err = theirMessenger.EditMessage(context.Background(), editedMessage)
|
|
|
|
s.Require().NoError(err)
|
|
s.Require().Len(sendResponse.Messages(), 1)
|
|
s.Require().NotEmpty(sendResponse.Messages()[0].EditedAt)
|
|
s.Require().Equal(sendResponse.Messages()[0].Text, editedText)
|
|
s.Require().Len(sendResponse.Chats(), 1)
|
|
s.Require().NotNil(sendResponse.Chats()[0].LastMessage)
|
|
s.Require().NotEmpty(sendResponse.Chats()[0].LastMessage.EditedAt)
|
|
s.Require().False(sendResponse.Messages()[0].Mentioned) // Sender is still not mentioned
|
|
|
|
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)
|
|
s.Require().NotEmpty(response.Messages()[0].EditedAt)
|
|
s.Require().False(response.Messages()[0].New)
|
|
// Receiver (us) is no longer mentioned
|
|
s.Require().False(response.Messages()[0].Mentioned)
|
|
s.Require().Equal(int(response.Chats()[0].UnviewedMessagesCount), 1) // We still have an unread message though
|
|
s.Require().Equal(int(response.Chats()[0].UnviewedMentionsCount), 1)
|
|
}
|
|
|
|
func (s *MessengerEditMessageSuite) TestEditMessageWithLinkPreviews() {
|
|
theirMessenger := s.newMessenger()
|
|
|
|
theirChat := CreateOneToOneChat("Their 1TO1", &s.privateKey.PublicKey, s.m.getTimesource())
|
|
err := theirMessenger.SaveChat(theirChat)
|
|
s.Require().NoError(err)
|
|
|
|
ourChat := CreateOneToOneChat("Our 1TO1", &theirMessenger.identity.PublicKey, s.m.getTimesource())
|
|
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)
|
|
|
|
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]
|
|
|
|
messageID, err := types.DecodeHex(ogMessage.ID)
|
|
s.Require().NoError(err)
|
|
|
|
contactPublicKey, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&contactPublicKey.PublicKey))
|
|
|
|
editedText := "edited text"
|
|
editedMessage := &requests.EditMessage{
|
|
ID: messageID,
|
|
Text: editedText,
|
|
LinkPreviews: []common.LinkPreview{
|
|
{
|
|
Type: protobuf.UnfurledLink_LINK,
|
|
Description: "GitHub is where people build software.",
|
|
Hostname: "github.com",
|
|
Title: "Build software better, together",
|
|
URL: "https://github.com",
|
|
Thumbnail: common.LinkPreviewThumbnail{
|
|
Width: 100,
|
|
Height: 200,
|
|
URL: "http://localhost:9999",
|
|
DataURI: "data:image/png;base64,iVBORw0KGgoAAAANSUg=",
|
|
}},
|
|
},
|
|
StatusLinkPreviews: []common.StatusLinkPreview{
|
|
{
|
|
URL: "https://status.app/u/TestUrl",
|
|
Contact: &common.StatusContactLinkPreview{
|
|
PublicKey: contactID,
|
|
DisplayName: "TestDisplayName",
|
|
Description: "Test description",
|
|
Icon: common.LinkPreviewThumbnail{
|
|
Width: 100,
|
|
Height: 200,
|
|
DataURI: "data:image/png;base64,iVBORw0KGgoAAAANSUg=",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
sendResponse, err = theirMessenger.EditMessage(context.Background(), editedMessage)
|
|
|
|
s.Require().NoError(err)
|
|
s.Require().Len(sendResponse.Messages(), 1)
|
|
s.Require().Len(sendResponse.Messages()[0].LinkPreviews, 1)
|
|
s.Require().NotNil(sendResponse.Messages()[0].UnfurledStatusLinks)
|
|
s.Require().Len(sendResponse.Messages()[0].UnfurledStatusLinks.UnfurledStatusLinks, 1)
|
|
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)
|
|
|
|
responseMessage := response.Messages()[0]
|
|
s.Require().NotEmpty(responseMessage.EditedAt)
|
|
s.Require().Len(responseMessage.UnfurledLinks, 1)
|
|
s.Require().NotNil(responseMessage.UnfurledStatusLinks)
|
|
s.Require().Len(responseMessage.UnfurledStatusLinks.UnfurledStatusLinks, 1)
|
|
s.Require().False(responseMessage.New)
|
|
}
|