package protocol import ( "context" "crypto/ecdsa" "errors" "fmt" "testing" 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/images" "github.com/status-im/status-go/protocol/encryption/multidevice" "github.com/status-im/status-go/protocol/tt" "github.com/status-im/status-go/waku" "github.com/stretchr/testify/suite" "go.uber.org/zap" "github.com/status-im/status-go/eth-node/types" ) func TestMessengerSyncProfilePictureSuite(t *testing.T) { suite.Run(t, new(MessengerSyncProfilePictureSuite)) } type MessengerSyncProfilePictureSuite 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 *MessengerSyncProfilePictureSuite) 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 *MessengerSyncProfilePictureSuite) TearDownTest() { s.Require().NoError(s.m.Shutdown()) } func (s *MessengerSyncProfilePictureSuite) 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 *MessengerSyncProfilePictureSuite) TestSyncProfilePicture() { // Add identity images keyUID := s.m.account.KeyUID // pair theirMessenger, err := newMessengerWithKey(s.shh, s.privateKey, s.logger, nil) s.Require().NoError(err) err = theirMessenger.SetInstallationMetadata(theirMessenger.installationID, &multidevice.InstallationMetadata{ Name: "their-name", DeviceType: "their-device-type", }) s.Require().NoError(err) response, err := theirMessenger.SendPairInstallation(context.Background()) s.Require().NoError(err) s.Require().NotNil(response) s.Require().Len(response.Chats(), 1) s.Require().False(response.Chats()[0].Active) // Wait for the message to reach its destination response, err = WaitOnMessengerResponse( s.m, func(r *MessengerResponse) bool { return len(r.Installations) > 0 }, "installation not received", ) s.Require().NoError(err) actualInstallation := response.Installations[0] s.Require().Equal(theirMessenger.installationID, actualInstallation.ID) s.Require().NotNil(actualInstallation.InstallationMetadata) s.Require().Equal("their-name", actualInstallation.InstallationMetadata.Name) s.Require().Equal("their-device-type", actualInstallation.InstallationMetadata.DeviceType) err = s.m.EnableInstallation(theirMessenger.installationID) s.Require().NoError(err) // Sync happens via subscription triggered from within StoreIdentityImages const ( lowClock = 5 highClock = 10 ) iis := images.SampleIdentityImages() for _, img := range iis { img.Clock = highClock } s.Require().NoError(s.m.multiAccounts.StoreIdentityImages(keyUID, iis, true)) // Wait for the message to reach its destination err = tt.RetryWithBackOff(func() error { response, err = theirMessenger.RetrieveAll() if err != nil { return err } syncedImages, err := theirMessenger.multiAccounts.GetIdentityImages(keyUID) if err != nil { return err } if len(syncedImages) == 2 { return nil } return errors.New("Not received all identity images") }) s.Require().NoError(err) syncedImages, err := theirMessenger.multiAccounts.GetIdentityImages(keyUID) s.Require().NoError(err) s.Require().Equal(2, len(syncedImages)) s.Require().Equal(2, len(response.IdentityImages)) // Check that we don't update images with earlier clock values for _, img := range iis { img.Clock = lowClock } iis2 := images.SampleIdentityImages() for i, img := range iis2 { img.Name = fmt.Sprintf("newimg%d", i) img.Clock = highClock } iis = append(iis, iis2...) s.Require().NoError(s.m.multiAccounts.StoreIdentityImages(keyUID, iis, true)) err = tt.RetryWithBackOff(func() error { response, err = theirMessenger.RetrieveAll() if err != nil { return err } syncedImages, err := theirMessenger.multiAccounts.GetIdentityImages(keyUID) if err != nil { return err } if len(syncedImages) == 4 { return nil } return errors.New("Not received all identity images") }) syncedImages, err = theirMessenger.multiAccounts.GetIdentityImages(keyUID) s.Require().NoError(err) s.Require().Equal(4, len(syncedImages)) for _, img := range syncedImages { s.Require().NotEqual(img.Clock, lowClock) } s.Require().NoError(theirMessenger.Shutdown()) }