status-go/protocol/messenger_status_updates_test.go

package protocol

import (
	"crypto/ecdsa"
	"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/protobuf"

	"github.com/status-im/status-go/protocol/tt"
	"github.com/status-im/status-go/waku"
)

func TestMessengerStatusUpdatesSuite(t *testing.T) {
	suite.Run(t, new(MessengerStatusUpdatesSuite))
}

type MessengerStatusUpdatesSuite struct {
	suite.Suite
	m          *Messenger
	privateKey *ecdsa.PrivateKey // private key for the main instance of 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 *MessengerStatusUpdatesSuite) 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.m = s.newMessenger()
	s.privateKey = s.m.identity
	_, err := s.m.Start()
	s.Require().NoError(err)

}

func (s *MessengerStatusUpdatesSuite) TearDownTest() {
	s.Require().NoError(s.m.Shutdown())
}

func (s *MessengerStatusUpdatesSuite) newMessenger() *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 *MessengerStatusUpdatesSuite) TestNextHigherClockValueOfAutomaticStatusUpdates() {

	statusUpdate1 := UserStatus{
		StatusType: int(protobuf.StatusUpdate_AUTOMATIC),
		Clock:      100,
		CustomText: "",
		PublicKey:  "pub-key1",
	}

	err := s.m.persistence.InsertStatusUpdate(statusUpdate1)
	s.Require().NoError(err)

	statusUpdate2 := UserStatus{
		StatusType: int(protobuf.StatusUpdate_AUTOMATIC),
		Clock:      200,
		CustomText: "",
		PublicKey:  "pub-key2",
	}

	err = s.m.persistence.InsertStatusUpdate(statusUpdate2)
	s.Require().NoError(err)

	statusUpdate3 := UserStatus{
		StatusType: int(protobuf.StatusUpdate_AUTOMATIC),
		Clock:      300,
		CustomText: "",
		PublicKey:  "pub-key3",
	}

	err = s.m.persistence.InsertStatusUpdate(statusUpdate3)
	s.Require().NoError(err)

	// nextClock: clock value next higher than passed clock, of status update of type StatusUpdate_AUTOMATIC
	nextClock, err := s.m.persistence.NextHigherClockValueOfAutomaticStatusUpdates(100)
	s.Require().NoError(err)

	s.Require().Equal(nextClock, uint64(200))

}

func (s *MessengerStatusUpdatesSuite) TestDeactivatedStatusUpdates() {

	statusUpdate1 := UserStatus{
		StatusType: int(protobuf.StatusUpdate_AUTOMATIC),
		Clock:      100,
		CustomText: "",
		PublicKey:  "pub-key1",
	}

	err := s.m.persistence.InsertStatusUpdate(statusUpdate1)
	s.Require().NoError(err)

	statusUpdate2 := UserStatus{
		StatusType: int(protobuf.StatusUpdate_AUTOMATIC),
		Clock:      200,
		CustomText: "",
		PublicKey:  "pub-key2",
	}

	err = s.m.persistence.InsertStatusUpdate(statusUpdate2)
	s.Require().NoError(err)

	statusUpdate3 := UserStatus{
		StatusType: int(protobuf.StatusUpdate_AUTOMATIC),
		Clock:      400,
		CustomText: "",
		PublicKey:  "pub-key3",
	}

	err = s.m.persistence.InsertStatusUpdate(statusUpdate3)
	s.Require().NoError(err)

	statusUpdate4 := UserStatus{
		StatusType: int(protobuf.StatusUpdate_AUTOMATIC),
		Clock:      400, // Adding duplicate clock value for testing
		CustomText: "",
		PublicKey:  "pub-key4",
	}

	err = s.m.persistence.InsertStatusUpdate(statusUpdate4)
	s.Require().NoError(err)

	statusUpdate5 := UserStatus{
		StatusType: int(protobuf.StatusUpdate_AUTOMATIC),
		Clock:      500,
		CustomText: "",
		PublicKey:  "pub-key5",
	}

	err = s.m.persistence.InsertStatusUpdate(statusUpdate5)
	s.Require().NoError(err)

	// Lower limit is not included, but upper limit is included
	// So every status update in this range (lowerClock upperClock] will be deactivated
	deactivatedAutomaticStatusUpdates, err := s.m.persistence.DeactivatedAutomaticStatusUpdates(100, 400)
	s.Require().NoError(err)

	count := len(deactivatedAutomaticStatusUpdates)
	s.Require().Equal(3, count)

	// Status is deactivated
	s.Require().Equal(int(protobuf.StatusUpdate_INACTIVE), deactivatedAutomaticStatusUpdates[0].StatusType)

	// Lower range starts at 201 (clock + 1)
	// (clock is bumped, so that client replaces old status update with new one)
	s.Require().Equal(uint64(201), deactivatedAutomaticStatusUpdates[0].Clock)

	//Upper rannge ends at 401 (clock + 1)
	s.Require().Equal(uint64(401), deactivatedAutomaticStatusUpdates[count-1].Clock)
}