status-go/protocol/pushnotificationserver/gorush_test.go

package pushnotificationserver

import (
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/status-im/status-go/eth-node/types"
	"github.com/status-im/status-go/protocol/protobuf"
)

func TestPushNotificationRegistrationToGoRushRequest(t *testing.T) {
	message1 := []byte("message-1")
	message2 := []byte("message-2")
	message3 := []byte("message-3")
	hexMessage1 := types.EncodeHex(message1)
	hexMessage2 := types.EncodeHex(message2)
	hexMessage3 := types.EncodeHex(message3)
	chatID := []byte("chat-id")
	publicKey1 := []byte("public-key-1")
	publicKey2 := []byte("public-key-2")
	installationID1 := "installation-id-1"
	installationID2 := "installation-id-2"
	installationID3 := "installation-id-3"
	var platform1 uint = 1
	var platform2 uint = 2
	var platform3 uint = 2
	token1 := "token-1"
	token2 := "token-2"
	token3 := "token-3"

	requestAndRegistrations := []*RequestAndRegistration{
		{
			Request: &protobuf.PushNotification{
				ChatId:         chatID,
				Type:           protobuf.PushNotification_MESSAGE,
				PublicKey:      publicKey1,
				InstallationId: installationID1,
				Message:        message1,
			},
			Registration: &protobuf.PushNotificationRegistration{
				DeviceToken: token1,
				TokenType:   protobuf.PushNotificationRegistration_APN_TOKEN,
			},
		},
		{
			Request: &protobuf.PushNotification{
				ChatId:         chatID,
				Type:           protobuf.PushNotification_MESSAGE,
				PublicKey:      publicKey1,
				InstallationId: installationID2,
				Message:        message2,
			},
			Registration: &protobuf.PushNotificationRegistration{
				DeviceToken: token2,
				TokenType:   protobuf.PushNotificationRegistration_FIREBASE_TOKEN,
			},
		},
		{
			Request: &protobuf.PushNotification{
				ChatId:         chatID,
				Type:           protobuf.PushNotification_MENTION,
				PublicKey:      publicKey2,
				InstallationId: installationID3,
				Message:        message3,
			},
			Registration: &protobuf.PushNotificationRegistration{
				DeviceToken: token3,
				TokenType:   protobuf.PushNotificationRegistration_FIREBASE_TOKEN,
			},
		},
	}

	expectedRequests := &GoRushRequest{
		Notifications: []*GoRushRequestNotification{
			{
				Tokens:   []string{token1},
				Platform: platform1,
				Message:  defaultNewMessageNotificationText,
				Data: &GoRushRequestData{
					EncryptedMessage: hexMessage1,
					ChatID:           types.EncodeHex(chatID),
					PublicKey:        types.EncodeHex(publicKey1),
				},
			},
			{
				Tokens:   []string{token2},
				Platform: platform2,
				Message:  defaultNewMessageNotificationText,
				Data: &GoRushRequestData{
					EncryptedMessage: hexMessage2,
					ChatID:           types.EncodeHex(chatID),
					PublicKey:        types.EncodeHex(publicKey1),
				},
			},
			{
				Tokens:   []string{token3},
				Platform: platform3,
				Message:  defaultMentionNotificationText,
				Data: &GoRushRequestData{
					EncryptedMessage: hexMessage3,
					ChatID:           types.EncodeHex(chatID),
					PublicKey:        types.EncodeHex(publicKey2),
				},
			},
		},
	}
	actualRequests := PushNotificationRegistrationToGoRushRequest(requestAndRegistrations)
	require.Equal(t, expectedRequests, actualRequests)
}