285 lines
8.7 KiB
Go
285 lines
8.7 KiB
Go
package pushnotificationclient
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/ecdsa"
|
|
"io/ioutil"
|
|
"math/rand"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"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"
|
|
)
|
|
|
|
const testDeviceToken = "test-token"
|
|
|
|
type ClientSuite struct {
|
|
suite.Suite
|
|
tmpFile *os.File
|
|
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() {
|
|
tmpFile, err := ioutil.TempFile("", "")
|
|
s.Require().NoError(err)
|
|
s.tmpFile = tmpFile
|
|
|
|
database, err := sqlite.Open(s.tmpFile.Name(), "", sqlite.ReducedKDFIterationsNumber)
|
|
s.Require().NoError(err)
|
|
s.persistence = NewPersistence(database)
|
|
|
|
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)
|
|
}
|