108 lines
3.5 KiB
Go
108 lines
3.5 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"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/multiaccounts/common"
|
|
"github.com/status-im/status-go/protocol/encryption/multidevice"
|
|
"github.com/status-im/status-go/protocol/requests"
|
|
"github.com/status-im/status-go/protocol/tt"
|
|
"github.com/status-im/status-go/waku"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
func TestMessengerSettings(t *testing.T) {
|
|
suite.Run(t, new(MessengerSettingsSuite))
|
|
}
|
|
|
|
type MessengerSettingsSuite struct {
|
|
MessengerBaseTestSuite
|
|
m2 *Messenger
|
|
}
|
|
|
|
func (s *MessengerSettingsSuite) 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.m, err = newMessengerWithKey(s.shh, pk, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
|
|
s.m2, err = newMessengerWithKey(s.shh, s.m.identity, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
|
|
prepareMessengersForPairing(&s.Suite, s.m, s.m2)
|
|
}
|
|
|
|
func (s *MessengerSettingsSuite) TearDownTest() {
|
|
TearDownMessenger(&s.Suite, s.m)
|
|
TearDownMessenger(&s.Suite, s.m2)
|
|
_ = s.logger.Sync()
|
|
}
|
|
|
|
func prepareMessengersForPairing(s *suite.Suite, m1, m2 *Messenger) {
|
|
// Set m's installation metadata
|
|
aim := &multidevice.InstallationMetadata{
|
|
Name: "m's-device",
|
|
DeviceType: "m's-device-type",
|
|
}
|
|
err := m1.SetInstallationMetadata(m1.installationID, aim)
|
|
s.Require().NoError(err)
|
|
|
|
// Set m 2's installation metadata
|
|
a2im := &multidevice.InstallationMetadata{
|
|
Name: "m's-other-device",
|
|
DeviceType: "m's-other-device-type",
|
|
}
|
|
err = m2.SetInstallationMetadata(m2.installationID, a2im)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func (s *MessengerSettingsSuite) TestSetCustomizationColor() {
|
|
PairDevices(&s.Suite, s.m2, s.m)
|
|
PairDevices(&s.Suite, s.m, s.m2)
|
|
|
|
s.Require().Equal(s.m.account.KeyUID, s.m2.account.KeyUID)
|
|
|
|
err := s.m.multiAccounts.SaveAccount(*s.m.account)
|
|
s.Require().NoError(err)
|
|
err = s.m2.multiAccounts.SaveAccount(*s.m2.account)
|
|
s.Require().NoError(err)
|
|
|
|
// check that accounts have no customization color
|
|
acc, err := s.m.multiAccounts.GetAccount(s.m.account.KeyUID)
|
|
s.Require().NoError(err)
|
|
acc2, err := s.m2.multiAccounts.GetAccount(s.m2.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))
|
|
|
|
err = s.m.SetCustomizationColor(context.TODO(), &requests.SetCustomizationColor{KeyUID: s.m.account.KeyUID, CustomizationColor: common.CustomizationColorBlue})
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(s.m.account.GetCustomizationColor(), common.CustomizationColorBlue)
|
|
_, err = WaitOnMessengerResponse(s.m2, func(r *MessengerResponse) bool {
|
|
return len(r.CustomizationColor) > 0
|
|
}, "message syncAccountCustomizationColor not received")
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(s.m2.account.GetCustomizationColor(), common.CustomizationColorBlue)
|
|
acc, err = s.m2.multiAccounts.GetAccount(s.m.account.KeyUID)
|
|
s.Require().NoError(err)
|
|
acc2, err = s.m2.multiAccounts.GetAccount(s.m2.account.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(common.CustomizationColorBlue, acc.CustomizationColor)
|
|
s.Require().Equal(acc.CustomizationColor, acc2.CustomizationColor)
|
|
}
|