1598 lines
59 KiB
Go
1598 lines
59 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/status-im/status-go/deprecation"
|
|
"github.com/status-im/status-go/eth-node/crypto"
|
|
"github.com/status-im/status-go/eth-node/types"
|
|
multiaccountscommon "github.com/status-im/status-go/multiaccounts/common"
|
|
"github.com/status-im/status-go/multiaccounts/settings"
|
|
"github.com/status-im/status-go/protocol/common"
|
|
"github.com/status-im/status-go/protocol/protobuf"
|
|
"github.com/status-im/status-go/protocol/requests"
|
|
v1protocol "github.com/status-im/status-go/protocol/v1"
|
|
)
|
|
|
|
func TestMessengerContactRequestSuite(t *testing.T) {
|
|
suite.Run(t, new(MessengerContactRequestSuite))
|
|
}
|
|
|
|
type MessengerContactRequestSuite struct {
|
|
MessengerBaseTestSuite
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) findFirstByContentType(messages []*common.Message, contentType protobuf.ChatMessage_ContentType) *common.Message {
|
|
return FindFirstByContentType(messages, contentType)
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) sendContactRequest(request *requests.SendContactRequest, messenger *Messenger) {
|
|
s.logger.Info("sendContactRequest", zap.String("sender", messenger.IdentityPublicKeyString()), zap.String("receiver", request.ID))
|
|
|
|
// Send contact request
|
|
resp, err := messenger.SendContactRequest(context.Background(), request)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
|
|
// Check CR and mutual state update messages
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
mutualStateUpdate := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_SYSTEM_MESSAGE_MUTUAL_EVENT_SENT)
|
|
s.Require().NotNil(mutualStateUpdate)
|
|
|
|
s.Require().NotNil(mutualStateUpdate.ID)
|
|
s.Require().Equal(mutualStateUpdate.From, messenger.myHexIdentity())
|
|
s.Require().Equal(mutualStateUpdate.ChatId, request.ID)
|
|
s.Require().Equal(mutualStateUpdate.Text, fmt.Sprintf(outgoingMutualStateEventSentDefaultText, request.ID))
|
|
|
|
contactRequest := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
s.Require().Equal(common.ContactRequestStatePending, contactRequest.ContactRequestState)
|
|
s.Require().Equal(request.Message, contactRequest.Text)
|
|
|
|
// Check pending notification
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(ActivityCenterNotificationTypeContactRequest, resp.ActivityCenterNotifications()[0].Type)
|
|
s.Require().Equal(contactRequest.ID, resp.ActivityCenterNotifications()[0].Message.ID)
|
|
s.Require().Equal(contactRequest.ContactRequestState, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, true)
|
|
|
|
// Check contacts
|
|
s.Require().Len(resp.Contacts, 1)
|
|
contact := resp.Contacts[0]
|
|
s.Require().False(contact.mutual())
|
|
|
|
// Make sure it's not returned as coming from us
|
|
contactRequests, _, err := messenger.PendingContactRequests("", 10)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(contactRequests, 0)
|
|
|
|
// Make sure contact is added on the sender side
|
|
contacts := messenger.AddedContacts()
|
|
s.Require().Len(contacts, 1)
|
|
s.Require().Equal(ContactRequestStateSent, contacts[0].ContactRequestLocalState)
|
|
s.Require().NotNil(contacts[0].DisplayName)
|
|
|
|
// Check contact's primary name matches notification's name
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Name, contacts[0].PrimaryName())
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) receiveContactRequest(messageText string, theirMessenger *Messenger) *common.Message {
|
|
s.logger.Info("receiveContactRequest", zap.String("receiver", theirMessenger.IdentityPublicKeyString()))
|
|
|
|
// Wait for the message to reach its destination
|
|
resp, err := WaitOnMessengerResponse(
|
|
theirMessenger,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) == 1 && len(r.Messages()) >= 2 && len(r.ActivityCenterNotifications()) == 1
|
|
},
|
|
"no messages",
|
|
)
|
|
|
|
// Check contact request has been received
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
|
|
// Check CR and mutual state update messages
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
contactRequest := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
s.Require().Equal(common.ContactRequestStatePending, contactRequest.ContactRequestState)
|
|
s.Require().Equal(messageText, contactRequest.Text)
|
|
|
|
mutualStateUpdate := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_SYSTEM_MESSAGE_MUTUAL_EVENT_SENT)
|
|
s.Require().NotNil(mutualStateUpdate)
|
|
|
|
s.Require().Equal(mutualStateUpdate.From, contactRequest.From)
|
|
s.Require().Equal(mutualStateUpdate.ChatId, contactRequest.From)
|
|
s.Require().Equal(mutualStateUpdate.Text, fmt.Sprintf(incomingMutualStateEventSentDefaultText, contactRequest.From))
|
|
|
|
// Check activity center notification is of the right type
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(ActivityCenterNotificationTypeContactRequest, resp.ActivityCenterNotifications()[0].Type)
|
|
s.Require().Equal(contactRequest.ID, resp.ActivityCenterNotifications()[0].Message.ID)
|
|
s.Require().Equal(contactRequest.ContactRequestState, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, false)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false)
|
|
|
|
notifications, err := theirMessenger.ActivityCenterNotifications(ActivityCenterNotificationsRequest{
|
|
Cursor: "",
|
|
Limit: 10,
|
|
ActivityTypes: []ActivityCenterType{ActivityCenterNotificationTypeContactRequest},
|
|
ReadType: ActivityCenterQueryParamsReadUnread,
|
|
},
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(notifications.Notifications, 1)
|
|
s.Require().Equal(contactRequest.ID, notifications.Notifications[0].Message.ID)
|
|
s.Require().Equal(contactRequest.ContactRequestState, notifications.Notifications[0].Message.ContactRequestState)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
contact := resp.Contacts[0]
|
|
s.Require().Equal(ContactRequestStateReceived, contact.ContactRequestRemoteState)
|
|
|
|
// Check contact's primary name matches notification's name
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Name, contact.PrimaryName())
|
|
|
|
// Make sure it's the latest pending contact requests
|
|
contactRequests, _, err := theirMessenger.PendingContactRequests("", 10)
|
|
s.Require().NoError(err)
|
|
s.Require().Greater(len(contactRequests), 0)
|
|
s.Require().Equal(contactRequests[0].ID, contactRequest.ID)
|
|
|
|
// Confirm latest pending contact request
|
|
resp, err = theirMessenger.GetLatestContactRequestForContact(contactRequest.From)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(resp.Messages(), 1)
|
|
s.Require().Equal(contactRequest.ID, resp.Messages()[0].ID)
|
|
s.Require().Equal(common.ContactRequestStatePending, resp.Messages()[0].ContactRequestState)
|
|
|
|
return contactRequest
|
|
}
|
|
|
|
// This function partially logs given MessengerResponse with description.
|
|
// This is helpful for testing response content during long tests.
|
|
// Logged contents: Messages, Contacts, ActivityCenterNotifications
|
|
func (s *MessengerContactRequestSuite) logResponse(response *MessengerResponse, description string) {
|
|
s.logger.Debug("MessengerResponse", zap.String("description", description))
|
|
|
|
for i, message := range response.Messages() {
|
|
s.logger.Debug("message",
|
|
zap.Int("index", i),
|
|
zap.String("Text", message.Text),
|
|
zap.Any("ContentType", message.ContentType),
|
|
)
|
|
}
|
|
|
|
for i, contact := range response.Contacts {
|
|
s.logger.Debug("contact",
|
|
zap.Int("index", i),
|
|
zap.Bool("Blocked", contact.Blocked),
|
|
zap.Bool("Removed", contact.Removed),
|
|
zap.Any("crRemoteState", contact.ContactRequestLocalState),
|
|
zap.Any("crLocalState", contact.ContactRequestRemoteState),
|
|
)
|
|
}
|
|
|
|
for i, notification := range response.ActivityCenterNotifications() {
|
|
messageText := ""
|
|
if notification.Message != nil {
|
|
messageText = notification.Message.Text
|
|
}
|
|
s.logger.Debug("acNotification",
|
|
zap.Int("index", i),
|
|
zap.Any("id", notification.ID),
|
|
zap.Any("Type", notification.Type),
|
|
zap.String("Message", messageText),
|
|
zap.String("Name", notification.Name),
|
|
zap.String("Author", notification.Author),
|
|
)
|
|
}
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) acceptContactRequest(
|
|
contactRequest *common.Message, sender *Messenger, receiver *Messenger) {
|
|
s.logger.Info("acceptContactRequest",
|
|
zap.String("sender", sender.IdentityPublicKeyString()),
|
|
zap.String("receiver", receiver.IdentityPublicKeyString()))
|
|
|
|
// Accept contact request, receiver side
|
|
resp, err := receiver.AcceptContactRequest(context.Background(), &requests.AcceptContactRequest{ID: types.Hex2Bytes(contactRequest.ID)})
|
|
s.Require().NoError(err)
|
|
|
|
// Chack updated contact request message and mutual state update
|
|
s.Require().NotNil(resp)
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
contactRequestMsg := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequestMsg)
|
|
|
|
mutualStateUpdate := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_SYSTEM_MESSAGE_MUTUAL_EVENT_ACCEPTED)
|
|
s.Require().NotNil(mutualStateUpdate)
|
|
|
|
s.Require().Equal(contactRequestMsg.ID, contactRequest.ID)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, contactRequestMsg.ContactRequestState)
|
|
|
|
s.Require().Equal(mutualStateUpdate.ChatId, contactRequestMsg.From)
|
|
s.Require().Equal(mutualStateUpdate.From, contactRequestMsg.ChatId)
|
|
s.Require().Equal(mutualStateUpdate.Text, fmt.Sprintf(outgoingMutualStateEventAcceptedDefaultText, contactRequestMsg.From))
|
|
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), contactRequest.ID)
|
|
s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
s.Require().True(resp.Contacts[0].mutual())
|
|
|
|
// Check contact's primary name matches notification's name
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Name, resp.Contacts[0].PrimaryName())
|
|
|
|
// Check we have active chat in the response
|
|
s.Require().Len(resp.Chats(), 1)
|
|
s.Require().True(resp.Chats()[0].Active)
|
|
|
|
// Make sure the sender is added to our contacts
|
|
contacts := receiver.AddedContacts()
|
|
s.Require().Len(contacts, 1)
|
|
|
|
// Make sure we consider them a mutual contact, receiver side
|
|
mutualContacts := receiver.MutualContacts()
|
|
s.Require().Len(mutualContacts, 1)
|
|
|
|
// Confirm latest pending contact request
|
|
resp, err = receiver.GetLatestContactRequestForContact(sender.IdentityPublicKeyString())
|
|
s.Require().NoError(err)
|
|
s.Require().Len(resp.Messages(), 1)
|
|
s.Require().Equal(contactRequest.ID, resp.Messages()[0].ID)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, resp.Messages()[0].ContactRequestState)
|
|
|
|
// Wait for the message to reach its destination
|
|
resp, err = WaitOnMessengerResponse(sender,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) == 1 && len(r.Messages()) == 2
|
|
},
|
|
"contact request acceptance not received",
|
|
)
|
|
s.logResponse(resp, "acceptContactRequest")
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
|
|
// Check activity center notification is of the right type
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(ActivityCenterNotificationTypeContactRequest, resp.ActivityCenterNotifications()[0].Type)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, true)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, true)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false)
|
|
s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message)
|
|
|
|
// Make sure the message is updated, sender side
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
contactRequestMsg = s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequestMsg)
|
|
|
|
mutualStateUpdate = s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_SYSTEM_MESSAGE_MUTUAL_EVENT_ACCEPTED)
|
|
s.Require().NotNil(mutualStateUpdate)
|
|
|
|
s.Require().Equal(contactRequest.ID, contactRequestMsg.ID)
|
|
s.Require().Equal(contactRequest.Text, contactRequestMsg.Text)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, contactRequestMsg.ContactRequestState)
|
|
|
|
s.Require().Equal(mutualStateUpdate.From, contactRequestMsg.ChatId)
|
|
s.Require().Equal(mutualStateUpdate.ChatId, contactRequestMsg.ChatId)
|
|
s.Require().Equal(mutualStateUpdate.Text, fmt.Sprintf(incomingMutualStateEventAcceptedDefaultText, contactRequestMsg.ChatId))
|
|
|
|
// Make sure we consider them a mutual contact, sender side
|
|
mutualContacts = s.m.MutualContacts()
|
|
s.Require().Len(mutualContacts, 1)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
contact := resp.Contacts[0]
|
|
s.Require().True(contact.mutual())
|
|
|
|
// Check contact's primary name matches notification's name
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Name, contact.PrimaryName())
|
|
|
|
// Sender's side chat should be active after the accepting the CR
|
|
chat, ok := s.m.allChats.Load(contact.ID)
|
|
s.Require().True(ok)
|
|
s.Require().NotNil(chat)
|
|
s.Require().True(chat.Active)
|
|
|
|
// Receiver's side chat should be also active after the accepting the CR
|
|
myID := types.EncodeHex(crypto.FromECDSAPub(&s.m.identity.PublicKey))
|
|
chat, ok = receiver.allChats.Load(myID)
|
|
s.Require().True(ok)
|
|
s.Require().NotNil(chat)
|
|
s.Require().True(chat.Active)
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) checkMutualContact(messenger *Messenger, contactPublicKey string) {
|
|
contacts := messenger.AddedContacts()
|
|
s.Require().Len(contacts, 1)
|
|
contact := contacts[0]
|
|
s.Require().Equal(contactPublicKey, contact.ID)
|
|
s.Require().True(contact.mutual())
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) createContactRequest(contactPublicKey string, messageText string) *requests.SendContactRequest {
|
|
return &requests.SendContactRequest{
|
|
ID: contactPublicKey,
|
|
Message: messageText,
|
|
}
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) declineContactRequest(contactRequest *common.Message, theirMessenger *Messenger) {
|
|
// Dismiss contact request, receiver side
|
|
resp, err := theirMessenger.DeclineContactRequest(context.Background(), &requests.DeclineContactRequest{ID: types.Hex2Bytes(contactRequest.ID)})
|
|
s.Require().NoError(err)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
s.Require().Equal(ContactRequestStateDismissed, resp.Contacts[0].ContactRequestLocalState)
|
|
|
|
// Make sure the message is updated
|
|
s.Require().NotNil(resp)
|
|
s.Require().Len(resp.Messages(), 1)
|
|
s.Require().Equal(resp.Messages()[0].ID, contactRequest.ID)
|
|
s.Require().Equal(common.ContactRequestStateDismissed, resp.Messages()[0].ContactRequestState)
|
|
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), contactRequest.ID)
|
|
s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, true)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, true)
|
|
s.Require().Equal(common.ContactRequestStateDismissed, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
|
|
// Check contact's primary name matches notification's name
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Name, resp.Contacts[0].PrimaryName())
|
|
|
|
// Make sure the sender is not added to our contacts
|
|
contacts := theirMessenger.AddedContacts()
|
|
s.Require().Len(contacts, 0)
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) retractContactRequest(contactID string, theirMessenger *Messenger) {
|
|
resp, err := s.m.RetractContactRequest(&requests.RetractContactRequest{ID: types.Hex2Bytes(contactID)})
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
s.Require().Len(resp.Contacts, 1)
|
|
s.Require().False(resp.Contacts[0].hasAddedUs())
|
|
s.Require().False(resp.Contacts[0].added())
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
s.Require().Equal(ContactRequestStateNone, resp.Contacts[0].ContactRequestLocalState)
|
|
s.Require().Equal(ContactRequestStateNone, resp.Contacts[0].ContactRequestRemoteState)
|
|
|
|
// Check outgoing mutual state message
|
|
s.Require().Len(resp.Messages(), 1)
|
|
mutualStateUpdate := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_SYSTEM_MESSAGE_MUTUAL_EVENT_REMOVED)
|
|
s.Require().NotNil(mutualStateUpdate)
|
|
|
|
myID := types.EncodeHex(crypto.FromECDSAPub(&s.m.identity.PublicKey))
|
|
s.Require().Equal(mutualStateUpdate.From, myID)
|
|
s.Require().Equal(mutualStateUpdate.ChatId, contactID)
|
|
s.Require().Equal(mutualStateUpdate.Text, fmt.Sprintf(outgoingMutualStateEventRemovedDefaultText, contactID))
|
|
|
|
// Wait for the message to reach its destination
|
|
resp, err = WaitOnMessengerResponse(
|
|
theirMessenger,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) > 0 && len(r.ActivityCenterNotifications()) == 1
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
s.Require().Len(resp.Contacts, 1)
|
|
|
|
s.Require().Equal(myID, resp.Contacts[0].ID)
|
|
|
|
s.Require().False(resp.Contacts[0].added())
|
|
s.Require().False(resp.Contacts[0].hasAddedUs())
|
|
s.Require().Equal(ContactRequestStateNone, resp.Contacts[0].ContactRequestLocalState)
|
|
s.Require().Equal(ContactRequestStateNone, resp.Contacts[0].ContactRequestRemoteState)
|
|
|
|
// Check pending notification
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(ActivityCenterNotificationTypeContactRemoved, resp.ActivityCenterNotifications()[0].Type)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, false)
|
|
|
|
// Check incoming mutual state message
|
|
s.Require().Len(resp.Messages(), 1)
|
|
mutualStateUpdate = s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_SYSTEM_MESSAGE_MUTUAL_EVENT_REMOVED)
|
|
s.Require().NotNil(mutualStateUpdate)
|
|
|
|
s.Require().Equal(mutualStateUpdate.From, myID)
|
|
s.Require().Equal(mutualStateUpdate.ChatId, myID)
|
|
s.Require().Equal(mutualStateUpdate.Text, fmt.Sprintf(incomingMutualStateEventRemovedDefaultText, myID))
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) syncInstallationContactV2FromContact(contact *Contact) protobuf.SyncInstallationContactV2 {
|
|
return protobuf.SyncInstallationContactV2{
|
|
LastUpdatedLocally: contact.LastUpdatedLocally,
|
|
LastUpdated: contact.LastUpdated,
|
|
Id: contact.ID,
|
|
DisplayName: contact.DisplayName,
|
|
EnsName: contact.EnsName,
|
|
LocalNickname: contact.LocalNickname,
|
|
Added: contact.added(),
|
|
Blocked: contact.Blocked,
|
|
Muted: false,
|
|
HasAddedUs: contact.hasAddedUs(),
|
|
Removed: contact.Removed,
|
|
ContactRequestLocalState: int64(contact.ContactRequestLocalState),
|
|
ContactRequestRemoteState: int64(contact.ContactRequestRemoteState),
|
|
ContactRequestRemoteClock: int64(contact.ContactRequestRemoteClock),
|
|
ContactRequestLocalClock: int64(contact.ContactRequestLocalClock),
|
|
VerificationStatus: int64(contact.VerificationStatus),
|
|
TrustStatus: int64(contact.TrustStatus),
|
|
}
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) TestReceiveAndAcceptContactRequest() { //nolint: unused
|
|
messageText := "hello!"
|
|
|
|
theirMessenger := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, theirMessenger)
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
request := &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, s.m)
|
|
contactRequest := s.receiveContactRequest(messageText, theirMessenger)
|
|
s.acceptContactRequest(contactRequest, s.m, theirMessenger)
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) TestReceiveAndDismissContactRequest() {
|
|
messageText := "hello!"
|
|
|
|
theirMessenger := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, theirMessenger)
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
request := &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, s.m)
|
|
contactRequest := s.receiveContactRequest(messageText, theirMessenger)
|
|
s.declineContactRequest(contactRequest, theirMessenger)
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) TestReceiveAcceptAndRetractContactRequest() { //nolint: unused
|
|
messageText := "hello!"
|
|
|
|
theirMessenger := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, theirMessenger)
|
|
|
|
s.Require().NoError(theirMessenger.settings.SaveSettingField(settings.MutualContactEnabled, true))
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
request := &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, s.m)
|
|
contactRequest := s.receiveContactRequest(messageText, theirMessenger)
|
|
s.acceptContactRequest(contactRequest, s.m, theirMessenger)
|
|
s.retractContactRequest(contactID, theirMessenger)
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1. Repeat 5 times:
|
|
// 2.1) Alice sends a contact request to Bob
|
|
// 2.2) Bob accepts the contact request
|
|
// 2.3) Alice removes bob from contacts
|
|
func (s *MessengerContactRequestSuite) TestAcceptCRRemoveAndRepeat() {
|
|
theirMessenger := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, theirMessenger)
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
|
|
for i := 0; i < 5; i++ {
|
|
messageText := fmt.Sprintf("hello %d", i)
|
|
request := &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, s.m)
|
|
contactRequest := s.receiveContactRequest(messageText, theirMessenger)
|
|
s.acceptContactRequest(contactRequest, s.m, theirMessenger)
|
|
s.retractContactRequest(contactID, theirMessenger)
|
|
}
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1) Alice sends a contact request to Bob
|
|
// 2) Bob declines the contact request
|
|
// 3) Alice fails to send a new contact request to Bob
|
|
func (s *MessengerContactRequestSuite) TestAliceTriesToSpamBobWithContactRequests() {
|
|
messageTextAlice := "You wanna play with fire, Bobby?!"
|
|
alice := s.m
|
|
|
|
bob := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob)
|
|
|
|
bobID := types.EncodeHex(crypto.FromECDSAPub(&bob.identity.PublicKey))
|
|
|
|
// Alice sends a contact request to Bob
|
|
request := &requests.SendContactRequest{
|
|
ID: bobID,
|
|
Message: messageTextAlice,
|
|
}
|
|
s.sendContactRequest(request, alice)
|
|
|
|
contactRequest := s.receiveContactRequest(messageTextAlice, bob)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
// Bob declines the contact request
|
|
s.declineContactRequest(contactRequest, bob)
|
|
|
|
// Alice sends a new contact request
|
|
resp, err := alice.SendContactRequest(context.Background(), request)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
|
|
// Check CR and mutual state update messages
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
contactRequest = s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
s.Require().Equal(common.ContactRequestStatePending, contactRequest.ContactRequestState)
|
|
s.Require().Equal(request.Message, contactRequest.Text)
|
|
|
|
// We should not receive a CR from a rejected contact
|
|
_, err = WaitOnMessengerResponse(
|
|
bob,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Messages()) > 0 &&
|
|
s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST) != nil
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().Error(err)
|
|
s.Require().ErrorContains(err, "no messages")
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1) Alice sends a contact request to Bob
|
|
// 2) Bob accepts the contact
|
|
// 3) Bob accepts the contact request (again!)
|
|
// 4) No extra mesages on Alice's side
|
|
func (s *MessengerContactRequestSuite) TestAliceSeesOnlyOneAcceptFromBob() {
|
|
messageTextAlice := "You wanna play with fire, Bobby?!"
|
|
alice := s.m
|
|
|
|
bob := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob)
|
|
|
|
bobID := types.EncodeHex(crypto.FromECDSAPub(&bob.identity.PublicKey))
|
|
|
|
// Alice sends a contact request to Bob
|
|
request := &requests.SendContactRequest{
|
|
ID: bobID,
|
|
Message: messageTextAlice,
|
|
}
|
|
s.sendContactRequest(request, alice)
|
|
|
|
contactRequest := s.receiveContactRequest(messageTextAlice, bob)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
// Bob accepts the contact request
|
|
s.acceptContactRequest(contactRequest, alice, bob)
|
|
|
|
// Accept contact request again
|
|
_, err := bob.AcceptContactRequest(context.Background(), &requests.AcceptContactRequest{ID: types.Hex2Bytes(contactRequest.ID)})
|
|
s.Require().NoError(err)
|
|
|
|
// Check we don't have extra messages on Alice's side
|
|
resp, err := WaitOnMessengerResponse(alice,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.ActivityCenterNotifications()) == 1 && len(r.Messages()) == 1
|
|
},
|
|
"contact request acceptance not received",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
|
|
// Check activity center notification is of the right type
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(ActivityCenterNotificationTypeContactRequest, resp.ActivityCenterNotifications()[0].Type)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, true)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, true)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false)
|
|
s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message)
|
|
|
|
// Make sure the message is updated, sender side
|
|
s.Require().Len(resp.Messages(), 1)
|
|
|
|
contactRequest = s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
s.Require().Equal(common.ContactRequestStateAccepted, contactRequest.ContactRequestState)
|
|
s.Require().Equal(request.Message, contactRequest.Text)
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) TestReceiveAndAcceptContactRequestTwice() { //nolint: unused
|
|
messageText := "hello!"
|
|
|
|
theirMessenger := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, theirMessenger)
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
request := &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, s.m)
|
|
contactRequest := s.receiveContactRequest(messageText, theirMessenger)
|
|
s.acceptContactRequest(contactRequest, s.m, theirMessenger)
|
|
|
|
// Resend contact request with higher clock value
|
|
resp, err := s.m.SendContactRequest(context.Background(), request)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
|
|
// Check CR and mutual state update messages
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
contactRequest = s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
s.Require().Equal(common.ContactRequestStateAccepted, contactRequest.ContactRequestState)
|
|
s.Require().Equal(request.Message, contactRequest.Text)
|
|
|
|
// We should not receive a CR from a mutual contact
|
|
_, err = WaitOnMessengerResponse(
|
|
theirMessenger,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Messages()) > 0 &&
|
|
s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST) != nil
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().Error(err)
|
|
s.Require().ErrorContains(err, "no messages")
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) TestAcceptLatestContactRequestForContact() {
|
|
messageText := "hello!"
|
|
|
|
theirMessenger := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, theirMessenger)
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
request := &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, s.m)
|
|
contactRequest := s.receiveContactRequest(messageText, theirMessenger)
|
|
|
|
// Accept latest contact request, receiver side
|
|
myID := types.EncodeHex(crypto.FromECDSAPub(&s.m.identity.PublicKey))
|
|
resp, err := theirMessenger.AcceptLatestContactRequestForContact(context.Background(), &requests.AcceptLatestContactRequestForContact{ID: types.Hex2Bytes(myID)})
|
|
s.Require().NoError(err)
|
|
|
|
// Make sure the message is updated
|
|
s.Require().NotNil(resp)
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
contactRequestMsg := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequestMsg)
|
|
|
|
mutualStateUpdate := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_SYSTEM_MESSAGE_MUTUAL_EVENT_ACCEPTED)
|
|
s.Require().NotNil(mutualStateUpdate)
|
|
|
|
s.Require().Equal(contactRequestMsg.ID, contactRequest.ID)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, contactRequestMsg.ContactRequestState)
|
|
|
|
s.Require().Equal(mutualStateUpdate.From, contactRequest.ChatId)
|
|
s.Require().Equal(mutualStateUpdate.ChatId, contactRequest.From)
|
|
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), contactRequest.ID)
|
|
s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
s.Require().True(resp.Contacts[0].mutual())
|
|
|
|
// Make sure the sender is added to our contacts
|
|
contacts := theirMessenger.AddedContacts()
|
|
s.Require().Len(contacts, 1)
|
|
|
|
// Make sure we consider them a mutual contact, receiver side
|
|
mutualContacts := theirMessenger.MutualContacts()
|
|
s.Require().Len(mutualContacts, 1)
|
|
|
|
// Wait for the message to reach its destination
|
|
resp, err = WaitOnMessengerResponse(
|
|
s.m,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) == 1 && len(r.Messages()) == 2 && len(r.ActivityCenterNotifications()) == 1
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
// Make sure the message is updated, sender side
|
|
s.Require().NotNil(resp)
|
|
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
contactRequestMsg = s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequestMsg)
|
|
|
|
mutualStateUpdate = s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_SYSTEM_MESSAGE_MUTUAL_EVENT_ACCEPTED)
|
|
s.Require().NotNil(mutualStateUpdate)
|
|
|
|
s.Require().Equal(common.ContactRequestStateAccepted, contactRequestMsg.ContactRequestState)
|
|
|
|
s.Require().Equal(mutualStateUpdate.From, contactRequest.ChatId)
|
|
s.Require().Equal(mutualStateUpdate.ChatId, contactRequest.ChatId)
|
|
|
|
// Check activity center notification is of the right type
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(ActivityCenterNotificationTypeContactRequest, resp.ActivityCenterNotifications()[0].Type)
|
|
s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
|
|
// Make sure we consider them a mutual contact, sender side
|
|
mutualContacts = s.m.MutualContacts()
|
|
s.Require().Len(mutualContacts, 1)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
s.Require().True(resp.Contacts[0].mutual())
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) TestDismissLatestContactRequestForContact() {
|
|
messageText := "hello!"
|
|
|
|
theirMessenger := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, theirMessenger)
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
request := &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, s.m)
|
|
contactRequest := s.receiveContactRequest(messageText, theirMessenger)
|
|
|
|
// Dismiss latest contact request, receiver side
|
|
myID := types.EncodeHex(crypto.FromECDSAPub(&s.m.identity.PublicKey))
|
|
resp, err := theirMessenger.DismissLatestContactRequestForContact(context.Background(), &requests.DismissLatestContactRequestForContact{ID: types.Hex2Bytes(myID)})
|
|
s.Require().NoError(err)
|
|
|
|
// Make sure the message is updated
|
|
s.Require().NotNil(resp)
|
|
s.Require().Len(resp.Messages(), 1)
|
|
s.Require().Equal(resp.Messages()[0].ID, contactRequest.ID)
|
|
s.Require().Equal(common.ContactRequestStateDismissed, resp.Messages()[0].ContactRequestState)
|
|
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), contactRequest.ID)
|
|
s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message)
|
|
s.Require().Equal(common.ContactRequestStateDismissed, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) TestPairedDevicesRemoveContact() {
|
|
messageText := "hello!"
|
|
|
|
alice1 := s.m
|
|
alice2, err := newMessengerWithKey(s.shh, s.m.identity, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
defer TearDownMessenger(&s.Suite, alice2)
|
|
|
|
prepAliceMessengersForPairing(&s.Suite, alice1, alice2)
|
|
|
|
PairDevices(&s.Suite, alice1, alice2)
|
|
PairDevices(&s.Suite, alice2, alice1)
|
|
|
|
bob := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob)
|
|
|
|
// Alice sends a contact request to bob
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&bob.identity.PublicKey))
|
|
request := &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, alice1)
|
|
contactRequest := s.receiveContactRequest(messageText, bob)
|
|
s.acceptContactRequest(contactRequest, alice1, bob)
|
|
|
|
// Wait for the message to reach its destination
|
|
resp, err := WaitOnMessengerResponse(
|
|
alice2,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) > 0
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
// Make sure we consider them a mutual contact, sender side
|
|
mutualContacts := alice2.MutualContacts()
|
|
s.Require().Len(mutualContacts, 1)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
s.Require().True(resp.Contacts[0].mutual())
|
|
|
|
s.retractContactRequest(contactID, bob)
|
|
|
|
// Check on alice2 side
|
|
resp, err = WaitOnMessengerResponse(
|
|
alice2,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) > 0
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
s.Require().Len(resp.Contacts, 1)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Equal(ContactRequestStateNone, resp.Contacts[0].ContactRequestLocalState)
|
|
s.Require().Equal(ContactRequestStateNone, resp.Contacts[0].ContactRequestRemoteState)
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1) Alice sends a contact request to Bob
|
|
// 2) Bob accepts the contact request
|
|
// 3) Alice restores state on a different device
|
|
// 4) Alice sends a contact request to bob
|
|
// Bob will need to help Alice recover her state, since as far as he can see
|
|
// that's an already accepted contact request
|
|
func (s *MessengerContactRequestSuite) TestAliceRecoverStateSendContactRequest() {
|
|
messageText := "hello!"
|
|
|
|
alice1 := s.m
|
|
|
|
bob := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob)
|
|
|
|
bobID := types.EncodeHex(crypto.FromECDSAPub(&bob.identity.PublicKey))
|
|
|
|
// Alice sends a contact request to bob
|
|
request := &requests.SendContactRequest{
|
|
ID: bobID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, alice1)
|
|
|
|
contactRequest := s.receiveContactRequest(messageText, bob)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
// Bob accepts the contact request
|
|
s.acceptContactRequest(contactRequest, alice1, bob)
|
|
|
|
// Alice resets her device
|
|
alice2, err := newMessengerWithKey(s.shh, s.m.identity, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
defer TearDownMessenger(&s.Suite, alice2)
|
|
|
|
// adds bob again to her device
|
|
s.sendContactRequest(request, alice2)
|
|
|
|
// Wait for the message to reach its destination
|
|
_, err = WaitOnMessengerResponse(
|
|
bob,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) > 0
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
// Bob should be a mutual contact with alice, nothing has changed
|
|
s.Require().Len(bob.MutualContacts(), 1)
|
|
|
|
// Alice retrieves her messages, she should have been notified by
|
|
// dear bobby that they were contacts
|
|
resp, err := WaitOnMessengerResponse(
|
|
alice2,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) > 0
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
s.Require().Len(resp.Contacts, 1)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().True(resp.Contacts[0].mutual())
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1) Alice sends a contact request to Bob
|
|
// 2) Bob accepts the contact request
|
|
// 3) Alice restores state on a different device
|
|
// 4) Bob sends a message to alice
|
|
// Alice will show a contact request from bob
|
|
func (s *MessengerContactRequestSuite) TestAliceRecoverStateReceiveContactRequest() {
|
|
messageText := "hello!"
|
|
|
|
alice1 := s.m
|
|
|
|
bob := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob)
|
|
|
|
bobID := types.EncodeHex(crypto.FromECDSAPub(&bob.identity.PublicKey))
|
|
|
|
// Alice sends a contact request to bob
|
|
request := &requests.SendContactRequest{
|
|
ID: bobID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, alice1)
|
|
|
|
contactRequest := s.receiveContactRequest(messageText, bob)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
// Bob accepts the contact request
|
|
s.acceptContactRequest(contactRequest, alice1, bob)
|
|
|
|
// Alice resets her device
|
|
alice2, err := newMessengerWithKey(s.shh, s.m.identity, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
defer TearDownMessenger(&s.Suite, alice2)
|
|
|
|
// We want to facilitate the discovery of the x3dh bundl here, since bob does not know about alice device
|
|
|
|
alice2Bundle, err := alice2.encryptor.GetBundle(alice2.identity)
|
|
s.Require().NoError(err)
|
|
|
|
_, err = bob.encryptor.ProcessPublicBundle(bob.identity, alice2Bundle)
|
|
s.Require().NoError(err)
|
|
|
|
// Bob sends a chat message to alice
|
|
|
|
var chat Chat
|
|
chats := bob.Chats()
|
|
for i, c := range chats {
|
|
if c.ID == alice1.myHexIdentity() && c.OneToOne() {
|
|
chat = *chats[i]
|
|
}
|
|
}
|
|
s.Require().NotNil(chat)
|
|
|
|
inputMessage := buildTestMessage(chat)
|
|
_, err = bob.SendChatMessage(context.Background(), inputMessage)
|
|
s.NoError(err)
|
|
|
|
// Alice retrieves the chat message, it should be
|
|
resp, err := WaitOnMessengerResponse(
|
|
alice2,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.ActivityCenterNotifications()) == 1
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
s.Require().Equal(ActivityCenterNotificationTypeContactRequest, resp.ActivityCenterNotifications()[0].Type)
|
|
s.Require().Len(resp.Contacts, 1)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Equal(ContactRequestStateNone, resp.Contacts[0].ContactRequestLocalState)
|
|
s.Require().Equal(ContactRequestStateReceived, resp.Contacts[0].ContactRequestRemoteState)
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1) Alice sends a contact request to Bob
|
|
// 2) Bob accepts the contact request
|
|
// 3) Bob goes offline
|
|
// 4) Alice retracts the contact request
|
|
// 5) Alice adds bob back to her contacts
|
|
// 6) Bob goes online, they receive 4 and 5 in the correct order
|
|
func (s *MessengerContactRequestSuite) TestAliceOfflineRetractsAndAddsCorrectOrder() {
|
|
messageText := "hello!"
|
|
|
|
alice1 := s.m
|
|
|
|
bob := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob)
|
|
|
|
bobID := types.EncodeHex(crypto.FromECDSAPub(&bob.identity.PublicKey))
|
|
|
|
// Alice sends a contact request to bob
|
|
request := &requests.SendContactRequest{
|
|
ID: bobID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, alice1)
|
|
|
|
contactRequest := s.receiveContactRequest(messageText, bob)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
// Bob accepts the contact request
|
|
s.acceptContactRequest(contactRequest, alice1, bob)
|
|
|
|
// Alice removes Bob from contacts
|
|
_, err := alice1.RetractContactRequest(&requests.RetractContactRequest{ID: types.Hex2Bytes(bob.myHexIdentity())})
|
|
s.Require().NoError(err)
|
|
|
|
// Adds bob again to her device
|
|
s.sendContactRequest(request, alice1)
|
|
|
|
// Wait for the message to reach its destination
|
|
_, err = WaitOnMessengerResponse(
|
|
bob,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.ActivityCenterNotifications()) > 0
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1) Alice sends a contact request to Bob
|
|
// 2) Bob accepts the contact request
|
|
// 3) Bob goes offline
|
|
// 4) Alice retracts the contact request
|
|
// 5) Alice adds bob back to her contacts
|
|
// 6) Bob goes online, they receive 4 and 5 in the wrong order
|
|
func (s *MessengerContactRequestSuite) TestAliceOfflineRetractsAndAddsWrongOrder() {
|
|
messageText := "hello!"
|
|
|
|
alice1 := s.m
|
|
|
|
bob := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob)
|
|
|
|
bobID := types.EncodeHex(crypto.FromECDSAPub(&bob.identity.PublicKey))
|
|
|
|
// Alice sends a contact request to bob
|
|
request := &requests.SendContactRequest{
|
|
ID: bobID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(request, alice1)
|
|
|
|
contactRequest := s.receiveContactRequest(messageText, bob)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
// Bob accepts the contact request
|
|
s.acceptContactRequest(contactRequest, alice1, bob)
|
|
|
|
// Alice removes Bob from contacts
|
|
_, err := alice1.RetractContactRequest(&requests.RetractContactRequest{ID: types.Hex2Bytes(bob.myHexIdentity())})
|
|
s.Require().NoError(err)
|
|
|
|
// Adds bob again to her device
|
|
s.sendContactRequest(request, alice1)
|
|
|
|
// Get alice perspective of bob
|
|
bobFromAlice := alice1.AddedContacts()[0]
|
|
|
|
// Get bob perspective of alice
|
|
aliceFromBob := bob.MutualContacts()[0]
|
|
|
|
s.Require().NotNil(bobFromAlice)
|
|
s.Require().NotNil(aliceFromBob)
|
|
|
|
// We can't simulate out-of-order messages easily, so we need to do
|
|
// things manually here
|
|
result := aliceFromBob.ContactRequestPropagatedStateReceived(bobFromAlice.ContactRequestPropagatedState())
|
|
s.Require().True(result.newContactRequestReceived)
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1) Alice sends a contact request to Bob
|
|
// 2) Bob accepts the contact request
|
|
// 3) Alice removes Bob from contacts
|
|
// 4) Make sure Alice and Bob are not mutual contacts
|
|
// 5) Alice sends new contact request
|
|
// 6) Bob accepts new contact request
|
|
func (s *MessengerContactRequestSuite) TestAliceResendsContactRequestAfterRemovingBobFromContacts() {
|
|
messageTextFirst := "hello 1!"
|
|
|
|
theirMessenger := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, theirMessenger)
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
|
|
// Alice sends a contact request to Bob
|
|
request := &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageTextFirst,
|
|
}
|
|
s.sendContactRequest(request, s.m)
|
|
|
|
// Bob accepts the contact request
|
|
contactRequest := s.receiveContactRequest(messageTextFirst, theirMessenger)
|
|
s.Require().NotNil(contactRequest)
|
|
s.acceptContactRequest(contactRequest, s.m, theirMessenger)
|
|
|
|
// Alice removes Bob from contacts
|
|
s.retractContactRequest(contactID, theirMessenger)
|
|
|
|
// Send new contact request
|
|
messageTextSecond := "hello 2!"
|
|
|
|
// Alice sends new contact request
|
|
request = &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageTextSecond,
|
|
}
|
|
s.sendContactRequest(request, s.m)
|
|
|
|
// Make sure bob and alice are not mutual after sending CR
|
|
s.Require().Len(s.m.MutualContacts(), 0)
|
|
s.Require().Len(theirMessenger.MutualContacts(), 0)
|
|
|
|
// Bob accepts new contact request
|
|
contactRequest = s.receiveContactRequest(messageTextSecond, theirMessenger)
|
|
s.Require().NotNil(contactRequest)
|
|
s.acceptContactRequest(contactRequest, s.m, theirMessenger)
|
|
|
|
// Make sure bob and alice are not mutual after sending CR
|
|
s.Require().Len(s.m.MutualContacts(), 1)
|
|
s.Require().Len(theirMessenger.MutualContacts(), 1)
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1) Alice sends a contact request to Bob
|
|
// 2) Bob declines the contact request from Alice
|
|
// 3) Bob sends a contact request to Alice
|
|
// 4) Alice and Bob are mutual contacts (because Alice's CR is "pending" on her side), Both CRs are accepted
|
|
func (s *MessengerContactRequestSuite) TestBobSendsContactRequestAfterDecliningOneFromAlice() {
|
|
messageTextAlice := "hello, Bobby!"
|
|
|
|
alice := s.m
|
|
|
|
bob := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob)
|
|
|
|
bobID := types.EncodeHex(crypto.FromECDSAPub(&bob.identity.PublicKey))
|
|
|
|
// Alice sends a contact request to bob
|
|
requestFromAlice := &requests.SendContactRequest{
|
|
ID: bobID,
|
|
Message: messageTextAlice,
|
|
}
|
|
s.sendContactRequest(requestFromAlice, alice)
|
|
|
|
contactRequest := s.receiveContactRequest(messageTextAlice, bob)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
// Bob declines the contact request
|
|
s.declineContactRequest(contactRequest, bob)
|
|
|
|
messageTextBob := "hello, Alice!"
|
|
|
|
aliceID := types.EncodeHex(crypto.FromECDSAPub(&alice.identity.PublicKey))
|
|
|
|
// Bob sends a contact request to Alice
|
|
requestFromBob := &requests.SendContactRequest{
|
|
ID: aliceID,
|
|
Message: messageTextBob,
|
|
}
|
|
|
|
// Send contact request
|
|
resp, err := bob.SendContactRequest(context.Background(), requestFromBob)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
|
|
// Check CR message, it should be accepted
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
contactRequest = s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
s.Require().Equal(common.ContactRequestStateAccepted, contactRequest.ContactRequestState)
|
|
s.Require().Equal(requestFromBob.Message, contactRequest.Text)
|
|
|
|
// Check pending notification
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(ActivityCenterNotificationTypeContactRequest, resp.ActivityCenterNotifications()[0].Type)
|
|
s.Require().Equal(contactRequest.ID, resp.ActivityCenterNotifications()[0].Message.ID)
|
|
s.Require().Equal(contactRequest.ContactRequestState, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, true)
|
|
|
|
// Check contacts Bob's side
|
|
s.Require().Len(resp.Contacts, 1)
|
|
contact := resp.Contacts[0]
|
|
s.Require().True(contact.mutual())
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) TestBuildContact() {
|
|
contactKey, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&contactKey.PublicKey))
|
|
|
|
contact, err := s.m.BuildContact(&requests.BuildContact{PublicKey: contactID})
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(contact.EnsName, "")
|
|
s.Require().False(contact.ENSVerified)
|
|
|
|
contact, err = s.m.BuildContact(&requests.BuildContact{PublicKey: contactID, ENSName: "foobar"})
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(contact.EnsName, "foobar")
|
|
s.Require().True(contact.ENSVerified)
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) TestReceiveAcceptAndRetractContactRequestOutOfOrder() {
|
|
message := protobuf.ChatMessage{
|
|
Clock: 4,
|
|
Timestamp: 1,
|
|
Text: "some text",
|
|
ChatId: common.PubkeyToHex(&s.m.identity.PublicKey),
|
|
MessageType: protobuf.MessageType_ONE_TO_ONE,
|
|
ContentType: protobuf.ChatMessage_CONTACT_REQUEST,
|
|
}
|
|
|
|
contactKey, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
|
|
contact, err := BuildContactFromPublicKey(&contactKey.PublicKey)
|
|
s.Require().NoError(err)
|
|
|
|
state := s.m.buildMessageState()
|
|
|
|
state.CurrentMessageState = &CurrentMessageState{
|
|
PublicKey: &contactKey.PublicKey,
|
|
MessageID: "0xa",
|
|
StatusMessage: &v1protocol.StatusMessage{TransportLayer: v1protocol.TransportLayer{Message: &types.Message{Timestamp: 1}}, ApplicationLayer: v1protocol.ApplicationLayer{ID: []byte("test-id")}},
|
|
Contact: contact,
|
|
WhisperTimestamp: 1,
|
|
}
|
|
|
|
response := state.Response
|
|
err = s.m.HandleChatMessage(state, &message, nil, false)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(response.ActivityCenterNotifications(), 1)
|
|
contacts := s.m.Contacts()
|
|
s.Require().Len(contacts, 1)
|
|
s.Require().Equal(ContactRequestStateReceived, contacts[0].ContactRequestRemoteState)
|
|
|
|
retract := protobuf.RetractContactRequest{
|
|
Clock: 2,
|
|
}
|
|
err = s.m.HandleRetractContactRequest(state, &retract, nil)
|
|
s.Require().NoError(err)
|
|
|
|
// Nothing should have changed
|
|
contacts = s.m.Contacts()
|
|
s.Require().Len(contacts, 1)
|
|
s.Require().Equal(ContactRequestStateReceived, contacts[0].ContactRequestRemoteState)
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1) Alice sends a contact request to Bob
|
|
// 2) Bob receives CR from Alice
|
|
// 3) Bob resets his device
|
|
// 4) Bob restores Alice's contact from backup, CR is created
|
|
// 5) Bob succesefully accepts restored contact request
|
|
// 6) Alice get notified properly
|
|
func (s *MessengerContactRequestSuite) TestBobRestoresIncomingContactRequestFromSyncInstallationContactV2() {
|
|
messageText := "hello, Bobby!"
|
|
|
|
alice := s.m
|
|
alice.account.CustomizationColor = multiaccountscommon.CustomizationColorBeige
|
|
|
|
bob1 := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob1)
|
|
|
|
aliceID := types.EncodeHex(crypto.FromECDSAPub(&alice.identity.PublicKey))
|
|
bobID := types.EncodeHex(crypto.FromECDSAPub(&bob1.identity.PublicKey))
|
|
|
|
// Alice sends a contact request to bob
|
|
requestFromAlice := &requests.SendContactRequest{
|
|
ID: bobID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(requestFromAlice, alice)
|
|
|
|
// Bob receives CR from Alice
|
|
contactRequest := s.receiveContactRequest(messageText, bob1)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
// Bob resets his device
|
|
bob2, err := newMessengerWithKey(s.shh, bob1.identity, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
defer TearDownMessenger(&s.Suite, bob2)
|
|
|
|
// Get bob perspective of alice for backup
|
|
aliceFromBob := bob1.Contacts()[0]
|
|
s.Require().Equal(aliceFromBob.CustomizationColor, alice.account.GetCustomizationColor())
|
|
state := bob2.buildMessageState()
|
|
|
|
// Restore alice's contact from backup
|
|
sync := s.syncInstallationContactV2FromContact(aliceFromBob)
|
|
err = bob2.HandleSyncInstallationContactV2(state, &sync, nil)
|
|
s.Require().NoError(err)
|
|
|
|
// Accept latest CR for a contact
|
|
resp, err := bob2.AcceptLatestContactRequestForContact(context.Background(), &requests.AcceptLatestContactRequestForContact{ID: types.Hex2Bytes(aliceID)})
|
|
s.Require().NoError(err)
|
|
s.Require().Len(resp.Contacts, 1)
|
|
|
|
// Make sure the message is updated
|
|
s.Require().NotNil(resp)
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
contactRequestMsg := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequestMsg)
|
|
|
|
// NOTE: We don't restore CR message
|
|
// s.Require().Equal(resp.Messages()[0].ID, contactRequest.ID)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, contactRequestMsg.ContactRequestState)
|
|
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
s.Require().True(resp.Contacts[0].mutual())
|
|
|
|
// Make sure the sender is added to our contacts
|
|
contacts := bob2.AddedContacts()
|
|
s.Require().Len(contacts, 1)
|
|
|
|
// Make sure we consider them a mutual contact, receiver side
|
|
mutualContacts := bob2.MutualContacts()
|
|
s.Require().Len(mutualContacts, 1)
|
|
}
|
|
|
|
// The scenario tested is as follow:
|
|
// 1) Alice sends a contact request to Bob
|
|
// 2) Bob receives CR from Alice
|
|
// 3) Alice resets her device
|
|
// 4) Alice restores Bob's contact from backup, CR is created
|
|
// 5) Bob accepts contact request
|
|
// 6) Alice get notified properly
|
|
func (s *MessengerContactRequestSuite) TestAliceRestoresOutgoingContactRequestFromSyncInstallationContactV2() {
|
|
messageText := "hello, Bobby!"
|
|
|
|
alice1 := s.m
|
|
|
|
bob := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob)
|
|
|
|
aliceID := types.EncodeHex(crypto.FromECDSAPub(&alice1.identity.PublicKey))
|
|
bobID := types.EncodeHex(crypto.FromECDSAPub(&bob.identity.PublicKey))
|
|
|
|
// Alice sends a contact request to bob
|
|
requestFromAlice := &requests.SendContactRequest{
|
|
ID: bobID,
|
|
Message: messageText,
|
|
}
|
|
s.sendContactRequest(requestFromAlice, alice1)
|
|
|
|
// Bob receives CR from Alice
|
|
contactRequest := s.receiveContactRequest(messageText, bob)
|
|
s.Require().NotNil(contactRequest)
|
|
|
|
// Bob resets his device
|
|
alice2, err := newMessengerWithKey(s.shh, alice1.identity, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
defer TearDownMessenger(&s.Suite, alice2)
|
|
|
|
// Get bob perspective of alice for backup
|
|
bobFromAlice := alice1.Contacts()[0]
|
|
state := alice2.buildMessageState()
|
|
|
|
// Restore alice's contact from backup
|
|
sync := s.syncInstallationContactV2FromContact(bobFromAlice)
|
|
err = alice2.HandleSyncInstallationContactV2(state, &sync, nil)
|
|
s.Require().NoError(err)
|
|
|
|
// Accept latest CR for a contact
|
|
resp, err := bob.AcceptLatestContactRequestForContact(context.Background(), &requests.AcceptLatestContactRequestForContact{ID: types.Hex2Bytes(aliceID)})
|
|
s.Require().NoError(err)
|
|
|
|
// Make sure the message is updated
|
|
s.Require().NotNil(resp)
|
|
s.Require().Len(resp.Messages(), 2)
|
|
|
|
contactRequestMsg := s.findFirstByContentType(resp.Messages(), protobuf.ChatMessage_CONTACT_REQUEST)
|
|
s.Require().NotNil(contactRequestMsg)
|
|
|
|
// NOTE: We don't restore CR message
|
|
// s.Require().Equal(resp.Messages()[0].ID, contactRequest.ID)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, contactRequestMsg.ContactRequestState)
|
|
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message)
|
|
s.Require().Equal(common.ContactRequestStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactRequestState)
|
|
|
|
// Check the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
s.Require().True(resp.Contacts[0].mutual())
|
|
|
|
// Make sure the sender is added to our contacts
|
|
contacts := bob.AddedContacts()
|
|
s.Require().Len(contacts, 1)
|
|
|
|
// Make sure we consider them a mutual contact, receiver side
|
|
mutualContacts := bob.MutualContacts()
|
|
s.Require().Len(mutualContacts, 1)
|
|
}
|
|
|
|
/*
|
|
Makes Alice and Bob mutual contacts.
|
|
Verifies that Alice device-2 receives mutual contact information.
|
|
Contact request is sent from Alice device 1.
|
|
*/
|
|
func (s *MessengerContactRequestSuite) makeMutualContactsAndSync(alice1 *Messenger, alice2 *Messenger, bob *Messenger, messageText string) {
|
|
bobPublicKey := bob.IdentityPublicKeyString()
|
|
|
|
cr := s.createContactRequest(bobPublicKey, messageText)
|
|
s.sendContactRequest(cr, alice1)
|
|
receivedCR := s.receiveContactRequest(cr.Message, bob)
|
|
s.acceptContactRequest(receivedCR, alice1, bob)
|
|
s.checkMutualContact(alice1, bobPublicKey)
|
|
|
|
// Wait for Alice-2 to sync new contact
|
|
resp, _ := WaitOnMessengerResponse(alice2, func(r *MessengerResponse) bool {
|
|
// FIXME: https://github.com/status-im/status-go/issues/3803
|
|
// No condition here. There are randomly received 1-3 messages.
|
|
return false // len(r.Contacts) == 1 && len(r.Messages()) == 3
|
|
}, "alice-2 didn't receive bob contact")
|
|
s.logResponse(resp, "Wait for Alice-2 to sync new contact")
|
|
s.Require().NotNil(resp)
|
|
//s.Require().NoError(err) // WARNING: Uncomment when bug fixed. https://github.com/status-im/status-go/issues/3803
|
|
|
|
// Check that Alice-2 has Bob as a contact
|
|
s.Require().Len(alice2.Contacts(), 1)
|
|
s.Require().Equal(bobPublicKey, alice2.Contacts()[0].ID)
|
|
|
|
// TODO: https://github.com/status-im/status-go/issues/3803
|
|
// Check response messages and AC notifications when
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) blockContactAndSync(alice1 *Messenger, alice2 *Messenger, bob *Messenger) {
|
|
bobPublicKey := bob.IdentityPublicKeyString()
|
|
bobDisplayName, err := bob.settings.DisplayName()
|
|
s.Require().NoError(err)
|
|
|
|
// Alice-1 blocks Bob
|
|
_, err = alice1.BlockContact(context.Background(), bobPublicKey, false)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(alice1.BlockedContacts(), 1)
|
|
s.Require().Equal(bobPublicKey, alice1.BlockedContacts()[0].ID)
|
|
|
|
// Wait for Bob to receive message that he was removed as contact
|
|
resp, err := WaitOnMessengerResponse(bob, func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) == 1 && len(r.Messages()) == 1
|
|
}, "Bob didn't receive a message that he was removed as contact")
|
|
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
s.logResponse(resp, "Wait for Bob to receive message that he was removed as contact")
|
|
|
|
// Check response contacts
|
|
s.Require().Len(resp.Contacts, 1)
|
|
respContact := resp.Contacts[0]
|
|
s.Require().Equal(respContact.ID, alice1.IdentityPublicKeyString())
|
|
s.Require().Equal(ContactRequestStateNone, respContact.ContactRequestLocalState)
|
|
s.Require().Equal(ContactRequestStateNone, respContact.ContactRequestRemoteState)
|
|
|
|
// Check response messages
|
|
s.Require().Len(resp.Messages(), 1)
|
|
s.Require().Equal(resp.Messages()[0].Text, fmt.Sprintf(incomingMutualStateEventRemovedDefaultText, alice1.IdentityPublicKeyString()))
|
|
|
|
// Check response AC notifications
|
|
s.Require().Len(resp.ActivityCenterNotifications(), 1)
|
|
s.Require().Equal(resp.ActivityCenterNotifications()[0].Type, ActivityCenterNotificationTypeContactRemoved)
|
|
|
|
alice2.logger.Info("STARTING")
|
|
// Wait for Alice-2 to sync Bob blocked state
|
|
resp, err = WaitOnMessengerResponse(alice2, func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) == 1
|
|
}, "Alice-2 didn't receive blocking bob")
|
|
s.logResponse(resp, "Wait for Alice-2 to sync Bob blocked state")
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
|
|
// Check that Bob contact is synced with correct display name and blocked
|
|
s.Require().Len(alice2.Contacts(), 1)
|
|
respContact = alice2.Contacts()[0]
|
|
s.Require().True(respContact.Blocked)
|
|
s.Require().True(respContact.Removed)
|
|
s.Require().Equal(bobPublicKey, respContact.ID)
|
|
s.Require().Equal(bobDisplayName, respContact.DisplayName)
|
|
s.Require().Equal(ContactRequestStateDismissed, respContact.ContactRequestLocalState)
|
|
s.Require().Equal(ContactRequestStateReceived, respContact.ContactRequestRemoteState)
|
|
|
|
// Check chats list
|
|
s.Require().Len(alice2.Chats(), deprecation.AddChatsCount(2))
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) unblockContactAndSync(alice1 *Messenger, alice2 *Messenger, bob *Messenger) {
|
|
bobPublicKey := bob.IdentityPublicKeyString()
|
|
|
|
_, err := alice1.UnblockContact(bobPublicKey)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(alice1.BlockedContacts(), 0)
|
|
|
|
// Bob doesn't receive any message on blocking.
|
|
// No response wait here.
|
|
|
|
// Wait for Alice-2 to receive Bob unblocked state
|
|
resp, err := WaitOnMessengerResponse(alice2, func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) == 1
|
|
}, "Alice-2 didn't receive Bob unblocked state")
|
|
s.logResponse(resp, "Wait for Alice-2 to receive Bob unblocked state")
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(resp)
|
|
|
|
// Check that Alice-2 has Bob unblocked and removed
|
|
s.Require().Len(alice2.Contacts(), 1)
|
|
respContact := alice2.Contacts()[0]
|
|
s.Require().Equal(bobPublicKey, respContact.ID)
|
|
s.Require().False(respContact.Blocked)
|
|
s.Require().True(respContact.Removed)
|
|
s.Require().Equal(respContact.ContactRequestLocalState, ContactRequestStateNone)
|
|
s.Require().Equal(respContact.ContactRequestRemoteState, ContactRequestStateNone)
|
|
|
|
// Check chats list
|
|
s.Require().Len(alice2.Chats(), deprecation.AddChatsCount(2))
|
|
}
|
|
|
|
func (s *MessengerContactRequestSuite) TestBlockedContactSyncing() {
|
|
// Setup Bob
|
|
bob := s.newMessenger()
|
|
defer TearDownMessenger(&s.Suite, bob)
|
|
_ = bob.SetDisplayName("bob-1")
|
|
s.logger.Info("Bob account set up", zap.String("publicKey", bob.IdentityPublicKeyString()))
|
|
|
|
// Setup Alice-1
|
|
alice1 := s.m
|
|
s.logger.Info("Alice account set up", zap.String("publicKey", alice1.IdentityPublicKeyString()))
|
|
|
|
// Setup Alice-2
|
|
alice2, err := newMessengerWithKey(s.shh, s.m.identity, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
defer TearDownMessenger(&s.Suite, alice2)
|
|
|
|
// Pair alice-1 <-> alice-2
|
|
// NOTE: This doesn't include initial data sync. Local pairing could be used.
|
|
s.logger.Info("pairing Alice-1 and Alice-2")
|
|
prepAliceMessengersForPairing(&s.Suite, alice1, alice2)
|
|
PairDevices(&s.Suite, alice1, alice2)
|
|
PairDevices(&s.Suite, alice2, alice1)
|
|
s.logger.Info("pairing Alice-1 and Alice-2 finished")
|
|
|
|
// Loop cr-block-unblock. Some bugs happen at second iteration.
|
|
for i := 0; i < 2; i++ {
|
|
crText := fmt.Sprintf("hello-%d", i)
|
|
s.makeMutualContactsAndSync(alice1, alice2, bob, crText)
|
|
s.blockContactAndSync(alice1, alice2, bob)
|
|
s.unblockContactAndSync(alice1, alice2, bob)
|
|
}
|
|
}
|