status-im/status-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)
}