146 lines
4.7 KiB
Go
146 lines
4.7 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/status-im/status-go/multiaccounts/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/protocol/encryption/multidevice"
|
|
"github.com/status-im/status-go/protocol/tt"
|
|
"github.com/status-im/status-go/waku"
|
|
)
|
|
|
|
func TestMessengerAccountCustomizationColor(t *testing.T) {
|
|
suite.Run(t, new(MessengerSyncAccountCustomizationColorSuite))
|
|
}
|
|
|
|
type MessengerSyncAccountCustomizationColorSuite struct {
|
|
suite.Suite
|
|
alice *Messenger
|
|
alice2 *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 *MessengerSyncAccountCustomizationColorSuite) 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())
|
|
|
|
pk, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
s.alice, err = newMessengerWithKey(s.shh, pk, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
|
|
s.alice2, err = newMessengerWithKey(s.shh, s.alice.identity, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
|
|
prepareAliceMessengersForPairing(&s.Suite, s.alice, s.alice2)
|
|
}
|
|
|
|
func (s *MessengerSyncAccountCustomizationColorSuite) TearDownTest() {
|
|
s.Require().NoError(s.alice.Shutdown())
|
|
s.Require().NoError(s.alice2.Shutdown())
|
|
_ = s.logger.Sync()
|
|
}
|
|
|
|
func pairDevices(s *suite.Suite, device1, device2 *Messenger) {
|
|
// Send pairing data
|
|
response, err := device1.SendPairInstallation(context.Background(), nil)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(response)
|
|
s.Len(response.Chats(), 1)
|
|
s.False(response.Chats()[0].Active)
|
|
|
|
i, ok := device1.allInstallations.Load(device1.installationID)
|
|
s.Require().True(ok)
|
|
|
|
// Wait for the message to reach its destination
|
|
response, err = WaitOnMessengerResponse(
|
|
device2,
|
|
func(r *MessengerResponse) bool {
|
|
for _, installation := range r.Installations {
|
|
if installation.ID == device1.installationID {
|
|
return installation.InstallationMetadata != nil &&
|
|
i.InstallationMetadata.Name == installation.InstallationMetadata.Name &&
|
|
i.InstallationMetadata.DeviceType == installation.InstallationMetadata.DeviceType
|
|
}
|
|
}
|
|
return false
|
|
|
|
},
|
|
"installation not received",
|
|
)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(response)
|
|
|
|
// Ensure installation is enabled
|
|
err = device2.EnableInstallation(device1.installationID)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func prepareAliceMessengersForPairing(s *suite.Suite, alice1, alice2 *Messenger) {
|
|
// Set Alice's installation metadata
|
|
aim := &multidevice.InstallationMetadata{
|
|
Name: "alice's-device",
|
|
DeviceType: "alice's-device-type",
|
|
}
|
|
err := alice1.SetInstallationMetadata(alice1.installationID, aim)
|
|
s.Require().NoError(err)
|
|
|
|
// Set Alice 2's installation metadata
|
|
a2im := &multidevice.InstallationMetadata{
|
|
Name: "alice's-other-device",
|
|
DeviceType: "alice's-other-device-type",
|
|
}
|
|
err = alice2.SetInstallationMetadata(alice2.installationID, a2im)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func (s *MessengerSyncAccountCustomizationColorSuite) TestSyncCustomizationColor() {
|
|
pairDevices(&s.Suite, s.alice2, s.alice)
|
|
pairDevices(&s.Suite, s.alice, s.alice2)
|
|
|
|
s.Require().Equal(s.alice.account.KeyUID, s.alice2.account.KeyUID)
|
|
|
|
err := s.alice.multiAccounts.SaveAccount(*s.alice.account)
|
|
s.Require().NoError(err)
|
|
err = s.alice2.multiAccounts.SaveAccount(*s.alice2.account)
|
|
s.Require().NoError(err)
|
|
|
|
// check that accounts have no customization color
|
|
acc, err := s.alice.multiAccounts.GetAccount(s.alice.account.KeyUID)
|
|
s.Require().NoError(err)
|
|
acc2, err := s.alice2.multiAccounts.GetAccount(s.alice2.account.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(acc.CustomizationColor, common.CustomizationColor(""))
|
|
s.Require().Equal(acc.CustomizationColorClock, uint64(0))
|
|
s.Require().Equal(acc2.CustomizationColor, common.CustomizationColor(""))
|
|
s.Require().Equal(acc2.CustomizationColorClock, uint64(0))
|
|
|
|
acc.CustomizationColor = common.CustomizationColorBlue
|
|
acc.CustomizationColorClock = 1
|
|
err = s.alice.syncAccountCustomizationColor(context.TODO(), acc)
|
|
s.Require().NoError(err)
|
|
_, err = WaitOnMessengerResponse(s.alice2, func(r *MessengerResponse) bool {
|
|
return len(r.CustomizationColor) > 0
|
|
}, "message syncAccountCustomizationColor not received")
|
|
s.Require().NoError(err)
|
|
acc2, err = s.alice2.multiAccounts.GetAccount(s.alice2.account.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(acc.CustomizationColor, acc2.CustomizationColor)
|
|
}
|