status-go/protocol/messenger_settings_test.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)
}