status-go/protocol/messenger_settings_test.go

106 lines
3.3 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)
_, err = WaitOnMessengerResponse(s.m2, func(r *MessengerResponse) bool {
return len(r.CustomizationColor) > 0
}, "message syncAccountCustomizationColor not received")
s.Require().NoError(err)
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)
}