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/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 TestMessengerBackupSuite(t *testing.T) { suite.Run(t, new(MessengerBackupSuite)) } type MessengerBackupSuite 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 *MessengerBackupSuite) 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.privateKey = s.m.identity _, err := s.m.Start() s.Require().NoError(err) } func (s *MessengerBackupSuite) TearDownTest() { s.Require().NoError(s.m.Shutdown()) } func (s *MessengerBackupSuite) newMessenger() *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 *MessengerBackupSuite) TestBackupContacts() { bob1 := s.m // Create bob2 bob2, err := newMessengerWithKey(s.shh, bob1.identity, s.logger, nil) s.Require().NoError(err) _, err = bob2.Start() s.Require().NoError(err) // Create 2 contacts contact1Key, err := crypto.GenerateKey() s.Require().NoError(err) contactID1 := types.EncodeHex(crypto.FromECDSAPub(&contact1Key.PublicKey)) _, err = bob1.AddContact(context.Background(), &requests.AddContact{ID: types.Hex2Bytes(contactID1)}) s.Require().NoError(err) contact2Key, err := crypto.GenerateKey() s.Require().NoError(err) contactID2 := types.EncodeHex(crypto.FromECDSAPub(&contact2Key.PublicKey)) _, err = bob1.AddContact(context.Background(), &requests.AddContact{ID: types.Hex2Bytes(contactID2)}) s.Require().NoError(err) s.Require().Len(bob1.Contacts(), 2) actualContacts := bob1.Contacts() if actualContacts[0].ID == contactID1 { s.Require().Equal(actualContacts[0].ID, contactID1) s.Require().Equal(actualContacts[1].ID, contactID2) } else { s.Require().Equal(actualContacts[0].ID, contactID2) s.Require().Equal(actualContacts[1].ID, contactID1) } // Backup clock, err := bob1.BackupData(context.Background()) s.Require().NoError(err) // Safety check s.Require().Len(bob2.Contacts(), 0) // Wait for the message to reach its destination _, err = WaitOnMessengerResponse( bob2, func(r *MessengerResponse) bool { _, err := s.m.RetrieveAll() if err != nil { s.logger.Info("Failed") return false } if len(bob2.Contacts()) < 2 { return false } return true }, "contacts not backed up", ) s.Require().NoError(err) s.Require().Len(bob2.AddedContacts(), 2) actualContacts = bob2.AddedContacts() if actualContacts[0].ID == contactID1 { s.Require().Equal(actualContacts[0].ID, contactID1) s.Require().Equal(actualContacts[1].ID, contactID2) } else { s.Require().Equal(actualContacts[0].ID, contactID2) s.Require().Equal(actualContacts[1].ID, contactID1) } lastBackup, err := bob1.lastBackup() s.Require().NoError(err) s.Require().NotEmpty(lastBackup) s.Require().Equal(clock, lastBackup) } func (s *MessengerBackupSuite) TestBackupContactsGreaterThanBatch() { bob1 := s.m // Create bob2 bob2, err := newMessengerWithKey(s.shh, bob1.identity, s.logger, nil) s.Require().NoError(err) _, err = bob2.Start() s.Require().NoError(err) // Create contacts for i := 0; i < BackupContactsPerBatch*2; i++ { contactKey, err := crypto.GenerateKey() s.Require().NoError(err) contactID := types.EncodeHex(crypto.FromECDSAPub(&contactKey.PublicKey)) _, err = bob1.AddContact(context.Background(), &requests.AddContact{ID: types.Hex2Bytes(contactID)}) s.Require().NoError(err) } // Backup _, err = bob1.BackupData(context.Background()) s.Require().NoError(err) // Safety check s.Require().Len(bob2.Contacts(), 0) // Wait for the message to reach its destination _, err = WaitOnMessengerResponse( bob2, func(r *MessengerResponse) bool { _, err := s.m.RetrieveAll() if err != nil { s.logger.Info("Failed") return false } if len(bob2.Contacts()) < BackupContactsPerBatch*2 { return false } return true }, "contacts not backed up", ) s.Require().NoError(err) s.Require().Len(bob2.AddedContacts(), BackupContactsPerBatch*2) } func (s *MessengerBackupSuite) TestBackupRemovedContact() { bob1 := s.m // Create bob2 bob2, err := newMessengerWithKey(s.shh, bob1.identity, s.logger, nil) s.Require().NoError(err) _, err = bob2.Start() s.Require().NoError(err) // Create 2 contacts on bob 1 contact1Key, err := crypto.GenerateKey() s.Require().NoError(err) contactID1 := types.EncodeHex(crypto.FromECDSAPub(&contact1Key.PublicKey)) _, err = bob1.AddContact(context.Background(), &requests.AddContact{ID: types.Hex2Bytes(contactID1)}) s.Require().NoError(err) contact2Key, err := crypto.GenerateKey() s.Require().NoError(err) contactID2 := types.EncodeHex(crypto.FromECDSAPub(&contact2Key.PublicKey)) _, err = bob1.AddContact(context.Background(), &requests.AddContact{ID: types.Hex2Bytes(contactID2)}) s.Require().NoError(err) s.Require().Len(bob1.Contacts(), 2) actualContacts := bob1.Contacts() if actualContacts[0].ID == contactID1 { s.Require().Equal(actualContacts[0].ID, contactID1) s.Require().Equal(actualContacts[1].ID, contactID2) } else { s.Require().Equal(actualContacts[0].ID, contactID2) s.Require().Equal(actualContacts[1].ID, contactID1) } // Bob 2 add one of the same contacts _, err = bob2.AddContact(context.Background(), &requests.AddContact{ID: types.Hex2Bytes(contactID2)}) s.Require().NoError(err) // Bob 1 now removes one of the contact that was also on bob 2 _, err = bob1.RemoveContact(context.Background(), contactID2) s.Require().NoError(err) // Backup clock, err := bob1.BackupData(context.Background()) s.Require().NoError(err) // Safety check s.Require().Len(bob2.Contacts(), 1) // Wait for the message to reach its destination _, err = WaitOnMessengerResponse( bob2, func(r *MessengerResponse) bool { _, err := s.m.RetrieveAll() if err != nil { s.logger.Info("Failed") return false } if len(bob2.Contacts()) != 1 && bob2.Contacts()[0].ID != contactID1 { return false } return true }, "contacts not backed up", ) // Bob 2 should remove the contact s.Require().NoError(err) s.Require().Len(bob2.AddedContacts(), 1) s.Require().Equal(contactID1, bob2.AddedContacts()[0].ID) lastBackup, err := bob1.lastBackup() s.Require().NoError(err) s.Require().NotEmpty(lastBackup) s.Require().Equal(clock, lastBackup) } func (s *MessengerBackupSuite) TestBackupLocalNickname() { bob1 := s.m // Create bob2 bob2, err := newMessengerWithKey(s.shh, bob1.identity, s.logger, nil) nickname := "don van vliet" s.Require().NoError(err) _, err = bob2.Start() s.Require().NoError(err) // Set contact nickname contact1Key, err := crypto.GenerateKey() s.Require().NoError(err) contactID1 := types.EncodeHex(crypto.FromECDSAPub(&contact1Key.PublicKey)) _, err = bob1.SetContactLocalNickname(&requests.SetContactLocalNickname{ID: types.Hex2Bytes(contactID1), Nickname: nickname}) s.Require().NoError(err) // Backup clock, err := bob1.BackupData(context.Background()) s.Require().NoError(err) // Safety check s.Require().Len(bob2.Contacts(), 0) var actualContact *Contact // Wait for the message to reach its destination _, err = WaitOnMessengerResponse( bob2, func(r *MessengerResponse) bool { _, err := s.m.RetrieveAll() if err != nil { s.logger.Info("Failed") return false } for _, c := range bob2.Contacts() { if c.ID == contactID1 { actualContact = c return true } } return false }, "contacts not backed up", ) s.Require().NoError(err) s.Require().Equal(actualContact.LocalNickname, nickname) lastBackup, err := bob1.lastBackup() s.Require().NoError(err) s.Require().NotEmpty(lastBackup) s.Require().Equal(clock, lastBackup) } func (s *MessengerBackupSuite) TestBackupBlockedContacts() { bob1 := s.m // Create bob2 bob2, err := newMessengerWithKey(s.shh, bob1.identity, s.logger, nil) s.Require().NoError(err) _, err = bob2.Start() s.Require().NoError(err) // Create contact contact1Key, err := crypto.GenerateKey() s.Require().NoError(err) contactID1 := types.EncodeHex(crypto.FromECDSAPub(&contact1Key.PublicKey)) _, err = bob1.AddContact(context.Background(), &requests.AddContact{ID: types.Hex2Bytes(contactID1)}) s.Require().NoError(err) // Backup _, err = bob1.BackupData(context.Background()) s.Require().NoError(err) // Safety check s.Require().Len(bob2.Contacts(), 0) // Wait for the message to reach its destination _, err = WaitOnMessengerResponse( bob2, func(r *MessengerResponse) bool { return len(bob2.AddedContacts()) >= 1 }, "contacts not backed up", ) s.Require().NoError(err) s.Require().Len(bob2.AddedContacts(), 1) actualContacts := bob2.AddedContacts() s.Require().Equal(actualContacts[0].ID, contactID1) _, err = bob1.BlockContact(contactID1) s.Require().NoError(err) // Backup _, err = bob1.BackupData(context.Background()) s.Require().NoError(err) // Wait for the message to reach its destination _, err = WaitOnMessengerResponse( bob2, func(r *MessengerResponse) bool { return len(bob2.BlockedContacts()) == 1 }, "blocked contact not received", ) s.Require().NoError(err) s.Require().Len(bob2.BlockedContacts(), 1) } func (s *MessengerBackupSuite) TestBackupCommunities() { bob1 := s.m // Create bob2 bob2, err := newMessengerWithKey(s.shh, bob1.identity, s.logger, nil) s.Require().NoError(err) _, err = bob2.Start() s.Require().NoError(err) // Create a communitie description := &requests.CreateCommunity{ Membership: protobuf.CommunityPermissions_NO_MEMBERSHIP, Name: "status", Color: "#ffffff", Description: "status community description", } // Create a community chat response, err := bob1.CreateCommunity(description, true) s.Require().NoError(err) s.Require().NotNil(response) s.Require().Len(response.Communities(), 1) // Backup clock, err := bob1.BackupData(context.Background()) s.Require().NoError(err) // Safety check communities, err := bob2.Communities() s.Require().NoError(err) s.Require().Len(communities, 1) // Wait for the message to reach its destination _, err = WaitOnMessengerResponse( bob2, func(r *MessengerResponse) bool { _, err := s.m.RetrieveAll() if err != nil { s.logger.Info("Failed") return false } communities, err := bob2.Communities() s.Require().NoError(err) return len(communities) >= 2 }, "communities not backed up", ) s.Require().NoError(err) communities, err = bob2.JoinedCommunities() s.Require().NoError(err) s.Require().Len(communities, 1) lastBackup, err := bob1.lastBackup() s.Require().NoError(err) s.Require().NotEmpty(lastBackup) s.Require().Equal(clock, lastBackup) }