409 lines
14 KiB
Go
409 lines
14 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"crypto/ecdsa"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
"go.uber.org/zap"
|
|
|
|
gethbridge "github.com/status-im/status-go/eth-node/bridge/geth"
|
|
"github.com/status-im/status-go/eth-node/crypto"
|
|
"github.com/status-im/status-go/eth-node/types"
|
|
"github.com/status-im/status-go/multiaccounts/accounts"
|
|
"github.com/status-im/status-go/protocol/common"
|
|
"github.com/status-im/status-go/protocol/protobuf"
|
|
"github.com/status-im/status-go/protocol/requests"
|
|
"github.com/status-im/status-go/protocol/tt"
|
|
"github.com/status-im/status-go/waku"
|
|
)
|
|
|
|
func TestMessengerProfileShowcaseSuite(t *testing.T) { // nolint: deadcode,unused
|
|
suite.Run(t, new(TestMessengerProfileShowcase))
|
|
}
|
|
|
|
type TestMessengerProfileShowcase struct {
|
|
suite.Suite
|
|
m *Messenger // main instance of Messenger
|
|
privateKey *ecdsa.PrivateKey // private key for the main instance of Messenger
|
|
|
|
// If one wants to send messages between different instances of Messenger,
|
|
// a single Waku service should be shared.
|
|
shh types.Waku
|
|
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func (s *TestMessengerProfileShowcase) SetupTest() {
|
|
s.logger = tt.MustCreateTestLogger()
|
|
config := waku.DefaultConfig
|
|
config.MinimumAcceptedPoW = 0
|
|
shh := waku.New(&config, s.logger)
|
|
s.shh = gethbridge.NewGethWakuWrapper(shh)
|
|
s.Require().NoError(shh.Start())
|
|
s.m = s.newMessenger(s.shh)
|
|
s.privateKey = s.m.identity
|
|
// We start the messenger in order to receive installations
|
|
_, err := s.m.Start()
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func (s *TestMessengerProfileShowcase) TearDownTest() {
|
|
s.Require().NoError(s.m.Shutdown())
|
|
}
|
|
|
|
func (s *TestMessengerProfileShowcase) 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 *TestMessengerProfileShowcase) mutualContact(theirMessenger *Messenger) {
|
|
messageText := "hello!"
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
request := &requests.SendContactRequest{
|
|
ID: contactID,
|
|
Message: messageText,
|
|
}
|
|
|
|
// Send contact request
|
|
_, err := s.m.SendContactRequest(context.Background(), request)
|
|
s.Require().NoError(err)
|
|
|
|
// Wait for the message to reach its destination
|
|
_, err = WaitOnMessengerResponse(
|
|
theirMessenger,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.Contacts) > 0 && len(r.Messages()) > 0
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
// 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
|
|
_, err = theirMessenger.AcceptContactRequest(context.Background(), &requests.AcceptContactRequest{ID: types.Hex2Bytes(contactRequests[0].ID)})
|
|
s.Require().NoError(err)
|
|
|
|
// 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 the contact state is correctly set
|
|
s.Require().Len(resp.Contacts, 1)
|
|
s.Require().True(resp.Contacts[0].mutual())
|
|
}
|
|
|
|
func (s *TestMessengerProfileShowcase) verifiedContact(theirMessenger *Messenger) {
|
|
theirPk := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))
|
|
challenge := "Want to see what I'm hiding in my profile showcase?"
|
|
|
|
_, err := s.m.SendContactVerificationRequest(context.Background(), theirPk, challenge)
|
|
s.Require().NoError(err)
|
|
|
|
// 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)
|
|
verificationRequestID := resp.VerificationRequests()[0].ID
|
|
|
|
_, err = theirMessenger.AcceptContactVerificationRequest(context.Background(), verificationRequestID, "For sure!")
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NoError(err)
|
|
|
|
// 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)
|
|
|
|
resp, err = s.m.VerifiedTrusted(context.Background(), &requests.VerifiedTrusted{ID: types.FromHex(verificationRequestID)})
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Len(resp.Messages(), 1)
|
|
s.Require().Equal(common.ContactVerificationStateTrusted, resp.Messages()[0].ContactVerificationState)
|
|
}
|
|
|
|
func (s *TestMessengerProfileShowcase) prepareShowcasePreferences() *ProfileShowcasePreferences {
|
|
communityEntry1 := &ProfileShowcaseCommunityPreference{
|
|
CommunityID: "0x01312357798976434",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityEveryone,
|
|
Order: 10,
|
|
}
|
|
|
|
communityEntry2 := &ProfileShowcaseCommunityPreference{
|
|
CommunityID: "0x01312357798976535",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityContacts,
|
|
Order: 11,
|
|
}
|
|
|
|
communityEntry3 := &ProfileShowcaseCommunityPreference{
|
|
CommunityID: "0x01312353452343552",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityIDVerifiedContacts,
|
|
Order: 12,
|
|
}
|
|
|
|
accountEntry := &ProfileShowcaseAccountPreference{
|
|
Address: "0cx34662234",
|
|
Name: "Status Account",
|
|
ColorID: "blue",
|
|
Emoji: ">:-]",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityEveryone,
|
|
Order: 17,
|
|
}
|
|
|
|
collectibleEntry := &ProfileShowcaseCollectiblePreference{
|
|
UID: "0x12378534257568678487683576",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityIDVerifiedContacts,
|
|
Order: 17,
|
|
}
|
|
|
|
assetEntry := &ProfileShowcaseAssetPreference{
|
|
Symbol: "SNT",
|
|
ShowcaseVisibility: ProfileShowcaseVisibilityNoOne,
|
|
Order: 17,
|
|
}
|
|
|
|
return &ProfileShowcasePreferences{
|
|
Communities: []*ProfileShowcaseCommunityPreference{communityEntry1, communityEntry2, communityEntry3},
|
|
Accounts: []*ProfileShowcaseAccountPreference{accountEntry},
|
|
Collectibles: []*ProfileShowcaseCollectiblePreference{collectibleEntry},
|
|
Assets: []*ProfileShowcaseAssetPreference{assetEntry},
|
|
}
|
|
}
|
|
|
|
func (s *TestMessengerProfileShowcase) TestSetAndGetProfileShowcasePreferences() {
|
|
request := s.prepareShowcasePreferences()
|
|
err := s.m.SetProfileShowcasePreferences(request)
|
|
s.Require().NoError(err)
|
|
|
|
// Restored preferences shoulf be same as stored
|
|
response, err := s.m.GetProfileShowcasePreferences()
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Len(response.Communities, 3)
|
|
s.Require().Equal(response.Communities[0], request.Communities[0])
|
|
s.Require().Equal(response.Communities[1], request.Communities[1])
|
|
s.Require().Equal(response.Communities[2], request.Communities[2])
|
|
|
|
s.Require().Len(response.Accounts, 1)
|
|
s.Require().Equal(response.Accounts[0], request.Accounts[0])
|
|
|
|
s.Require().Len(response.Collectibles, 1)
|
|
s.Require().Equal(response.Collectibles[0], request.Collectibles[0])
|
|
|
|
s.Require().Len(response.Assets, 1)
|
|
s.Require().Equal(response.Assets[0], request.Assets[0])
|
|
}
|
|
|
|
func (s *TestMessengerProfileShowcase) TestEncryptAndDecryptProfileShowcaseEntries() {
|
|
// Add mutual contact
|
|
theirMessenger := s.newMessenger(s.shh)
|
|
_, err := theirMessenger.Start()
|
|
s.Require().NoError(err)
|
|
defer theirMessenger.Shutdown() // nolint: errcheck
|
|
|
|
s.mutualContact(theirMessenger)
|
|
|
|
entries := &protobuf.ProfileShowcaseEntries{
|
|
Communities: []*protobuf.ProfileShowcaseCommunity{
|
|
&protobuf.ProfileShowcaseCommunity{
|
|
CommunityId: "0x01312357798976535235432345",
|
|
Order: 12,
|
|
},
|
|
&protobuf.ProfileShowcaseCommunity{
|
|
CommunityId: "0x12378534257568678487683576",
|
|
Order: 11,
|
|
},
|
|
},
|
|
Accounts: []*protobuf.ProfileShowcaseAccount{
|
|
&protobuf.ProfileShowcaseAccount{
|
|
Address: "0x00000323245",
|
|
Name: "Default",
|
|
ColorId: "red",
|
|
Emoji: "(=^ ◡ ^=)",
|
|
Order: 1,
|
|
},
|
|
},
|
|
Assets: []*protobuf.ProfileShowcaseAsset{
|
|
&protobuf.ProfileShowcaseAsset{
|
|
Symbol: "ETH",
|
|
Order: 2,
|
|
},
|
|
&protobuf.ProfileShowcaseAsset{
|
|
Symbol: "DAI",
|
|
Order: 3,
|
|
},
|
|
&protobuf.ProfileShowcaseAsset{
|
|
Symbol: "SNT",
|
|
Order: 1,
|
|
},
|
|
},
|
|
}
|
|
data, err := s.m.EncryptProfileShowcaseEntriesWithContactPubKeys(entries, s.m.Contacts())
|
|
s.Require().NoError(err)
|
|
|
|
entriesBack, err := theirMessenger.DecryptProfileShowcaseEntriesWithPubKey(&s.m.identity.PublicKey, data)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Equal(2, len(entriesBack.Communities))
|
|
s.Require().Equal(entries.Communities[0].CommunityId, entriesBack.Communities[0].CommunityId)
|
|
s.Require().Equal(entries.Communities[0].Order, entriesBack.Communities[0].Order)
|
|
s.Require().Equal(entries.Communities[1].CommunityId, entriesBack.Communities[1].CommunityId)
|
|
s.Require().Equal(entries.Communities[1].Order, entriesBack.Communities[1].Order)
|
|
|
|
s.Require().Equal(1, len(entriesBack.Accounts))
|
|
s.Require().Equal(entries.Accounts[0].Address, entriesBack.Accounts[0].Address)
|
|
s.Require().Equal(entries.Accounts[0].Name, entriesBack.Accounts[0].Name)
|
|
s.Require().Equal(entries.Accounts[0].ColorId, entriesBack.Accounts[0].ColorId)
|
|
s.Require().Equal(entries.Accounts[0].Emoji, entriesBack.Accounts[0].Emoji)
|
|
s.Require().Equal(entries.Accounts[0].Order, entriesBack.Accounts[0].Order)
|
|
|
|
s.Require().Equal(0, len(entriesBack.Collectibles))
|
|
|
|
s.Require().Equal(3, len(entriesBack.Assets))
|
|
s.Require().Equal(entries.Assets[0].Symbol, entriesBack.Assets[0].Symbol)
|
|
s.Require().Equal(entries.Assets[0].Order, entriesBack.Assets[0].Order)
|
|
s.Require().Equal(entries.Assets[1].Symbol, entriesBack.Assets[1].Symbol)
|
|
s.Require().Equal(entries.Assets[1].Order, entriesBack.Assets[1].Order)
|
|
s.Require().Equal(entries.Assets[2].Symbol, entriesBack.Assets[2].Symbol)
|
|
s.Require().Equal(entries.Assets[2].Order, entriesBack.Assets[2].Order)
|
|
}
|
|
|
|
func (s *TestMessengerProfileShowcase) TestShareShowcasePreferences() {
|
|
// Set Display name to pass shouldPublishChatIdentity check
|
|
profileKp := accounts.GetProfileKeypairForTest(true, false, false)
|
|
profileKp.KeyUID = s.m.account.KeyUID
|
|
profileKp.Accounts[0].KeyUID = s.m.account.KeyUID
|
|
|
|
err := s.m.settings.SaveOrUpdateKeypair(profileKp)
|
|
s.Require().NoError(err)
|
|
|
|
err = s.m.SetDisplayName("bobby")
|
|
s.Require().NoError(err)
|
|
|
|
// Add mutual contact
|
|
mutualContact := s.newMessenger(s.shh)
|
|
_, err = mutualContact.Start()
|
|
s.Require().NoError(err)
|
|
defer mutualContact.Shutdown() // nolint: errcheck
|
|
|
|
s.mutualContact(mutualContact)
|
|
|
|
// Add identity verified contact
|
|
verifiedContact := s.newMessenger(s.shh)
|
|
_, err = verifiedContact.Start()
|
|
s.Require().NoError(err)
|
|
defer verifiedContact.Shutdown() // nolint: errcheck
|
|
|
|
s.mutualContact(verifiedContact)
|
|
s.verifiedContact(verifiedContact)
|
|
|
|
// Save preferences to dispatch changes
|
|
request := s.prepareShowcasePreferences()
|
|
err = s.m.SetProfileShowcasePreferences(request)
|
|
s.Require().NoError(err)
|
|
|
|
// Get summarised profile data for mutual contact
|
|
resp, err := WaitOnMessengerResponse(
|
|
mutualContact,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.updatedProfileShowcases) > 0
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(resp.updatedProfileShowcases, 1)
|
|
|
|
contactID := types.EncodeHex(crypto.FromECDSAPub(&s.m.identity.PublicKey))
|
|
profileShowcase := resp.updatedProfileShowcases[contactID]
|
|
|
|
s.Require().Len(profileShowcase.Communities, 2)
|
|
|
|
// For everyone
|
|
s.Require().Equal(profileShowcase.Communities[0].CommunityID, request.Communities[0].CommunityID)
|
|
s.Require().Equal(profileShowcase.Communities[0].Order, request.Communities[0].Order)
|
|
|
|
// For contacts
|
|
s.Require().Equal(profileShowcase.Communities[1].CommunityID, request.Communities[1].CommunityID)
|
|
s.Require().Equal(profileShowcase.Communities[1].Order, request.Communities[1].Order)
|
|
|
|
s.Require().Len(profileShowcase.Accounts, 1)
|
|
s.Require().Equal(profileShowcase.Accounts[0].Address, request.Accounts[0].Address)
|
|
s.Require().Equal(profileShowcase.Accounts[0].Name, request.Accounts[0].Name)
|
|
s.Require().Equal(profileShowcase.Accounts[0].ColorID, request.Accounts[0].ColorID)
|
|
s.Require().Equal(profileShowcase.Accounts[0].Emoji, request.Accounts[0].Emoji)
|
|
s.Require().Equal(profileShowcase.Accounts[0].Order, request.Accounts[0].Order)
|
|
|
|
s.Require().Len(profileShowcase.Collectibles, 0)
|
|
s.Require().Len(profileShowcase.Assets, 0)
|
|
|
|
// Get summarised profile data for verified contact
|
|
resp, err = WaitOnMessengerResponse(
|
|
verifiedContact,
|
|
func(r *MessengerResponse) bool {
|
|
return len(r.updatedProfileShowcases) > 0
|
|
},
|
|
"no messages",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(resp.updatedProfileShowcases, 1)
|
|
|
|
// Here let's try synchronous
|
|
profileShowcase, err = verifiedContact.GetProfileShowcaseForContact(contactID)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().Len(profileShowcase.Communities, 3)
|
|
|
|
// For everyone
|
|
s.Require().Equal(profileShowcase.Communities[0].CommunityID, request.Communities[0].CommunityID)
|
|
s.Require().Equal(profileShowcase.Communities[0].Order, request.Communities[0].Order)
|
|
|
|
// For contacts
|
|
s.Require().Equal(profileShowcase.Communities[1].CommunityID, request.Communities[1].CommunityID)
|
|
s.Require().Equal(profileShowcase.Communities[1].Order, request.Communities[1].Order)
|
|
|
|
// For id verified
|
|
s.Require().Equal(profileShowcase.Communities[2].CommunityID, request.Communities[2].CommunityID)
|
|
s.Require().Equal(profileShowcase.Communities[2].Order, request.Communities[2].Order)
|
|
|
|
s.Require().Len(profileShowcase.Accounts, 1)
|
|
s.Require().Equal(profileShowcase.Accounts[0].Address, request.Accounts[0].Address)
|
|
s.Require().Equal(profileShowcase.Accounts[0].Name, request.Accounts[0].Name)
|
|
s.Require().Equal(profileShowcase.Accounts[0].ColorID, request.Accounts[0].ColorID)
|
|
s.Require().Equal(profileShowcase.Accounts[0].Emoji, request.Accounts[0].Emoji)
|
|
s.Require().Equal(profileShowcase.Accounts[0].Order, request.Accounts[0].Order)
|
|
|
|
s.Require().Len(profileShowcase.Collectibles, 1)
|
|
s.Require().Equal(profileShowcase.Collectibles[0].UID, request.Collectibles[0].UID)
|
|
s.Require().Equal(profileShowcase.Collectibles[0].Order, request.Collectibles[0].Order)
|
|
|
|
s.Require().Len(profileShowcase.Assets, 0)
|
|
}
|