112 lines
3.4 KiB
Go
112 lines
3.4 KiB
Go
package push_test
|
|
|
|
import (
|
|
"database/sql"
|
|
"testing"
|
|
|
|
"allaboutapps.dev/aw/go-starter/internal/data/mapper"
|
|
"allaboutapps.dev/aw/go-starter/internal/models"
|
|
"allaboutapps.dev/aw/go-starter/internal/push"
|
|
"allaboutapps.dev/aw/go-starter/internal/push/provider"
|
|
"allaboutapps.dev/aw/go-starter/internal/test"
|
|
"allaboutapps.dev/aw/go-starter/internal/test/fixtures"
|
|
"github.com/aarondl/sqlboiler/v4/boil"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSendMessageSuccess(t *testing.T) {
|
|
test.WithTestPusher(t, func(service *push.Service, db *sql.DB) {
|
|
ctx := t.Context()
|
|
fix := fixtures.Fixtures()
|
|
|
|
err := service.SendToUser(ctx, mapper.LocalUserToDTO(fix.User1).Ptr(), "Hello", "World")
|
|
require.NoError(t, err)
|
|
|
|
tokenCount, err2 := fix.User1.PushTokens().Count(ctx, db)
|
|
require.NoError(t, err2)
|
|
assert.Equal(t, int64(2), tokenCount)
|
|
})
|
|
}
|
|
|
|
func TestSendMessageSuccessWithGenericError(t *testing.T) {
|
|
test.WithTestPusher(t, func(service *push.Service, db *sql.DB) {
|
|
ctx := t.Context()
|
|
fix := fixtures.Fixtures()
|
|
|
|
// provoke error from mock provider
|
|
err := service.SendToUser(ctx, mapper.LocalUserToDTO(fix.User1).Ptr(), "other error", "World")
|
|
require.NoError(t, err)
|
|
|
|
tokenCount, err2 := fix.User1.PushTokens().Count(ctx, db)
|
|
require.NoError(t, err2)
|
|
assert.Equal(t, int64(2), tokenCount)
|
|
})
|
|
}
|
|
|
|
func TestSendMessageWithInvalidToken(t *testing.T) {
|
|
test.WithTestPusher(t, func(service *push.Service, db *sql.DB) {
|
|
ctx := t.Context()
|
|
fix := fixtures.Fixtures()
|
|
|
|
user1InvalidPushToken := models.PushToken{
|
|
ID: "55c37bc8-f245-40b3-bdef-14dee35b10bd",
|
|
Token: "d5ded380-3285-4243-8a9c-72cc3f063fee",
|
|
UserID: fix.User1.ID,
|
|
Provider: models.ProviderTypeFCM,
|
|
}
|
|
err := user1InvalidPushToken.Insert(ctx, db, boil.Infer())
|
|
require.NoError(t, err)
|
|
|
|
tokenCount, err2 := fix.User1.PushTokens().Count(ctx, db)
|
|
require.NoError(t, err2)
|
|
require.Equal(t, int64(3), tokenCount)
|
|
|
|
err = service.SendToUser(ctx, mapper.LocalUserToDTO(fix.User1).Ptr(), "Hello", "World")
|
|
require.NoError(t, err)
|
|
|
|
tokenCount, err2 = fix.User1.PushTokens().Count(ctx, db)
|
|
require.NoError(t, err2)
|
|
assert.Equal(t, int64(2), tokenCount)
|
|
})
|
|
}
|
|
|
|
func TestSendMessageWithNoProvider(t *testing.T) {
|
|
test.WithTestPusher(t, func(service *push.Service, db *sql.DB) {
|
|
ctx := t.Context()
|
|
fix := fixtures.Fixtures()
|
|
|
|
service.ResetProviders()
|
|
require.Equal(t, 0, service.GetProviderCount())
|
|
|
|
err := service.SendToUser(ctx, mapper.LocalUserToDTO(fix.User1).Ptr(), "Hello", "World")
|
|
require.Error(t, err)
|
|
|
|
tokenCount, err2 := fix.User1.PushTokens().Count(ctx, db)
|
|
require.NoError(t, err2)
|
|
assert.Equal(t, int64(2), tokenCount)
|
|
})
|
|
}
|
|
|
|
func TestSendMessageWithMultipleProvider(t *testing.T) {
|
|
test.WithTestPusher(t, func(service *push.Service, db *sql.DB) {
|
|
ctx := t.Context()
|
|
fix := fixtures.Fixtures()
|
|
|
|
service.ResetProviders()
|
|
require.Equal(t, 0, service.GetProviderCount())
|
|
|
|
mockProviderFCM := provider.NewMock(push.ProviderTypeFCM)
|
|
mockProviderAPN := provider.NewMock(push.ProviderTypeAPN)
|
|
service.RegisterProvider(mockProviderAPN)
|
|
service.RegisterProvider(mockProviderFCM)
|
|
|
|
err := service.SendToUser(ctx, mapper.LocalUserToDTO(fix.User1).Ptr(), "Hello", "World")
|
|
require.NoError(t, err)
|
|
|
|
tokenCount, err2 := fix.User1.PushTokens().Count(ctx, db)
|
|
require.NoError(t, err2)
|
|
assert.Equal(t, int64(1), tokenCount)
|
|
})
|
|
}
|