status-im/status-go

package protocol

import (
    "context"
    "crypto/ecdsa"
    "errors"
    "testing"

    "github.com/stretchr/testify/suite"
    "go.uber.org/zap"

    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/eth-node/types"
    "github.com/status-im/status-go/protocol/common"
    "github.com/status-im/status-go/protocol/encryption/multidevice"
    "github.com/status-im/status-go/protocol/tt"
    "github.com/status-im/status-go/waku"
)

const publicChatName = "status"

func TestMessengerSyncChatSuite(t *testing.T) {
    suite.Run(t, new(MessengerSyncChatSuite))
}

type MessengerSyncChatSuite struct {
    suite.Suite
    privateKey *ecdsa.PrivateKey
    alice1     *Messenger
    alice2     *Messenger
    // If one wants to send messages between different instances of Messenger,
    // a single Waku service should be shared.
    shh    types.Waku
    logger *zap.Logger
}

func (s *MessengerSyncChatSuite) newMessenger() *Messenger {
    if s.privateKey == nil {
        privateKey, err := crypto.GenerateKey()
        s.Require().NoError(err)

        s.privateKey = privateKey
    }

    messenger, err := newMessengerWithKey(s.shh, s.privateKey, s.logger, nil)
    s.Require().NoError(err)
    return messenger
}

func (s *MessengerSyncChatSuite) otherNewMessenger() *Messenger {
    privateKey, err := crypto.GenerateKey()
    s.Require().NoError(err)

    messenger, err := newMessengerWithKey(s.shh, privateKey, s.logger, nil)
    s.Require().NoError(err)
    return messenger
}

func (s *MessengerSyncChatSuite) 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())

    s.alice1 = s.newMessenger()
    s.alice2 = s.newMessenger()
}

func (s *MessengerSyncChatSuite) TearDownTest() {
    TearDownMessenger(&s.Suite, s.alice1)
    TearDownMessenger(&s.Suite, s.alice2)
    _ = s.logger.Sync()
}

func (s *MessengerSyncChatSuite) Pair() {
    err := s.alice2.SetInstallationMetadata(s.alice2.installationID, &multidevice.InstallationMetadata{
        Name:       "alice2",
        DeviceType: "alice2",
    })
    s.Require().NoError(err)
    response, err := s.alice2.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.alice1,
        func(r *MessengerResponse) bool { return len(r.Installations()) > 0 },
        "installation not received",
    )

    s.Require().NoError(err)
    actualInstallation := response.Installations()[0]
    s.Require().Equal(s.alice2.installationID, actualInstallation.ID)
    s.Require().NotNil(actualInstallation.InstallationMetadata)
    s.Require().Equal("alice2", actualInstallation.InstallationMetadata.Name)
    s.Require().Equal("alice2", actualInstallation.InstallationMetadata.DeviceType)

    err = s.alice1.EnableInstallation(s.alice2.installationID)
    s.Require().NoError(err)
}

func (s *MessengerSyncChatSuite) TestRemovePubChat() {
    chat := CreatePublicChat(publicChatName, s.alice1.transport)
    err := s.alice1.SaveChat(chat)
    s.Require().NoError(err)

    chat = CreatePublicChat(publicChatName, s.alice2.transport)
    err = s.alice2.SaveChat(chat)
    s.Require().NoError(err)

    s.Pair()

    _, err = s.alice1.deactivateChat(publicChatName, 0, true, true)
    s.Require().NoError(err)

    var allChats []*Chat
    // Wait for the message to reach its destination
    err = tt.RetryWithBackOff(func() error {
        var err error
        response, err := s.alice2.RetrieveAll()
        if err != nil {
            return err
        }

        allChats = append(allChats, response.Chats()...)
        if len(allChats) >= 1 {
            return nil
        }

        return errors.New("Not received all chats & contacts")
    })
    s.Require().NoError(err)

    var statusChat *Chat
    for _, c := range allChats {
        if c.ID == publicChatName {
            statusChat = c
        }
    }

    s.Require().NotNil(statusChat)
    s.Require().False(statusChat.Active)
}

func (s *MessengerSyncChatSuite) TestMarkChatMessagesRead() {
    s.Pair()
    chatID := "foobarsynctest"
    _, err := s.alice1.createPublicChat(chatID, &MessengerResponse{})
    s.Require().NoError(err)

    _, err = s.alice2.createPublicChat(chatID, &MessengerResponse{})
    s.Require().NoError(err)

    otherMessenger := s.otherNewMessenger()
    _, err = otherMessenger.createPublicChat(chatID, &MessengerResponse{})
    s.Require().NoError(err)

    chat := otherMessenger.Chat(chatID)
    message := buildTestMessage(*chat)

    _, err = otherMessenger.SendChatMessage(context.Background(), message)
    s.Require().NoError(err)

    var receivedPubChatMessage *common.Message
    err = tt.RetryWithBackOff(func() error {
        var err error
        response, err := s.alice2.RetrieveAll()
        if err != nil {
            return err
        }

        messages := response.Messages()
        if len(messages) > 0 {
            receivedPubChatMessage = messages[0]
            return nil
        }

        return errors.New("Not received all messages")
    })
    s.Require().NoError(err)
    s.Require().Equal(receivedPubChatMessage.ChatId, chatID)

    alice2chat := s.alice2.Chat(chatID)
    s.Require().Equal(alice2chat.UnviewedMessagesCount, uint(1))

    _, err = s.alice1.MarkAllRead(context.TODO(), chatID)
    s.Require().NoError(err)

    var receivedChat *Chat
    err = tt.RetryWithBackOff(func() error {
        var err error
        response, err := s.alice2.RetrieveAll()
        if err != nil {
            return err
        }

        chats := response.Chats()
        if len(chats) > 0 {
            receivedChat = chats[0]
            return nil
        }

        return errors.New("Not received all messages")
    })
    s.Require().NoError(err)

    s.Require().Equal(receivedChat.ID, chatID)
    s.Require().Equal(receivedChat.UnviewedMessagesCount, uint(0))
}