package pushnotificationclient import ( "bytes" "crypto/ecdsa" "math/rand" "testing" "github.com/google/uuid" "github.com/stretchr/testify/suite" "github.com/status-im/status-go/appdatabase" "github.com/status-im/status-go/eth-node/crypto" "github.com/status-im/status-go/eth-node/crypto/ecies" "github.com/status-im/status-go/eth-node/types" "github.com/status-im/status-go/protocol/common" "github.com/status-im/status-go/protocol/protobuf" "github.com/status-im/status-go/protocol/sqlite" "github.com/status-im/status-go/protocol/tt" "github.com/status-im/status-go/t/helpers" ) const testDeviceToken = "test-token" type ClientSuite struct { suite.Suite persistence *Persistence identity *ecdsa.PrivateKey installationID string client *Client } func TestClientSuite(t *testing.T) { s := new(ClientSuite) s.installationID = "c6ae4fde-bb65-11ea-b3de-0242ac130004" suite.Run(t, s) } func (s *ClientSuite) SetupTest() { db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{}) s.Require().NoError(err) err = sqlite.Migrate(db) s.Require().NoError(err) s.persistence = NewPersistence(db) identity, err := crypto.GenerateKey() s.Require().NoError(err) s.identity = identity config := &Config{ Identity: identity, Logger: tt.MustCreateTestLogger(), RemoteNotificationsEnabled: true, InstallationID: s.installationID, } s.client = New(s.persistence, config, nil, nil) } func (s *ClientSuite) TestBuildPushNotificationRegisterMessage() { mutedChatList := []string{"a", "b"} blockedChatList := []string{"c", "d"} // build chat lish hashes var mutedChatListHashes [][]byte for _, chatID := range mutedChatList { mutedChatListHashes = append(mutedChatListHashes, common.Shake256([]byte(chatID))) } // Build Blocked chat list hashes var blockedChatListHashes [][]byte for _, chatID := range blockedChatList { blockedChatListHashes = append(blockedChatListHashes, common.Shake256([]byte(chatID))) } contactKey, err := crypto.GenerateKey() s.Require().NoError(err) contactIDs := []*ecdsa.PublicKey{&contactKey.PublicKey} options := &RegistrationOptions{ ContactIDs: contactIDs, MutedChatIDs: mutedChatList, BlockedChatIDs: blockedChatList, } // Set random generator for uuid var seed int64 = 1 uuid.SetRand(rand.New(rand.NewSource(seed))) // nolint: gosec // Get token expectedUUID := uuid.New().String() // Reset random generator uuid.SetRand(rand.New(rand.NewSource(seed))) // nolint: gosec s.client.deviceToken = testDeviceToken // Set reader s.client.reader = bytes.NewReader([]byte(expectedUUID)) registration := &protobuf.PushNotificationRegistration{ Version: 1, AccessToken: expectedUUID, DeviceToken: testDeviceToken, InstallationId: s.installationID, Enabled: true, MutedChatList: mutedChatListHashes, BlockedChatList: blockedChatListHashes, } actualMessage, err := s.client.buildPushNotificationRegistrationMessage(options) s.Require().NoError(err) s.Require().Equal(registration, actualMessage) } func (s *ClientSuite) TestBuildPushNotificationRegisterMessageAllowFromContactsOnly() { mutedChatList := []string{"a", "b"} publicChatList := []string{"c", "d"} blockedChatList := []string{"e", "f"} // build muted chat lish hashes var mutedChatListHashes [][]byte for _, chatID := range mutedChatList { mutedChatListHashes = append(mutedChatListHashes, common.Shake256([]byte(chatID))) } // build blocked chat lish hashes var blockedChatListHashes [][]byte for _, chatID := range blockedChatList { blockedChatListHashes = append(blockedChatListHashes, common.Shake256([]byte(chatID))) } // build public chat lish hashes var publicChatListHashes [][]byte for _, chatID := range publicChatList { publicChatListHashes = append(publicChatListHashes, common.Shake256([]byte(chatID))) } contactKey, err := crypto.GenerateKey() s.Require().NoError(err) contactIDs := []*ecdsa.PublicKey{&contactKey.PublicKey} options := &RegistrationOptions{ ContactIDs: contactIDs, MutedChatIDs: mutedChatList, BlockedChatIDs: blockedChatList, PublicChatIDs: publicChatList, } // Set random generator for uuid var seed int64 = 1 uuid.SetRand(rand.New(rand.NewSource(seed))) // nolint: gosec // Get token expectedUUID := uuid.New().String() // set up reader reader := bytes.NewReader([]byte(expectedUUID)) sharedKey, err := ecies.ImportECDSA(s.identity).GenerateShared( ecies.ImportECDSAPublic(&contactKey.PublicKey), accessTokenKeyLength, accessTokenKeyLength, ) s.Require().NoError(err) // build encrypted token encryptedToken, err := encryptAccessToken([]byte(expectedUUID), sharedKey, reader) s.Require().NoError(err) // Reset random generator uuid.SetRand(rand.New(rand.NewSource(seed))) // nolint: gosec s.client.config.AllowFromContactsOnly = true s.client.deviceToken = testDeviceToken // Set reader s.client.reader = bytes.NewReader([]byte(expectedUUID)) registration := &protobuf.PushNotificationRegistration{ Version: 1, AccessToken: expectedUUID, DeviceToken: testDeviceToken, InstallationId: s.installationID, AllowFromContactsOnly: true, Enabled: true, BlockedChatList: blockedChatListHashes, MutedChatList: mutedChatListHashes, AllowedKeyList: [][]byte{encryptedToken}, AllowedMentionsChatList: publicChatListHashes, } actualMessage, err := s.client.buildPushNotificationRegistrationMessage(options) s.Require().NoError(err) s.Require().Equal(registration, actualMessage) } func (s *ClientSuite) TestHandleMessageScheduled() { messageID := []byte("message-id") chatID := "chat-id" installationID1 := "1" installationID2 := "2" rawMessage := &common.RawMessage{ ID: types.EncodeHex(messageID), SendPushNotification: true, LocalChatID: chatID, } event := &common.MessageEvent{ RawMessage: rawMessage, } s.Require().NoError(s.client.handleMessageScheduled(event)) key1, err := crypto.GenerateKey() s.Require().NoError(err) // First time, should notify response, err := s.client.shouldNotifyOn(&key1.PublicKey, installationID1, messageID) s.Require().NoError(err) s.Require().True(response) // Save notification s.Require().NoError(s.client.notifiedOn(&key1.PublicKey, installationID1, messageID, chatID, protobuf.PushNotification_MESSAGE)) // Second time, should not notify response, err = s.client.shouldNotifyOn(&key1.PublicKey, installationID1, messageID) s.Require().NoError(err) s.Require().False(response) // Different installationID response, err = s.client.shouldNotifyOn(&key1.PublicKey, installationID2, messageID) s.Require().NoError(err) s.Require().True(response) key2, err := crypto.GenerateKey() s.Require().NoError(err) // different key, should notify response, err = s.client.shouldNotifyOn(&key2.PublicKey, installationID1, messageID) s.Require().NoError(err) s.Require().True(response) // non tracked message id response, err = s.client.shouldNotifyOn(&key1.PublicKey, installationID1, []byte("not-existing")) s.Require().NoError(err) s.Require().False(response) } func (s *ClientSuite) TestShouldRefreshToken() { key1, err := crypto.GenerateKey() s.Require().NoError(err) key2, err := crypto.GenerateKey() s.Require().NoError(err) key3, err := crypto.GenerateKey() s.Require().NoError(err) key4, err := crypto.GenerateKey() s.Require().NoError(err) // Contacts are added s.Require().False(s.client.shouldRefreshToken([]*ecdsa.PublicKey{&key1.PublicKey, &key2.PublicKey}, []*ecdsa.PublicKey{&key1.PublicKey, &key2.PublicKey, &key3.PublicKey, &key4.PublicKey}, true, true)) // everything the same s.Require().False(s.client.shouldRefreshToken([]*ecdsa.PublicKey{&key1.PublicKey, &key2.PublicKey}, []*ecdsa.PublicKey{&key2.PublicKey, &key1.PublicKey}, true, true)) // A contact is removed s.Require().True(s.client.shouldRefreshToken([]*ecdsa.PublicKey{&key1.PublicKey, &key2.PublicKey}, []*ecdsa.PublicKey{&key2.PublicKey}, true, true)) // allow from contacts only is disabled s.Require().False(s.client.shouldRefreshToken([]*ecdsa.PublicKey{&key1.PublicKey, &key2.PublicKey}, []*ecdsa.PublicKey{&key2.PublicKey, &key1.PublicKey}, true, false)) // allow from contacts only is enabled s.Require().True(s.client.shouldRefreshToken([]*ecdsa.PublicKey{&key1.PublicKey, &key2.PublicKey}, []*ecdsa.PublicKey{&key2.PublicKey, &key1.PublicKey}, false, true)) } func (s *ClientSuite) TestHandleMessageScheduledFromPairedDevice() { messageID := []byte("message-id") installationID1 := "1" // Should return nil response, err := s.client.shouldNotifyOn(&s.identity.PublicKey, installationID1, messageID) s.Require().NoError(err) s.Require().False(response) }