252 lines
8.6 KiB
Go
252 lines
8.6 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
utils "github.com/status-im/status-go/common"
|
|
"github.com/status-im/status-go/multiaccounts/accounts"
|
|
"github.com/status-im/status-go/protocol/encryption/multidevice"
|
|
"github.com/status-im/status-go/protocol/tt"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
const testDisplayName = "My New Display Name"
|
|
|
|
func TestMessengerProfileDisplayNameHandlerSuite(t *testing.T) {
|
|
suite.Run(t, new(MessengerProfileDisplayNameHandlerSuite))
|
|
}
|
|
|
|
type MessengerProfileDisplayNameHandlerSuite struct {
|
|
MessengerBaseTestSuite
|
|
}
|
|
|
|
func (s *MessengerProfileDisplayNameHandlerSuite) TestDisplayNameChange() {
|
|
// check display name for the created instance
|
|
displayName, err := s.m.settings.DisplayName()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(DefaultProfileDisplayName, displayName)
|
|
|
|
// add profile keypair
|
|
profileKp := accounts.GetProfileKeypairForTest(true, false, false)
|
|
profileKp.KeyUID = s.m.account.KeyUID
|
|
profileKp.Name = DefaultProfileDisplayName
|
|
profileKp.Accounts[0].KeyUID = s.m.account.KeyUID
|
|
|
|
err = s.m.settings.SaveOrUpdateKeypair(profileKp)
|
|
s.Require().NoError(err)
|
|
|
|
// check account is present in the db
|
|
dbProfileKp, err := s.m.settings.GetKeypairByKeyUID(profileKp.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().True(accounts.SameKeypairs(profileKp, dbProfileKp))
|
|
|
|
// save account will create the account
|
|
err = s.m.multiAccounts.SaveAccount(*s.m.account)
|
|
s.Require().NoError(err)
|
|
// set new display name
|
|
err = s.m.SetDisplayName(testDisplayName)
|
|
s.Require().NoError(err)
|
|
|
|
// check display name after change - mutliaccount
|
|
multiAcc, err := s.m.multiAccounts.GetAccount(s.m.account.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(testDisplayName, multiAcc.Name)
|
|
|
|
// check display name after change - settings
|
|
displayName, err = s.m.settings.DisplayName()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(testDisplayName, displayName)
|
|
|
|
// check display name after change - keypair
|
|
dbProfileKp, err = s.m.settings.GetKeypairByKeyUID(profileKp.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(testDisplayName, dbProfileKp.Name)
|
|
}
|
|
|
|
func (s *MessengerProfileDisplayNameHandlerSuite) TestDisplayNameSync() {
|
|
// check display name for the created instance
|
|
displayName, err := s.m.settings.DisplayName()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(DefaultProfileDisplayName, displayName)
|
|
|
|
// add profile keypair
|
|
profileKp := accounts.GetProfileKeypairForTest(true, true, false)
|
|
profileKp.KeyUID = s.m.account.KeyUID
|
|
profileKp.Name = DefaultProfileDisplayName
|
|
profileKp.Accounts[0].KeyUID = s.m.account.KeyUID
|
|
profileKp.Accounts[1].KeyUID = s.m.account.KeyUID
|
|
|
|
err = s.m.settings.SaveOrUpdateKeypair(profileKp)
|
|
s.Require().NoError(err, "profile keypair alicesDevice.settings.SaveOrUpdateKeypair")
|
|
|
|
// check account is present in the db
|
|
dbProfileKp, err := s.m.settings.GetKeypairByKeyUID(profileKp.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().True(accounts.SameKeypairs(profileKp, dbProfileKp))
|
|
|
|
// Create new device and add main account to
|
|
alicesOtherDevice, err := newMessengerWithKey(s.shh, s.m.identity, s.logger, nil)
|
|
s.Require().NoError(err)
|
|
|
|
// Store only chat and default wallet account on other device
|
|
profileKpOtherDevice := accounts.GetProfileKeypairForTest(true, true, false)
|
|
profileKpOtherDevice.KeyUID = s.m.account.KeyUID
|
|
profileKpOtherDevice.Name = DefaultProfileDisplayName
|
|
profileKpOtherDevice.Accounts[0].KeyUID = s.m.account.KeyUID
|
|
profileKpOtherDevice.Accounts[1].KeyUID = s.m.account.KeyUID
|
|
|
|
err = alicesOtherDevice.settings.SaveOrUpdateKeypair(profileKpOtherDevice)
|
|
s.Require().NoError(err, "profile keypair alicesOtherDevice.settings.SaveOrUpdateKeypair")
|
|
|
|
// Check account is present in the db
|
|
dbProfileKp2, err := alicesOtherDevice.settings.GetKeypairByKeyUID(profileKpOtherDevice.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().True(accounts.SameKeypairs(profileKpOtherDevice, dbProfileKp2))
|
|
|
|
// Pair devices
|
|
im1 := &multidevice.InstallationMetadata{
|
|
Name: "alice's-other-device",
|
|
DeviceType: "alice's-other-device-type",
|
|
}
|
|
err = alicesOtherDevice.SetInstallationMetadata(alicesOtherDevice.installationID, im1)
|
|
s.Require().NoError(err)
|
|
response, err := alicesOtherDevice.SendPairInstallation(context.Background(), nil)
|
|
s.Require().NoError(err)
|
|
s.Require().NotNil(response)
|
|
s.Require().Len(response.Chats(), 1)
|
|
s.Require().False(response.Chats()[0].Active)
|
|
|
|
// Wait for the message to reach its destination
|
|
response, err = WaitOnMessengerResponse(
|
|
s.m,
|
|
func(r *MessengerResponse) bool { return len(r.Installations()) > 0 },
|
|
"installation not received",
|
|
)
|
|
|
|
s.Require().NoError(err)
|
|
actualInstallation := response.Installations()[0]
|
|
s.Require().Equal(alicesOtherDevice.installationID, actualInstallation.ID)
|
|
s.Require().NotNil(actualInstallation.InstallationMetadata)
|
|
s.Require().Equal("alice's-other-device", actualInstallation.InstallationMetadata.Name)
|
|
s.Require().Equal("alice's-other-device-type", actualInstallation.InstallationMetadata.DeviceType)
|
|
|
|
err = s.m.EnableInstallation(alicesOtherDevice.installationID)
|
|
s.Require().NoError(err)
|
|
|
|
// Set new display name on alice's device
|
|
err = s.m.SetDisplayName(testDisplayName)
|
|
s.Require().NoError(err)
|
|
|
|
err = tt.RetryWithBackOff(func() error {
|
|
response, err := alicesOtherDevice.RetrieveAll()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
foundKp := false
|
|
for _, kp := range response.Keypairs {
|
|
if kp.KeyUID == profileKp.KeyUID {
|
|
foundKp = true
|
|
}
|
|
}
|
|
|
|
foundSetting := false
|
|
for _, s := range response.Settings {
|
|
v, _ := s.Value.(string)
|
|
if v == testDisplayName {
|
|
foundSetting = true
|
|
}
|
|
}
|
|
|
|
if foundKp && foundSetting {
|
|
return nil
|
|
}
|
|
return errors.New("no sync data received")
|
|
})
|
|
s.Require().NoError(err)
|
|
|
|
// check display name after change - mutliaccount
|
|
multiAcc, err := alicesOtherDevice.multiAccounts.GetAccount(s.m.account.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(testDisplayName, multiAcc.Name)
|
|
|
|
// check display name after change - settings
|
|
displayName, err = alicesOtherDevice.settings.DisplayName()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(testDisplayName, displayName)
|
|
|
|
// check display name after change - keypair
|
|
dbProfileKp, err = alicesOtherDevice.settings.GetKeypairByKeyUID(profileKp.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(testDisplayName, dbProfileKp.Name)
|
|
}
|
|
|
|
func (s *MessengerProfileDisplayNameHandlerSuite) TestDisplayNameRestrictions() {
|
|
// check display name for the created instance
|
|
displayName, err := s.m.settings.DisplayName()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(DefaultProfileDisplayName, displayName)
|
|
|
|
// add profile keypair
|
|
profileKp := accounts.GetProfileKeypairForTest(true, false, false)
|
|
profileKp.KeyUID = s.m.account.KeyUID
|
|
profileKp.Name = DefaultProfileDisplayName
|
|
profileKp.Accounts[0].KeyUID = s.m.account.KeyUID
|
|
|
|
err = s.m.settings.SaveOrUpdateKeypair(profileKp)
|
|
s.Require().NoError(err)
|
|
|
|
setInvalidName := func(invalidName string, expectedErr error) {
|
|
err = s.m.SetDisplayName(invalidName)
|
|
s.Require().ErrorIs(err, expectedErr)
|
|
}
|
|
|
|
setInvalidName("dot.not", utils.ErrInvalidDisplayNameRegExp)
|
|
setInvalidName("t", utils.ErrInvalidDisplayNameRegExp)
|
|
setInvalidName("tt", utils.ErrInvalidDisplayNameRegExp)
|
|
setInvalidName("ttt", utils.ErrInvalidDisplayNameRegExp)
|
|
setInvalidName("tttt", utils.ErrInvalidDisplayNameRegExp)
|
|
setInvalidName("name is bigger than 24 symb", utils.ErrInvalidDisplayNameRegExp)
|
|
|
|
err = s.m.SetDisplayName("name with space")
|
|
s.Require().NoError(err)
|
|
displayName, err = s.m.settings.DisplayName()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal("name with space", displayName)
|
|
}
|
|
|
|
func (s *MessengerProfileDisplayNameHandlerSuite) TestUpdateImageWhenEnsNameIsSet() {
|
|
err := s.m.settings.SaveSetting("preferred-name", "kounkou.stateofus.eth")
|
|
s.Require().NoError(err)
|
|
|
|
// check display name for the created instance
|
|
displayName, err := s.m.settings.GetPreferredUsername()
|
|
s.Require().NoError(err)
|
|
s.Require().Equal("kounkou.stateofus.eth", displayName)
|
|
|
|
// add profile keypair
|
|
profileKp := accounts.GetProfileKeypairForTest(true, false, false)
|
|
profileKp.KeyUID = s.m.account.KeyUID
|
|
profileKp.Name = DefaultProfileDisplayName
|
|
profileKp.Accounts[0].KeyUID = s.m.account.KeyUID
|
|
|
|
err = s.m.settings.SaveOrUpdateKeypair(profileKp)
|
|
s.Require().NoError(err)
|
|
|
|
// check account is present in the db
|
|
dbProfileKp, err := s.m.settings.GetKeypairByKeyUID(profileKp.KeyUID)
|
|
s.Require().NoError(err)
|
|
s.Require().True(accounts.SameKeypairs(profileKp, dbProfileKp))
|
|
|
|
// save account will create the account
|
|
err = s.m.multiAccounts.SaveAccount(*s.m.account)
|
|
s.Require().NoError(err)
|
|
|
|
// set new description
|
|
err = s.m.SetDisplayName("godfrain.stateofus.eth")
|
|
s.Require().NoError(err)
|
|
}
|