mirror of
https://github.com/status-im/status-go.git
synced 2025-01-10 06:36:32 +00:00
2df9df10ab
interface for initializing db, which is implemented for appdatabase and walletdatabase. TBD for multiaccounts DB. Unified DB initializion for all tests using helpers and new interface. Reduced sqlcipher kdf iterations for all tests to 1.
335 lines
10 KiB
Go
335 lines
10 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"crypto/ecdsa"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
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/settings"
|
|
"github.com/status-im/status-go/protocol/encryption/multidevice"
|
|
"github.com/status-im/status-go/protocol/tt"
|
|
"github.com/status-im/status-go/services/wallet"
|
|
"github.com/status-im/status-go/waku"
|
|
)
|
|
|
|
func TestMessengerSyncSavedAddressesSuite(t *testing.T) {
|
|
suite.Run(t, new(MessengerSyncSavedAddressesSuite))
|
|
}
|
|
|
|
type MessengerSyncSavedAddressesSuite struct {
|
|
suite.Suite
|
|
main *Messenger // main instance of Messenger paired with `other`
|
|
other *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 *MessengerSyncSavedAddressesSuite) 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.main = s.newMessenger(s.shh)
|
|
s.privateKey = s.main.identity
|
|
// Start the main messenger in order to receive installations
|
|
_, err := s.main.Start()
|
|
s.Require().NoError(err)
|
|
|
|
// Create new device and add main account to
|
|
s.other, err = newMessengerWithKey(s.shh, s.main.identity, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
|
|
// Pair devices (main and other)
|
|
imOther := &multidevice.InstallationMetadata{
|
|
Name: "other-device",
|
|
DeviceType: "other-device-type",
|
|
}
|
|
err = s.other.SetInstallationMetadata(s.other.installationID, imOther)
|
|
s.Require().NoError(err)
|
|
response, err := s.other.SendPairInstallation(context.Background(), nil)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(response)
|
|
|
|
// Wait for the message to reach its destination
|
|
_, err = WaitOnMessengerResponse(
|
|
s.main,
|
|
func(r *MessengerResponse) bool { return len(r.Installations) > 0 },
|
|
"installation not received",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
err = s.main.EnableInstallation(s.other.installationID)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func (s *MessengerSyncSavedAddressesSuite) TearDownTest() {
|
|
s.Require().NoError(s.main.Shutdown())
|
|
}
|
|
|
|
func (s *MessengerSyncSavedAddressesSuite) 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
|
|
}
|
|
|
|
// Helpers duplicate of wallet test. Could not import it from saved_addresses_test.go
|
|
|
|
func contains[T comparable](container []T, element T, isEqual func(T, T) bool) bool {
|
|
for _, e := range container {
|
|
if isEqual(e, element) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func haveSameElements[T comparable](a []T, b []T, isEqual func(T, T) bool) bool {
|
|
for _, v := range a {
|
|
if !contains(b, v, isEqual) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func savedAddressDataIsEqual(a, b wallet.SavedAddress) bool {
|
|
return a.Address == b.Address && a.IsTest == b.IsTest && a.Name == b.Name &&
|
|
a.Favourite == b.Favourite && a.ENSName == b.ENSName && a.ChainShortNames == b.ChainShortNames
|
|
}
|
|
|
|
func (s *MessengerSyncSavedAddressesSuite) TestSyncExistingSavedAddresses() {
|
|
var isTestChain1 bool = false
|
|
var isTestChain2 bool = true
|
|
var testAddress1 = common.Address{1}
|
|
|
|
// Add saved addresses to main device
|
|
sa1 := wallet.SavedAddress{
|
|
Address: testAddress1,
|
|
Name: "TestC1A1",
|
|
Favourite: false,
|
|
IsTest: isTestChain1,
|
|
}
|
|
sa2 := wallet.SavedAddress{
|
|
ENSName: "test.ens.eth",
|
|
Name: "TestC2A1",
|
|
Favourite: true,
|
|
IsTest: isTestChain2,
|
|
}
|
|
|
|
savedAddressesManager := s.main.savedAddressesManager
|
|
|
|
_, err := savedAddressesManager.UpdateMetadataAndUpsertSavedAddress(sa1)
|
|
s.Require().NoError(err)
|
|
_, err = savedAddressesManager.UpdateMetadataAndUpsertSavedAddress(sa2)
|
|
s.Require().NoError(err)
|
|
|
|
// Trigger's a sync between devices
|
|
err = s.main.SyncDevices(context.Background(), "ens-name", "profile-image", nil)
|
|
s.Require().NoError(err)
|
|
|
|
// Wait and check that saved addresses are synced
|
|
_, err = WaitOnMessengerResponse(
|
|
s.other,
|
|
func(r *MessengerResponse) bool {
|
|
fmt.Println("LENG", len(r.SavedAddresses()))
|
|
if len(r.SavedAddresses()) == 2 {
|
|
sas := r.SavedAddresses()
|
|
s.Require().True(haveSameElements([]wallet.SavedAddress{sa1, sa2}, []wallet.SavedAddress{*sas[0], *sas[1]}, savedAddressDataIsEqual))
|
|
return true
|
|
}
|
|
return false
|
|
},
|
|
"expected to receive two changes",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
savedAddresses, err := s.other.savedAddressesManager.GetSavedAddresses()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(2, len(savedAddresses))
|
|
s.Require().True(haveSameElements([]wallet.SavedAddress{sa1, sa2}, savedAddresses, savedAddressDataIsEqual))
|
|
}
|
|
|
|
func (s *MessengerSyncSavedAddressesSuite) TestSyncSavedAddresses() {
|
|
var isTestChain1 bool = true
|
|
var testAddress1 = common.Address{1}
|
|
|
|
// Add saved addresses to main device
|
|
sa1 := wallet.SavedAddress{
|
|
Address: testAddress1,
|
|
Name: "TestC1A1",
|
|
Favourite: false,
|
|
IsTest: isTestChain1,
|
|
}
|
|
sa2 := wallet.SavedAddress{
|
|
ENSName: "test.ens.eth",
|
|
Name: "TestC1A2",
|
|
Favourite: true,
|
|
IsTest: isTestChain1,
|
|
}
|
|
|
|
err := s.main.UpsertSavedAddress(context.Background(), sa1)
|
|
s.Require().NoError(err)
|
|
err = s.main.UpsertSavedAddress(context.Background(), sa2)
|
|
s.Require().NoError(err)
|
|
|
|
// Wait and check that saved addresses are synced
|
|
_, err = WaitOnMessengerResponse(
|
|
s.other,
|
|
func(r *MessengerResponse) bool {
|
|
if len(r.SavedAddresses()) == 2 {
|
|
sas := r.SavedAddresses()
|
|
s.Require().True(haveSameElements([]wallet.SavedAddress{sa1, sa2}, []wallet.SavedAddress{*sas[0], *sas[1]}, savedAddressDataIsEqual))
|
|
return true
|
|
}
|
|
return false
|
|
},
|
|
"expected to receive two changes",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
savedAddresses, err := s.other.savedAddressesManager.GetSavedAddresses()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(2, len(savedAddresses))
|
|
s.Require().True(haveSameElements([]wallet.SavedAddress{sa1, sa2}, savedAddresses, savedAddressDataIsEqual))
|
|
}
|
|
|
|
func (s *MessengerSyncSavedAddressesSuite) testSyncDeletesOfSavedAddressesWithTestModes(testModeMain bool, testModeOther bool) {
|
|
var isTestChain1 bool = true
|
|
var isTestChain2 bool = false
|
|
var testAddress1 = common.Address{1}
|
|
|
|
// Add saved addresses to main device
|
|
sa1 := wallet.SavedAddress{
|
|
Address: testAddress1,
|
|
Name: "TestC1A1",
|
|
Favourite: false,
|
|
IsTest: isTestChain1,
|
|
}
|
|
sa2 := wallet.SavedAddress{
|
|
ENSName: "test.ens.eth",
|
|
Name: "TestC1A2",
|
|
Favourite: true,
|
|
IsTest: isTestChain2,
|
|
}
|
|
|
|
err := s.main.settings.SaveSettingField(settings.TestNetworksEnabled, testModeMain)
|
|
s.Require().NoError(err)
|
|
err = s.other.settings.SaveSettingField(settings.TestNetworksEnabled, testModeOther)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NoError(err)
|
|
err = s.main.UpsertSavedAddress(context.Background(), sa1)
|
|
s.Require().NoError(err)
|
|
err = s.main.UpsertSavedAddress(context.Background(), sa2)
|
|
s.Require().NoError(err)
|
|
|
|
// Wait and check that saved addresses are synced
|
|
_, err = WaitOnMessengerResponse(
|
|
s.other,
|
|
func(r *MessengerResponse) bool {
|
|
if len(r.SavedAddresses()) == 2 {
|
|
sas := r.SavedAddresses()
|
|
s.Require().True(haveSameElements([]wallet.SavedAddress{sa1, sa2}, []wallet.SavedAddress{*sas[0], *sas[1]}, savedAddressDataIsEqual))
|
|
return true
|
|
}
|
|
return false
|
|
},
|
|
"expected to receive two changes",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
savedAddresses, err := s.other.savedAddressesManager.GetSavedAddresses()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(2, len(savedAddresses))
|
|
|
|
// Delete saved addresses with test mode = true and sync with the other device
|
|
err = s.main.DeleteSavedAddress(context.Background(), sa1.Address, sa1.ENSName, sa1.IsTest)
|
|
s.Require().NoError(err)
|
|
|
|
// Wait and check that saved addresses are synced
|
|
_, err = WaitOnMessengerResponse(
|
|
s.other,
|
|
func(r *MessengerResponse) bool {
|
|
if len(r.SavedAddresses()) == 1 {
|
|
sa := r.SavedAddresses()[0]
|
|
// We expect the deleted event to report address, ens, isTest
|
|
s.Require().Equal(sa1.Address, sa.Address)
|
|
s.Require().Equal(sa1.ENSName, sa.ENSName)
|
|
s.Require().Equal(sa1.IsTest, sa.IsTest)
|
|
s.Require().Equal("", sa.Name)
|
|
s.Require().Equal(false, sa.Favourite)
|
|
return true
|
|
}
|
|
return false
|
|
},
|
|
"expected to receive one change",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
savedAddresses, err = s.other.savedAddressesManager.GetSavedAddresses()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(1, len(savedAddresses))
|
|
s.Require().True(haveSameElements([]wallet.SavedAddress{sa2}, savedAddresses, savedAddressDataIsEqual))
|
|
|
|
// Delete saved addresses with test mode = false and sync with the other device
|
|
err = s.main.DeleteSavedAddress(context.Background(), sa2.Address, sa2.ENSName, sa2.IsTest)
|
|
s.Require().NoError(err)
|
|
|
|
// Wait and check that saved addresses are synced
|
|
_, err = WaitOnMessengerResponse(
|
|
s.other,
|
|
func(r *MessengerResponse) bool {
|
|
if len(r.SavedAddresses()) == 1 {
|
|
sa := r.SavedAddresses()[0]
|
|
// We expect the deleted event to report address, ens, isTest
|
|
s.Require().Equal(sa2.Address, sa.Address)
|
|
s.Require().Equal(sa2.ENSName, sa.ENSName)
|
|
s.Require().Equal(sa2.IsTest, sa.IsTest)
|
|
s.Require().Equal("", sa.Name)
|
|
s.Require().Equal(false, sa.Favourite)
|
|
return true
|
|
}
|
|
return false
|
|
},
|
|
"expected to receive one change",
|
|
)
|
|
s.Require().NoError(err)
|
|
|
|
savedAddresses, err = s.other.savedAddressesManager.GetSavedAddresses()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(0, len(savedAddresses))
|
|
}
|
|
|
|
func (s *MessengerSyncSavedAddressesSuite) TestSyncDeletesOfSavedAddressesSameTestModeOnBothDevices() {
|
|
testModeMain := true
|
|
testModeOther := testModeMain
|
|
s.testSyncDeletesOfSavedAddressesWithTestModes(testModeMain, testModeOther)
|
|
}
|
|
|
|
func (s *MessengerSyncSavedAddressesSuite) TestSyncDeletesOfSavedAddressesDifferentTestModeOnDevices() {
|
|
testModeMain := true
|
|
testModeOther := !testModeMain
|
|
s.testSyncDeletesOfSavedAddressesWithTestModes(testModeMain, testModeOther)
|
|
}
|