status-go/protocol/messenger_contact_verification_test.go

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
}

func (s *MessengerVerificationRequests) TestTrustStatus() {
	theirMessenger := s.newMessenger(s.shh)
	defer TearDownMessenger(&s.Suite, theirMessenger)

	s.mutualContact(theirMessenger)

	theirPk := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))

	// Test Mark as Trusted
	err := s.m.MarkAsTrusted(context.Background(), theirPk)
	s.Require().NoError(err)

	contactFromCache, ok := s.m.allContacts.Load(theirPk)
	s.Require().True(ok)
	s.Require().Equal(verification.TrustStatusTRUSTED, contactFromCache.TrustStatus)
	trustStatusFromDb, err := s.m.GetTrustStatus(theirPk)
	s.Require().NoError(err)
	s.Require().Equal(verification.TrustStatusTRUSTED, trustStatusFromDb)

	// Test Remove Trust Mark
	err = s.m.RemoveTrustStatus(context.Background(), theirPk)
	s.Require().NoError(err)

	contactFromCache, ok = s.m.allContacts.Load(theirPk)
	s.Require().True(ok)
	s.Require().Equal(verification.TrustStatusUNKNOWN, contactFromCache.TrustStatus)
	trustStatusFromDb, err = s.m.GetTrustStatus(theirPk)
	s.Require().NoError(err)
	s.Require().Equal(verification.TrustStatusUNKNOWN, trustStatusFromDb)

	// Test Mark as Untrustoworthy
	err = s.m.MarkAsUntrustworthy(context.Background(), theirPk)
	s.Require().NoError(err)

	contactFromCache, ok = s.m.allContacts.Load(theirPk)
	s.Require().True(ok)
	s.Require().Equal(verification.TrustStatusUNTRUSTWORTHY, contactFromCache.TrustStatus)
	trustStatusFromDb, err = s.m.GetTrustStatus(theirPk)
	s.Require().NoError(err)
	s.Require().Equal(verification.TrustStatusUNTRUSTWORTHY, trustStatusFromDb)

	// Test calling with an unknown contact
	err = s.m.MarkAsTrusted(context.Background(), "0x00000123")
	s.Require().Error(err)
	s.Require().Equal("contact not found", err.Error())
}