package protocol import ( "context" "testing" "github.com/status-im/status-go/eth-node/crypto" "github.com/status-im/status-go/protocol/common" "github.com/status-im/status-go/protocol/requests" "github.com/status-im/status-go/protocol/verification" "github.com/stretchr/testify/suite" "github.com/status-im/status-go/eth-node/types" ) func TestMessengerVerificationRequests(t *testing.T) { // nolint: deadcode,unused suite.Run(t, new(MessengerVerificationRequests)) } type MessengerVerificationRequests struct { MessengerBaseTestSuite } func (s *MessengerVerificationRequests) mutualContact(theirMessenger *Messenger) { messageText := "hello!" contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey)) request := &requests.SendContactRequest{ ID: contactID, Message: messageText, } // Send contact request resp, err := s.m.SendContactRequest(context.Background(), request) s.Require().NoError(err) s.Require().NotNil(resp) s.Require().Len(resp.Messages(), 2) // Make sure it's not returned as coming from us contactRequests, _, err := s.m.PendingContactRequests("", 10) s.Require().NoError(err) s.Require().Len(contactRequests, 0) // Make sure contact is added on the sender side contacts := s.m.AddedContacts() s.Require().Len(contacts, 1) s.Require().Equal(ContactRequestStateSent, contacts[0].ContactRequestLocalState) // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( theirMessenger, func(r *MessengerResponse) bool { return len(r.Contacts) > 0 && len(r.Messages()) > 0 && len(r.ActivityCenterNotifications()) > 0 }, "no messages", ) // Check contact request has been received s.Require().NoError(err) // 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.ContactRequestStatePending, resp.ActivityCenterNotifications()[0].Message.ContactRequestState) // Check the contact state is correctly set s.Require().Len(resp.Contacts, 1) s.Require().Equal(ContactRequestStateReceived, resp.Contacts[0].ContactRequestRemoteState) // Make sure it's the pending contact requests contactRequests, _, err = theirMessenger.PendingContactRequests("", 10) s.Require().NoError(err) s.Require().Len(contactRequests, 1) s.Require().Equal(contactRequests[0].ContactRequestState, common.ContactRequestStatePending) // Accept contact request, receiver side resp, err = theirMessenger.AcceptContactRequest(context.Background(), &requests.AcceptContactRequest{ID: types.Hex2Bytes(contactRequests[0].ID)}) s.Require().NoError(err) // Make sure the message is updated s.Require().NotNil(resp) s.Require().Len(resp.Messages(), 2) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), contactRequests[0].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) // Check activity center notification is of the right type 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 the message is updated, sender side s.Require().NotNil(resp) s.Require().Len(resp.Messages(), 2) // 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 *MessengerVerificationRequests) TestAcceptVerificationRequests() { theirMessenger := s.newMessenger(s.shh) defer TearDownMessenger(&s.Suite, theirMessenger) s.mutualContact(theirMessenger) theirPk := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey)) challenge := "challenge" resp, err := s.m.SendContactVerificationRequest(context.Background(), theirPk, challenge) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) verificationRequestID := resp.VerificationRequests()[0].ID s.Require().Len(resp.Messages(), 1) s.Require().NotEmpty(resp.Messages()[0].OutgoingStatus) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(common.ContactVerificationStatePending, resp.Messages()[0].ContactVerificationState) // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( theirMessenger, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 && len(r.ActivityCenterNotifications()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].Type, ActivityCenterNotificationTypeContactVerification) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusPENDING) s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false) s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message) s.Require().Equal(challenge, resp.ActivityCenterNotifications()[0].Message.Text) s.Require().Equal(common.ContactVerificationStatePending, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().Len(resp.Messages(), 1) s.Require().Empty(resp.Messages()[0].OutgoingStatus) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(common.ContactVerificationStatePending, resp.Messages()[0].ContactVerificationState) resp, err = theirMessenger.AcceptContactVerificationRequest(context.Background(), verificationRequestID, "hello back") s.Require().NoError(err) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.VerificationRequests()[0].RequestStatus, verification.RequestStatusACCEPTED) s.Require().NotEmpty(resp.VerificationRequests()[0].RepliedAt) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusACCEPTED) 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().Equal(common.ContactVerificationStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().Len(resp.Messages(), 2) s.Require().Equal(common.ContactVerificationStateAccepted, resp.Messages()[0].ContactVerificationState) s.Require().NotNil(resp.ActivityCenterNotifications()[0].ReplyMessage) s.Require().NotEmpty(resp.ActivityCenterNotifications()[0].ReplyMessage.OutgoingStatus) s.Require().Equal("hello back", resp.ActivityCenterNotifications()[0].ReplyMessage.Text) // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( s.m, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) messages := resp.Messages() s.Require().Len(messages, 2) var originalMessage *common.Message var replyMessage *common.Message if messages[0].ID == resp.VerificationRequests()[0].ID { originalMessage = messages[0] replyMessage = messages[1] } else { originalMessage = messages[1] replyMessage = messages[0] } s.Require().Equal(common.ContactVerificationStateAccepted, originalMessage.ContactVerificationState) // Not sure this is the right state for this, it should probably be pendingtrustedverification or something s.Require().Equal(common.ContactVerificationStateAccepted, replyMessage.ContactVerificationState) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusACCEPTED) s.Require().Equal(common.ContactVerificationStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().NotNil(resp.ActivityCenterNotifications()[0].ReplyMessage) s.Require().Empty(resp.ActivityCenterNotifications()[0].ReplyMessage.OutgoingStatus) s.Require().Equal("hello back", resp.ActivityCenterNotifications()[0].ReplyMessage.Text) s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false) // Mark as tusted resp, err = s.m.VerifiedTrusted(context.Background(), &requests.VerifiedTrusted{ID: types.FromHex(verificationRequestID)}) s.Require().NoError(err) s.Require().NotNil(resp) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusTRUSTED) s.Require().Equal(common.ContactVerificationStateTrusted, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().Len(resp.Messages(), 1) s.Require().Equal(common.ContactVerificationStateTrusted, resp.Messages()[0].ContactVerificationState) s.Require().Len(resp.Contacts, 1) s.Require().Equal(resp.Contacts[0].ID, theirPk) s.Require().Equal(resp.Contacts[0].VerificationStatus, VerificationStatusVERIFIED) } func (s *MessengerVerificationRequests) TestTrustedVerificationRequests() { theirMessenger := s.newMessenger(s.shh) defer TearDownMessenger(&s.Suite, theirMessenger) s.mutualContact(theirMessenger) theirPk := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey)) challenge := "challenge" resp, err := s.m.SendContactVerificationRequest(context.Background(), theirPk, challenge) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) verificationRequestID := resp.VerificationRequests()[0].ID s.Require().Len(resp.Messages(), 1) s.Require().NotEmpty(resp.Messages()[0].OutgoingStatus) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(common.ContactVerificationStatePending, resp.Messages()[0].ContactVerificationState) // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( theirMessenger, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 && len(r.ActivityCenterNotifications()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].Type, ActivityCenterNotificationTypeContactVerification) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusPENDING) s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false) s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message) s.Require().Equal(challenge, resp.ActivityCenterNotifications()[0].Message.Text) s.Require().Equal(common.ContactVerificationStatePending, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().Len(resp.Messages(), 1) s.Require().Empty(resp.Messages()[0].OutgoingStatus) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(common.ContactVerificationStatePending, resp.Messages()[0].ContactVerificationState) resp, err = theirMessenger.AcceptContactVerificationRequest(context.Background(), verificationRequestID, "hello back") s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.VerificationRequests()[0].RequestStatus, verification.RequestStatusACCEPTED) s.Require().NotEmpty(resp.VerificationRequests()[0].RepliedAt) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusACCEPTED) 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().Equal(common.ContactVerificationStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().Len(resp.Messages(), 2) s.Require().Equal(common.ContactVerificationStateAccepted, resp.Messages()[0].ContactVerificationState) s.Require().NotNil(resp.ActivityCenterNotifications()[0].ReplyMessage) s.Require().NotEmpty(resp.ActivityCenterNotifications()[0].ReplyMessage.OutgoingStatus) s.Require().Equal("hello back", resp.ActivityCenterNotifications()[0].ReplyMessage.Text) // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( s.m, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) messages := resp.Messages() s.Require().Len(messages, 2) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusACCEPTED) s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false) s.Require().Equal(common.ContactVerificationStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().NotNil(resp.ActivityCenterNotifications()[0].ReplyMessage) s.Require().Empty(resp.ActivityCenterNotifications()[0].ReplyMessage.OutgoingStatus) s.Require().Equal("hello back", resp.ActivityCenterNotifications()[0].ReplyMessage.Text) resp, err = s.m.VerifiedTrusted(context.Background(), &requests.VerifiedTrusted{ID: types.FromHex(verificationRequestID)}) s.Require().NoError(err) s.Require().NotNil(resp) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusTRUSTED) 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().Equal(common.ContactVerificationStateTrusted, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().Len(resp.Messages(), 1) s.Require().Equal(common.ContactVerificationStateTrusted, resp.Messages()[0].ContactVerificationState) } func (s *MessengerVerificationRequests) TestUnthrustworthyVerificationRequests() { theirMessenger := s.newMessenger(s.shh) defer TearDownMessenger(&s.Suite, theirMessenger) s.mutualContact(theirMessenger) theirPk := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey)) challenge := "challenge" resp, err := s.m.SendContactVerificationRequest(context.Background(), theirPk, challenge) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) verificationRequestID := resp.VerificationRequests()[0].ID s.Require().Len(resp.Messages(), 1) s.Require().NotEmpty(resp.Messages()[0].OutgoingStatus) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(common.ContactVerificationStatePending, resp.Messages()[0].ContactVerificationState) // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( theirMessenger, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 && len(r.ActivityCenterNotifications()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].Type, ActivityCenterNotificationTypeContactVerification) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusPENDING) s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false) s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message) s.Require().Equal(challenge, resp.ActivityCenterNotifications()[0].Message.Text) s.Require().Equal(common.ContactVerificationStatePending, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().Len(resp.Messages(), 1) s.Require().Empty(resp.Messages()[0].OutgoingStatus) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(common.ContactVerificationStatePending, resp.Messages()[0].ContactVerificationState) resp, err = theirMessenger.AcceptContactVerificationRequest(context.Background(), verificationRequestID, "hello back") s.Require().NoError(err) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.VerificationRequests()[0].RequestStatus, verification.RequestStatusACCEPTED) s.Require().NotEmpty(resp.VerificationRequests()[0].RepliedAt) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusACCEPTED) 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().Equal(common.ContactVerificationStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().Len(resp.Messages(), 2) s.Require().Equal(common.ContactVerificationStateAccepted, resp.Messages()[0].ContactVerificationState) s.Require().NotNil(resp.ActivityCenterNotifications()[0].ReplyMessage) s.Require().NotEmpty(resp.ActivityCenterNotifications()[0].ReplyMessage.OutgoingStatus) s.Require().Equal("hello back", resp.ActivityCenterNotifications()[0].ReplyMessage.Text) // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( s.m, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) messages := resp.Messages() s.Require().Len(messages, 2) var originalMessage *common.Message var replyMessage *common.Message if messages[0].ID == resp.VerificationRequests()[0].ID { originalMessage = messages[0] replyMessage = messages[1] } else { originalMessage = messages[1] replyMessage = messages[0] } s.Require().Equal(common.ContactVerificationStateAccepted, originalMessage.ContactVerificationState) // Not sure this is the right state for this, it should probably be pendingtrustedverification or something s.Require().Equal(common.ContactVerificationStateAccepted, replyMessage.ContactVerificationState) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusACCEPTED) s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false) s.Require().Equal(common.ContactVerificationStateAccepted, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().NotNil(resp.ActivityCenterNotifications()[0].ReplyMessage) s.Require().Empty(resp.ActivityCenterNotifications()[0].ReplyMessage.OutgoingStatus) s.Require().Equal("hello back", resp.ActivityCenterNotifications()[0].ReplyMessage.Text) resp, err = s.m.VerifiedUntrustworthy(context.Background(), &requests.VerifiedUntrustworthy{ID: types.FromHex(verificationRequestID)}) s.Require().NoError(err) s.Require().NotNil(resp) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusUNTRUSTWORTHY) 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().Equal(common.ContactVerificationStateUntrustworthy, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().Len(resp.Messages(), 1) s.Require().Equal(common.ContactVerificationStateUntrustworthy, resp.Messages()[0].ContactVerificationState) } func (s *MessengerVerificationRequests) TestRemoveTrustVerificationStatus() { // GIVEN theirMessenger := s.newMessenger(s.shh) defer TearDownMessenger(&s.Suite, theirMessenger) s.mutualContact(theirMessenger) theirPk := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey)) challenge := "challenge" resp, err := s.m.SendContactVerificationRequest(context.Background(), theirPk, challenge) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) verificationRequestID := resp.VerificationRequests()[0].ID // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( theirMessenger, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 && len(r.ActivityCenterNotifications()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) resp, err = theirMessenger.AcceptContactVerificationRequest(context.Background(), verificationRequestID, "hello back") s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) // Wait for the message to reach its destination _, err = WaitOnMessengerResponse( s.m, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 }, "no messages", ) s.Require().NoError(err) // Mark as trusted _, err = s.m.VerifiedTrusted(context.Background(), &requests.VerifiedTrusted{ID: types.FromHex(verificationRequestID)}) s.Require().NoError(err) // WHEN _, err = s.m.RemoveTrustVerificationStatus(context.Background(), theirPk) s.Require().NoError(err) // THEN trustStatus, err := s.m.GetTrustStatus(theirPk) s.Require().NoError(err) s.Require().Equal(verification.TrustStatusUNKNOWN, trustStatus) contact, _ := s.m.allContacts.Load(theirPk) s.Require().NotNil(contact) s.Require().Equal(VerificationStatusUNVERIFIED, contact.VerificationStatus) s.Require().Equal(verification.TrustStatusUNKNOWN, contact.TrustStatus) } func (s *MessengerVerificationRequests) TestDeclineVerificationRequests() { theirMessenger := s.newMessenger(s.shh) defer TearDownMessenger(&s.Suite, theirMessenger) s.mutualContact(theirMessenger) theirPk := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey)) challenge := "challenge" resp, err := s.m.SendContactVerificationRequest(context.Background(), theirPk, challenge) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) verificationRequestID := resp.VerificationRequests()[0].ID s.Require().Len(resp.Messages(), 1) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(common.ContactVerificationStatePending, resp.Messages()[0].ContactVerificationState) // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( theirMessenger, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 && len(r.ActivityCenterNotifications()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].Type, ActivityCenterNotificationTypeContactVerification) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusPENDING) s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false) s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message) s.Require().Equal(challenge, resp.ActivityCenterNotifications()[0].Message.Text) s.Require().Equal(resp.ActivityCenterNotifications()[0].Message.ContactVerificationState, common.ContactVerificationStatePending) s.Require().Len(resp.Messages(), 1) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(resp.Messages()[0].ContactVerificationState, common.ContactVerificationStatePending) // Make sure it's stored and retrieved correctly notification, err := theirMessenger.ActivityCenterNotification(types.FromHex(verificationRequestID)) s.Require().NoError(err) s.Require().NotNil(notification) s.Require().Equal(notification.ContactVerificationStatus, verification.RequestStatusPENDING) s.Require().Equal(notification.Message.ContactVerificationState, common.ContactVerificationStatePending) s.Require().Equal(notification.Read, false) s.Require().Equal(notification.Accepted, false) s.Require().Equal(notification.Dismissed, false) resp, err = theirMessenger.DeclineContactVerificationRequest(context.Background(), verificationRequestID) s.Require().NoError(err) s.Require().NotNil(resp) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.VerificationRequests()[0].RequestStatus, verification.RequestStatusDECLINED) s.Require().NotEmpty(resp.VerificationRequests()[0].RepliedAt) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusDECLINED) s.Require().Equal(resp.ActivityCenterNotifications()[0].Message.ContactVerificationState, common.ContactVerificationStateDeclined) 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().Len(resp.Messages(), 1) s.Require().Equal(resp.Messages()[0].ContactVerificationState, common.ContactVerificationStateDeclined) // Make sure it's stored and retrieved correctly notification, err = theirMessenger.ActivityCenterNotification(types.FromHex(verificationRequestID)) s.Require().NoError(err) s.Require().NotNil(notification) s.Require().Equal(notification.ContactVerificationStatus, verification.RequestStatusDECLINED) s.Require().Equal(notification.Message.ContactVerificationState, common.ContactVerificationStateDeclined) s.Require().Equal(notification.Read, true) s.Require().Equal(notification.Accepted, false) s.Require().Equal(notification.Dismissed, true) // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( s.m, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Len(resp.Messages(), 1) s.Require().Equal(resp.Messages()[0].ContactVerificationState, common.ContactVerificationStateDeclined) s.Require().Len(resp.Contacts, 1) s.Require().Equal(resp.Contacts[0].ID, theirPk) s.Require().Equal(resp.Contacts[0].VerificationStatus, VerificationStatusUNVERIFIED) s.Require().Len(resp.ActivityCenterNotifications(), 1) s.Require().Equal(resp.ActivityCenterNotifications()[0].ID.String(), verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusDECLINED) s.Require().Equal(resp.ActivityCenterNotifications()[0].Message.ContactVerificationState, common.ContactVerificationStateDeclined) s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, true) s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, true) } func (s *MessengerVerificationRequests) TestCancelVerificationRequest() { theirMessenger := s.newMessenger(s.shh) defer TearDownMessenger(&s.Suite, theirMessenger) s.mutualContact(theirMessenger) theirPk := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey)) challenge := "challenge" resp, err := s.m.SendContactVerificationRequest(context.Background(), theirPk, challenge) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) verificationRequestID := resp.VerificationRequests()[0].ID s.Require().Len(resp.Messages(), 1) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(common.ContactVerificationStatePending, resp.Messages()[0].ContactVerificationState) // Wait for the message to reach its destination resp, err = WaitOnMessengerResponse( theirMessenger, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 && len(r.ActivityCenterNotifications()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].Type, ActivityCenterNotificationTypeContactVerification) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusPENDING) s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false) s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message) s.Require().Equal(challenge, resp.ActivityCenterNotifications()[0].Message.Text) s.Require().Equal(common.ContactVerificationStatePending, resp.ActivityCenterNotifications()[0].Message.ContactVerificationState) s.Require().Len(resp.Messages(), 1) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(common.ContactVerificationStatePending, resp.Messages()[0].ContactVerificationState) // Make sure it's stored and retrieved correctly notifications, err := theirMessenger.ActivityCenterNotifications( ActivityCenterNotificationsRequest{ Cursor: "", Limit: 4, ActivityTypes: []ActivityCenterType{ActivityCenterNotificationTypeContactVerification}, ReadType: ActivityCenterQueryParamsReadUnread, }, ) s.Require().NoError(err) s.Require().Greater(len(notifications.Notifications), 0) s.Require().Equal(notifications.Notifications[0].ContactVerificationStatus, verification.RequestStatusPENDING) s.Require().Equal(common.ContactVerificationStatePending, notifications.Notifications[0].Message.ContactVerificationState) resp, err = s.m.CancelVerificationRequest(context.Background(), verificationRequestID) s.Require().NoError(err) s.Require().NotNil(resp) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.VerificationRequests()[0].RequestStatus, verification.RequestStatusCANCELED) s.Require().Len(resp.Contacts, 1) s.Require().Equal(resp.Contacts[0].ID, theirPk) s.Require().Equal(resp.Contacts[0].VerificationStatus, VerificationStatusUNVERIFIED) // Check canceled state on the receiver's side resp, err = WaitOnMessengerResponse( theirMessenger, func(r *MessengerResponse) bool { return len(r.VerificationRequests()) == 1 && len(r.ActivityCenterNotifications()) == 1 }, "no messages", ) s.Require().NoError(err) s.Require().Len(resp.VerificationRequests(), 1) s.Require().Equal(resp.VerificationRequests()[0].ID, verificationRequestID) s.Require().Equal(resp.ActivityCenterNotifications()[0].Type, ActivityCenterNotificationTypeContactVerification) s.Require().Equal(resp.ActivityCenterNotifications()[0].ContactVerificationStatus, verification.RequestStatusCANCELED) s.Require().Equal(resp.ActivityCenterNotifications()[0].Read, true) s.Require().Equal(resp.ActivityCenterNotifications()[0].Accepted, false) s.Require().Equal(resp.ActivityCenterNotifications()[0].Dismissed, false) s.Require().NotNil(resp.ActivityCenterNotifications()[0].Message) s.Require().Equal(challenge, resp.ActivityCenterNotifications()[0].Message.Text) s.Require().Equal(resp.ActivityCenterNotifications()[0].Message.ContactVerificationState, common.ContactVerificationStateCanceled) s.Require().Len(resp.Messages(), 1) s.Require().Equal(challenge, resp.Messages()[0].Text) s.Require().Equal(resp.Messages()[0].ContactVerificationState, common.ContactVerificationStateCanceled) } func (s *MessengerVerificationRequests) newMessenger(shh types.Waku) *Messenger { privateKey, err := crypto.GenerateKey() s.Require().NoError(err) messenger, err := newMessengerWithKey(s.shh, privateKey, s.logger, nil) s.Require().NoError(err) return messenger }