add task service

2025-12-17 10:40:23 +03:00
parent 843c50a103
commit 6537aded89
10 changed files with 697 additions and 3 deletions
@@ -0,0 +1,140 @@
+package service
+
+import (
+	"context"
+	"testing"
+
+	"datarush/internal/task/service/mocks"
+	pb "datarush/pkg/api/task"
+
+	"go.uber.org/mock/gomock"
+	"github.com/stretchr/testify/assert"
+
+	"google.golang.org/protobuf/types/known/timestamppb"
+)
+
+func TestTaskService(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	mockRepo := mocks.NewMockTaskRepository(ctrl)
+	service := NewTaskService(mockRepo)
+
+	ctx := context.Background()
+
+	t.Run("CreateTask", func(t *testing.T) {
+		task := &pb.Task{
+			CompetitionId:       "comp1",
+			Title:               "Test Task",
+			Description:         "This is a test task",
+			InCompetitionPosition: 1,
+			MaxPoints:           100,
+			MaxAttempts:         10,
+			Type:                pb.TaskType_TASK_TYPE_INPUT,
+		}
+
+		mockRepo.EXPECT().CreateTask(ctx, task).Return(task, nil)
+
+		createdTask, err := service.CreateTask(ctx, task)
+		assert.NoError(t, err)
+		assert.Equal(t, task, createdTask)
+	})
+
+	t.Run("GetTask", func(t *testing.T) {
+		taskID := "task1"
+		req := &pb.GetTaskRequest{TaskId: taskID}
+		expectedTask := &pb.Task{
+			Id:                  taskID,
+			CompetitionId:       "comp1",
+			Title:               "Test Task",
+			Description:         "This is a test task",
+			InCompetitionPosition: 1,
+			MaxPoints:           100,
+			MaxAttempts:         10,
+			Type:                pb.TaskType_TASK_TYPE_INPUT,
+			CreatedAt:           timestamppb.Now(),
+			UpdatedAt:           timestamppb.Now(),
+		}
+
+		mockRepo.EXPECT().GetTask(ctx, taskID).Return(expectedTask, nil)
+
+		task, err := service.GetTask(ctx, req)
+		assert.NoError(t, err)
+		assert.Equal(t, expectedTask, task)
+	})
+
+	t.Run("EditTask", func(t *testing.T) {
+		task := &pb.Task{
+			Id:                  "task1",
+			CompetitionId:       "comp1",
+			Title:               "Updated Test Task",
+			Description:         "This is an updated test task",
+			InCompetitionPosition: 1,
+			MaxPoints:           150,
+			MaxAttempts:         5,
+			Type:                pb.TaskType_TASK_TYPE_CHECKER,
+		}
+
+		mockRepo.EXPECT().EditTask(ctx, task).Return(task, nil)
+
+		updatedTask, err := service.EditTask(ctx, task)
+		assert.NoError(t, err)
+		assert.Equal(t, task, updatedTask)
+	})
+
+	t.Run("DeleteTask", func(t *testing.T) {
+		taskID := "task1"
+		req := &pb.DeleteTaskRequest{TaskId: taskID}
+
+		mockRepo.EXPECT().DeleteTask(ctx, taskID).Return(nil)
+
+		_, err := service.DeleteTask(ctx, req)
+		assert.NoError(t, err)
+	})
+
+	t.Run("ListCompetitionTasks", func(t *testing.T) {
+		competitionID := "comp1"
+		req := &pb.ListCompetitionTasksRequest{CompetitionId: competitionID}
+		expectedTasks := []*pb.Task{
+			{Id: "task1", CompetitionId: competitionID, Title: "Task 1"},
+			{Id: "task2", CompetitionId: competitionID, Title: "Task 2"},
+		}
+
+		mockRepo.EXPECT().ListCompetitionTasks(ctx, competitionID).Return(expectedTasks, nil)
+
+		resp, err := service.ListCompetitionTasks(ctx, req)
+		assert.NoError(t, err)
+		assert.Equal(t, expectedTasks, resp.Tasks)
+	})
+
+	t.Run("GetTaskAttachments", func(t *testing.T) {
+		taskID := "task1"
+		showPrivate := true
+		req := &pb.GetTaskAttachmentsRequest{TaskId: taskID, ShowPrivate: &showPrivate}
+		expectedAttachments := []*pb.TaskAttachment{
+			{Id: "att1", FileUrl: "url1", IsPublic: true},
+			{Id: "att2", FileUrl: "url2", IsPublic: false},
+		}
+
+		mockRepo.EXPECT().GetTaskAttachments(ctx, taskID, showPrivate).Return(expectedAttachments, nil)
+
+		resp, err := service.GetTaskAttachments(ctx, req)
+		assert.NoError(t, err)
+		assert.Equal(t, expectedAttachments, resp.Attachments)
+	})
+
+	t.Run("GetTaskAttachments - show public only", func(t *testing.T) {
+		taskID := "task1"
+		showPrivate := false
+		req := &pb.GetTaskAttachmentsRequest{TaskId: taskID, ShowPrivate: &showPrivate}
+		expectedAttachments := []*pb.TaskAttachment{
+			{Id: "att1", FileUrl: "url1", IsPublic: true},
+		}
+
+		mockRepo.EXPECT().GetTaskAttachments(ctx, taskID, showPrivate).Return(expectedAttachments, nil)
+
+		resp, err := service.GetTaskAttachments(ctx, req)
+		assert.NoError(t, err)
+		assert.Equal(t, expectedAttachments, resp.Attachments)
+	})
+}