test: Cover notification service

internal/queue/client.go

@@ -5,6 +5,10 @@ import (
 	"github.com/learn-video/dionysia/internal/config"
 )
 
-func NewClient(cfg *config.Config) *asynq.Client {
+type Client interface {
+	Enqueue(*asynq.Task, ...asynq.Option) (*asynq.TaskInfo, error)
+}
+
+func NewClient(cfg *config.Config) Client {
 	return asynq.NewClient(asynq.RedisClientOpt{Addr: cfg.RedisAddr})
 }

internal/service/notification.go

@@ -4,7 +4,7 @@ import (
 	"log"
 
 	"github.com/google/uuid"
-	"github.com/hibiken/asynq"
+	"github.com/learn-video/dionysia/internal/queue"
 	"github.com/learn-video/dionysia/internal/task"
 )
 
@@ -13,10 +13,10 @@ type NotificationHandler interface {
 }
 
 type notificationHandler struct {
-	client *asynq.Client
+	client queue.Client
 }
 
-func NewNotificationHandler(c *asynq.Client) NotificationHandler {
+func NewNotificationHandler(c queue.Client) NotificationHandler {
 	return &notificationHandler{
 		client: c,
 	}

internal/service/notification_test.go

@@ -0,0 +1,49 @@
+package service_test
+
+import (
+	"testing"
+
+	"github.com/google/uuid"
+	"github.com/hibiken/asynq"
+	"github.com/learn-video/dionysia/internal/mocks"
+	"github.com/learn-video/dionysia/internal/service"
+	"github.com/learn-video/dionysia/internal/task"
+	"github.com/stretchr/testify/assert"
+	"go.uber.org/mock/gomock"
+)
+
+func TestPackageStreamSuccess(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	mockClient := mocks.NewMockClient(ctrl)
+	handler := service.NewNotificationHandler(mockClient)
+
+	taskID := uuid.New()
+	expectedTask, _ := task.NewPackageTask(taskID)
+
+	expectedInfo := &asynq.TaskInfo{ID: "1", Queue: "default"}
+
+	mockClient.EXPECT().Enqueue(expectedTask).Return(expectedInfo, nil).Times(1)
+
+	err := handler.PackageStream(taskID)
+
+	assert.NoError(t, err)
+}
+
+func TestPackageStreamEnqueueFailure(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	mockClient := mocks.NewMockClient(ctrl)
+	handler := service.NewNotificationHandler(mockClient)
+
+	taskID := uuid.New()
+	expectedTask, _ := task.NewPackageTask(taskID)
+
+	mockClient.EXPECT().Enqueue(expectedTask).Return(nil, assert.AnError).Times(1)
+
+	err := handler.PackageStream(taskID)
+
+	assert.Error(t, err)
+}