(Feat): Initial Commit

internal/push/service_test.go

@@ -0,0 +1,111 @@
+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)
+	})
+}