status-im/status-go

package protocol

import (
    "context"
    "testing"

    "github.com/stretchr/testify/suite"

    "github.com/status-im/status-go/deprecation"
    "github.com/status-im/status-go/eth-node/crypto"
    "github.com/status-im/status-go/eth-node/types"
    multiaccountscommon "github.com/status-im/status-go/multiaccounts/common"
    "github.com/status-im/status-go/multiaccounts/settings"
    "github.com/status-im/status-go/protocol/requests"
)

func TestMessengerContactUpdateSuite(t *testing.T) {
    suite.Run(t, new(MessengerContactUpdateSuite))
}

type MessengerContactUpdateSuite struct {
    MessengerBaseTestSuite
}

func (s *MessengerContactUpdateSuite) TestReceiveContactUpdate() {
    theirName := "ens-name.stateofus.eth"

    contactID := types.EncodeHex(crypto.FromECDSAPub(&s.m.identity.PublicKey))

    theirMessenger := s.newMessenger()
    defer TearDownMessenger(&s.Suite, theirMessenger)

    // Set ENS name
    err := theirMessenger.settings.SaveSettingField(settings.PreferredName, theirName)
    s.Require().NoError(err)

    theirContactID := types.EncodeHex(crypto.FromECDSAPub(&theirMessenger.identity.PublicKey))

    response, err := theirMessenger.AddContact(context.Background(), &requests.AddContact{ID: contactID})
    s.Require().NoError(err)
    s.Require().NotNil(response)

    s.Require().Len(response.Contacts, 1)
    contact := response.Contacts[0]
    // It should add the contact
    s.Require().True(contact.added())

    if deprecation.ChatProfileDeprecated {
        // It should a one to one chat
        s.Require().Len(response.Chats(), 1)
        s.Require().False(response.Chats()[0].Active)
    } else {
        // It should create a profile chat & a one to one chat
        s.Require().Len(response.Chats(), 2)
        chats := response.Chats()
        if chats[0].ChatType == ChatTypeOneToOne {
            s.Require().False(chats[0].Active)
        } else {
            s.Require().False(chats[1].Active)
        }
    }

    //// It should a one to one chat
    //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.Contacts) > 0 },
        "contact request not received",
    )
    s.Require().NoError(err)

    receivedContact := response.Contacts[0]
    s.Require().Equal(theirName, receivedContact.EnsName)
    s.Require().False(receivedContact.ENSVerified)
    s.Require().NotEmpty(receivedContact.LastUpdated)
    s.Require().True(receivedContact.hasAddedUs())

    newPicture := "new-picture"
    err = theirMessenger.SendContactUpdates(context.Background(), newEnsName, newPicture, multiaccountscommon.CustomizationColorRed)
    s.Require().NoError(err)

    // Wait for the message to reach its destination
    response, err = WaitOnMessengerResponse(
        s.m,
        func(r *MessengerResponse) bool {
            return len(r.Contacts) > 0 && response.Contacts[0].ID == theirContactID
        },
        "contact request not received",
    )

    s.Require().NoError(err)

    receivedContact = response.Contacts[0]
    s.Require().Equal(theirContactID, receivedContact.ID)
    s.Require().Equal(newEnsName, receivedContact.EnsName)
    s.Require().False(receivedContact.ENSVerified)
    s.Require().Equal(receivedContact.CustomizationColor, multiaccountscommon.CustomizationColorRed)
    s.Require().NotEmpty(receivedContact.LastUpdated)
}

func (s *MessengerContactUpdateSuite) TestAddContact() {
    contactID := types.EncodeHex(crypto.FromECDSAPub(&s.m.identity.PublicKey))

    theirMessenger := s.newMessenger()
    defer TearDownMessenger(&s.Suite, theirMessenger)

    theirMessenger.account.CustomizationColor = multiaccountscommon.CustomizationColorSky
    response, err := theirMessenger.AddContact(context.Background(), &requests.AddContact{ID: contactID, CustomizationColor: string(multiaccountscommon.CustomizationColorRed)})
    s.Require().NoError(err)
    s.Require().NotNil(response)

    s.Require().Len(response.Contacts, 1)
    contact := response.Contacts[0]

    if deprecation.ChatProfileDeprecated {
        // It adds the one to one chat
        s.Require().Len(response.Chats(), 1)
    } else {
        // It adds the profile chat and the one to one chat
        s.Require().Len(response.Chats(), 2)
    }

    // It should add the contact
    s.Require().True(contact.added())
    s.Require().Equal(contact.CustomizationColor, multiaccountscommon.CustomizationColorRed)

    // Wait for the message to reach its destination
    response, err = WaitOnMessengerResponse(
        s.m,
        func(r *MessengerResponse) bool { return len(r.Contacts) > 0 },
        "contact request not received",
    )
    s.Require().NoError(err)

    receivedContact := response.Contacts[0]
    s.Require().NotEmpty(receivedContact.LastUpdated)
    s.Require().Equal(receivedContact.CustomizationColor, multiaccountscommon.CustomizationColorSky)
}

func (s *MessengerContactUpdateSuite) TestAddContactWithENS() {
    contactID := types.EncodeHex(crypto.FromECDSAPub(&s.m.identity.PublicKey))
    ensName := "blah.stateofus.eth"

    theirMessenger := s.newMessenger()
    defer TearDownMessenger(&s.Suite, theirMessenger)

    s.Require().NoError(theirMessenger.ENSVerified(contactID, ensName))

    response, err := theirMessenger.AddContact(context.Background(), &requests.AddContact{ID: contactID})
    s.Require().NoError(err)
    s.Require().NotNil(response)
    s.Require().Len(response.Contacts, 1)
    s.Require().Equal(ensName, response.Contacts[0].EnsName)
    s.Require().True(response.Contacts[0].ENSVerified)

    s.Require().Len(response.Contacts, 1)
    contact := response.Contacts[0]

    if deprecation.ChatProfileDeprecated {
        // It adds the one to one chat
        s.Require().Len(response.Chats(), 1)
    } else {
        // It adds the profile chat and the one to one chat
        s.Require().Len(response.Chats(), 2)
    }

    // It should add the contact
    s.Require().True(contact.added())

    // Wait for the message to reach its destination
    response, err = WaitOnMessengerResponse(
        s.m,
        func(r *MessengerResponse) bool { return len(r.Contacts) > 0 },
        "contact request not received",
    )
    s.Require().NoError(err)

    receivedContact := response.Contacts[0]
    s.Require().NotEmpty(receivedContact.LastUpdated)
}