add timeline test

internal/logic/timelineservice/add_post_logic_test.go

@@ -0,0 +1,137 @@
+package timelineservicelogic
+
+import (
+	"app-cloudep-tweeting-service/gen_result/pb/tweeting"
+	"app-cloudep-tweeting-service/internal/domain/repository"
+	mocklib "app-cloudep-tweeting-service/internal/mock/lib"
+	mockRepo "app-cloudep-tweeting-service/internal/mock/repository"
+	"app-cloudep-tweeting-service/internal/svc"
+	"context"
+	"errors"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"go.uber.org/mock/gomock"
+)
+
+func TestAddPostLogic_AddPost(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	// 初始化 mock 依賴
+	mockTimelineRepo := mockRepo.NewMockTimelineRepository(ctrl)
+	mockValidate := mocklib.NewMockValidate(ctrl)
+
+	// 初始化服務上下文
+	svcCtx := &svc.ServiceContext{
+		TimelineRepo: mockTimelineRepo,
+		Validate:     mockValidate,
+	}
+
+	// 測試數據集
+	tests := []struct {
+		name      string
+		input     *tweeting.AddPostToTimelineReq
+		prepare   func()
+		expectErr bool
+		wantResp  *tweeting.OKResp
+	}{
+		{
+			name: "成功加入貼文",
+			input: &tweeting.AddPostToTimelineReq{
+				Uid: "user123",
+				Posts: []*tweeting.PostTimelineItem{
+					{PostId: "post1", CreatedAt: 1627890123},
+					{PostId: "post2", CreatedAt: 1627890124},
+				},
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+				// 模擬成功加入貼文
+				mockTimelineRepo.EXPECT().AddPost(gomock.Any(), repository.AddPostRequest{
+					UID: "user123",
+					PostItems: []repository.TimelineItem{
+						{PostID: "post1", Score: 1627890123},
+						{PostID: "post2", Score: 1627890124},
+					},
+				}).Return(nil).Times(1)
+			},
+			expectErr: false,
+			wantResp:  &tweeting.OKResp{},
+		},
+		{
+			name: "驗證失敗",
+			input: &tweeting.AddPostToTimelineReq{
+				Uid: "",
+			},
+			prepare: func() {
+				// 模擬驗證失敗
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(errors.New("validation failed")).Times(1)
+			},
+			expectErr: true,
+			wantResp:  nil,
+		},
+		{
+			name: "沒有貼文資料",
+			input: &tweeting.AddPostToTimelineReq{
+				Uid:   "user123",
+				Posts: []*tweeting.PostTimelineItem{},
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+			},
+			expectErr: false,
+			wantResp:  &tweeting.OKResp{},
+		},
+		{
+			name: "加入貼文失敗",
+			input: &tweeting.AddPostToTimelineReq{
+				Uid: "user123",
+				Posts: []*tweeting.PostTimelineItem{
+					{PostId: "post1", CreatedAt: 1627890123},
+				},
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+				// 模擬加入貼文失敗
+				mockTimelineRepo.EXPECT().AddPost(gomock.Any(), repository.AddPostRequest{
+					UID: "user123",
+					PostItems: []repository.TimelineItem{
+						{PostID: "post1", Score: 1627890123},
+					},
+				}).Return(errors.New("repository error")).Times(1)
+			},
+			expectErr: true,
+			wantResp:  nil,
+		},
+	}
+
+	// 執行測試
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// 設置測試環境
+			tt.prepare()
+
+			// 初始化 AddPostLogic
+			logic := AddPostLogic{
+				svcCtx: svcCtx,
+				ctx:    context.TODO(),
+			}
+
+			// 執行 AddPost
+			got, err := logic.AddPost(tt.input)
+
+			// 驗證結果
+			if tt.expectErr {
+				assert.Error(t, err)
+				assert.Nil(t, got)
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tt.wantResp, got)
+			}
+		})
+	}
+}

internal/logic/timelineservice/clear_no_more_data_flag_logic_test.go

@@ -0,0 +1,105 @@
+package timelineservicelogic
+
+import (
+	"app-cloudep-tweeting-service/gen_result/pb/tweeting"
+	mocklib "app-cloudep-tweeting-service/internal/mock/lib"
+	mockRepo "app-cloudep-tweeting-service/internal/mock/repository"
+	"app-cloudep-tweeting-service/internal/svc"
+	"context"
+	"errors"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"go.uber.org/mock/gomock"
+)
+
+func TestClearNoMoreDataFlagLogic_ClearNoMoreDataFlag(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	// 初始化 mock 依賴
+	mockTimelineRepo := mockRepo.NewMockTimelineRepository(ctrl)
+	mockValidate := mocklib.NewMockValidate(ctrl)
+
+	// 初始化服務上下文
+	svcCtx := &svc.ServiceContext{
+		TimelineRepo: mockTimelineRepo,
+		Validate:     mockValidate,
+	}
+
+	// 測試數據集
+	tests := []struct {
+		name      string
+		input     *tweeting.DoNoMoreDataReq
+		prepare   func()
+		expectErr bool
+		wantResp  *tweeting.OKResp
+	}{
+		{
+			name: "成功清除 NoMoreData 標誌",
+			input: &tweeting.DoNoMoreDataReq{
+				Uid: "user123",
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+				// 模擬成功清除 NoMoreData 標誌
+				mockTimelineRepo.EXPECT().ClearNoMoreDataFlag(gomock.Any(), "user123").Return(nil).Times(1)
+			},
+			expectErr: false,
+			wantResp:  &tweeting.OKResp{},
+		},
+		{
+			name: "驗證失敗",
+			input: &tweeting.DoNoMoreDataReq{
+				Uid: "",
+			},
+			prepare: func() {
+				// 模擬驗證失敗
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(errors.New("validation failed")).Times(1)
+			},
+			expectErr: true,
+			wantResp:  nil,
+		},
+		{
+			name: "清除 NoMoreData 標誌失敗",
+			input: &tweeting.DoNoMoreDataReq{
+				Uid: "user123",
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+				// 模擬清除 NoMoreData 標誌失敗
+				mockTimelineRepo.EXPECT().ClearNoMoreDataFlag(gomock.Any(), "user123").Return(errors.New("repository error")).Times(1)
+			},
+			expectErr: true,
+			wantResp:  nil,
+		},
+	}
+
+	// 執行測試
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// 設置測試環境
+			tt.prepare()
+
+			// 初始化 ClearNoMoreDataFlagLogic
+			logic := ClearNoMoreDataFlagLogic{
+				svcCtx: svcCtx,
+				ctx:    context.TODO(),
+			}
+
+			// 執行 ClearNoMoreDataFlag
+			got, err := logic.ClearNoMoreDataFlag(tt.input)
+
+			// 驗證結果
+			if tt.expectErr {
+				assert.Error(t, err)
+				assert.Nil(t, got)
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tt.wantResp, got)
+			}
+		})
+	}
+}

internal/logic/timelineservice/fetch_timeline_logic_test.go

@@ -0,0 +1,140 @@
+package timelineservicelogic
+
+import (
+	"app-cloudep-tweeting-service/gen_result/pb/tweeting"
+	"app-cloudep-tweeting-service/internal/domain/repository"
+	mocklib "app-cloudep-tweeting-service/internal/mock/lib"
+	mockRepo "app-cloudep-tweeting-service/internal/mock/repository"
+	"app-cloudep-tweeting-service/internal/svc"
+	"context"
+	"errors"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"go.uber.org/mock/gomock"
+)
+
+func TestFetchTimelineLogic_FetchTimeline(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	// 初始化 mock 依賴
+	mockTimelineRepo := mockRepo.NewMockTimelineRepository(ctrl)
+	mockValidate := mocklib.NewMockValidate(ctrl)
+
+	// 初始化服務上下文
+	svcCtx := &svc.ServiceContext{
+		TimelineRepo: mockTimelineRepo,
+		Validate:     mockValidate,
+	}
+
+	// 測試數據集
+	tests := []struct {
+		name      string
+		input     *tweeting.GetTimelineReq
+		prepare   func()
+		expectErr bool
+		wantResp  *tweeting.FetchTimelineResponse
+	}{
+		{
+			name: "成功獲取動態時報",
+			input: &tweeting.GetTimelineReq{
+				Uid:       "user123",
+				PageSize:  10,
+				PageIndex: 1,
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+				// 模擬成功獲取動態時報
+				mockTimelineRepo.EXPECT().FetchTimeline(gomock.Any(), repository.FetchTimelineRequest{
+					UID:       "user123",
+					PageSize:  10,
+					PageIndex: 1,
+				}).Return(repository.FetchTimelineResponse{
+					Items: []repository.TimelineItem{
+						{PostID: "post1", Score: 100},
+						{PostID: "post2", Score: 200},
+					},
+					Page: tweeting.Pager{
+						Total: 2,
+						Size:  10,
+						Index: 1,
+					},
+				}, nil).Times(1)
+			},
+			expectErr: false,
+			wantResp: &tweeting.FetchTimelineResponse{
+				Posts: []*tweeting.FetchTimelineItem{
+					{PostId: "post1", Score: 100},
+					{PostId: "post2", Score: 200},
+				},
+				Page: &tweeting.Pager{
+					Total: 2,
+					Index: 1,
+					Size:  10,
+				},
+			},
+		},
+		{
+			name: "驗證失敗",
+			input: &tweeting.GetTimelineReq{
+				Uid:       "",
+				PageSize:  10,
+				PageIndex: 1,
+			},
+			prepare: func() {
+				// 模擬驗證失敗
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(errors.New("validation failed")).Times(1)
+			},
+			expectErr: true,
+			wantResp:  nil,
+		},
+		{
+			name: "獲取動態時報失敗",
+			input: &tweeting.GetTimelineReq{
+				Uid:       "user123",
+				PageSize:  10,
+				PageIndex: 1,
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+				// 模擬獲取動態時報失敗
+				mockTimelineRepo.EXPECT().FetchTimeline(gomock.Any(), repository.FetchTimelineRequest{
+					UID:       "user123",
+					PageSize:  10,
+					PageIndex: 1,
+				}).Return(repository.FetchTimelineResponse{}, errors.New("repository error")).Times(1)
+			},
+			expectErr: true,
+			wantResp:  nil,
+		},
+	}
+
+	// 執行測試
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// 設置測試環境
+			tt.prepare()
+
+			// 初始化 FetchTimelineLogic
+			logic := FetchTimelineLogic{
+				svcCtx: svcCtx,
+				ctx:    context.TODO(),
+			}
+
+			// 執行 FetchTimeline
+			got, err := logic.FetchTimeline(tt.input)
+
+			// 驗證結果
+			if tt.expectErr {
+				assert.Error(t, err)
+				assert.Nil(t, got)
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tt.wantResp, got)
+			}
+		})
+	}
+}

internal/logic/timelineservice/has_no_more_data_logic_test.go

@@ -0,0 +1,107 @@
+package timelineservicelogic
+
+import (
+	"app-cloudep-tweeting-service/gen_result/pb/tweeting"
+	mocklib "app-cloudep-tweeting-service/internal/mock/lib"
+	mockRepo "app-cloudep-tweeting-service/internal/mock/repository"
+	"app-cloudep-tweeting-service/internal/svc"
+	"context"
+	"errors"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"go.uber.org/mock/gomock"
+)
+
+func TestHasNoMoreDataLogic_HasNoMoreData(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	// 初始化 mock 依賴
+	mockTimelineRepo := mockRepo.NewMockTimelineRepository(ctrl)
+	mockValidate := mocklib.NewMockValidate(ctrl)
+
+	// 初始化服務上下文
+	svcCtx := &svc.ServiceContext{
+		TimelineRepo: mockTimelineRepo,
+		Validate:     mockValidate,
+	}
+
+	// 測試數據集
+	tests := []struct {
+		name      string
+		input     *tweeting.DoNoMoreDataReq
+		prepare   func()
+		expectErr bool
+		wantResp  *tweeting.HasNoMoreDataResp
+	}{
+		{
+			name: "成功檢查時間線是否完整",
+			input: &tweeting.DoNoMoreDataReq{
+				Uid: "user123",
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+				// 模擬成功檢查時間線狀態
+				mockTimelineRepo.EXPECT().HasNoMoreData(gomock.Any(), "user123").Return(true, nil).Times(1)
+			},
+			expectErr: false,
+			wantResp: &tweeting.HasNoMoreDataResp{
+				Status: true,
+			},
+		},
+		{
+			name: "驗證失敗",
+			input: &tweeting.DoNoMoreDataReq{
+				Uid: "",
+			},
+			prepare: func() {
+				// 模擬驗證失敗
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(errors.New("validation failed")).Times(1)
+			},
+			expectErr: true,
+			wantResp:  nil,
+		},
+		{
+			name: "檢查時間線狀態失敗",
+			input: &tweeting.DoNoMoreDataReq{
+				Uid: "user123",
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+				// 模擬檢查時間線狀態失敗
+				mockTimelineRepo.EXPECT().HasNoMoreData(gomock.Any(), "user123").Return(false, errors.New("repository error")).Times(1)
+			},
+			expectErr: true,
+			wantResp:  nil,
+		},
+	}
+
+	// 執行測試
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// 設置測試環境
+			tt.prepare()
+
+			// 初始化 HasNoMoreDataLogic
+			logic := HasNoMoreDataLogic{
+				svcCtx: svcCtx,
+				ctx:    context.TODO(),
+			}
+
+			// 執行 HasNoMoreData
+			got, err := logic.HasNoMoreData(tt.input)
+
+			// 驗證結果
+			if tt.expectErr {
+				assert.Error(t, err)
+				assert.Nil(t, got)
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tt.wantResp, got)
+			}
+		})
+	}
+}

internal/logic/timelineservice/set_no_more_data_flag_logic_test.go

@@ -0,0 +1,105 @@
+package timelineservicelogic
+
+import (
+	"app-cloudep-tweeting-service/gen_result/pb/tweeting"
+	mocklib "app-cloudep-tweeting-service/internal/mock/lib"
+	mockRepo "app-cloudep-tweeting-service/internal/mock/repository"
+	"app-cloudep-tweeting-service/internal/svc"
+	"context"
+	"errors"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"go.uber.org/mock/gomock"
+)
+
+func TestSetNoMoreDataFlagLogic_SetNoMoreDataFlag(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	// 初始化 mock 依賴
+	mockTimelineRepo := mockRepo.NewMockTimelineRepository(ctrl)
+	mockValidate := mocklib.NewMockValidate(ctrl)
+
+	// 初始化服務上下文
+	svcCtx := &svc.ServiceContext{
+		TimelineRepo: mockTimelineRepo,
+		Validate:     mockValidate,
+	}
+
+	// 測試數據集
+	tests := []struct {
+		name      string
+		input     *tweeting.DoNoMoreDataReq
+		prepare   func()
+		expectErr bool
+		wantResp  *tweeting.OKResp
+	}{
+		{
+			name: "成功標記時間線已完整",
+			input: &tweeting.DoNoMoreDataReq{
+				Uid: "user123",
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+				// 模擬成功標記時間線狀態
+				mockTimelineRepo.EXPECT().SetNoMoreDataFlag(gomock.Any(), "user123").Return(nil).Times(1)
+			},
+			expectErr: false,
+			wantResp:  &tweeting.OKResp{},
+		},
+		{
+			name: "驗證失敗",
+			input: &tweeting.DoNoMoreDataReq{
+				Uid: "",
+			},
+			prepare: func() {
+				// 模擬驗證失敗
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(errors.New("validation failed")).Times(1)
+			},
+			expectErr: true,
+			wantResp:  nil,
+		},
+		{
+			name: "標記時間線狀態失敗",
+			input: &tweeting.DoNoMoreDataReq{
+				Uid: "user123",
+			},
+			prepare: func() {
+				// 模擬驗證通過
+				mockValidate.EXPECT().ValidateAll(gomock.Any()).Return(nil).Times(1)
+				// 模擬標記時間線狀態失敗
+				mockTimelineRepo.EXPECT().SetNoMoreDataFlag(gomock.Any(), "user123").Return(errors.New("repository error")).Times(1)
+			},
+			expectErr: true,
+			wantResp:  nil,
+		},
+	}
+
+	// 執行測試
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// 設置測試環境
+			tt.prepare()
+
+			// 初始化 SetNoMoreDataFlagLogic
+			logic := SetNoMoreDataFlagLogic{
+				svcCtx: svcCtx,
+				ctx:    context.TODO(),
+			}
+
+			// 執行 SetNoMoreDataFlag
+			got, err := logic.SetNoMoreDataFlag(tt.input)
+
+			// 驗證結果
+			if tt.expectErr {
+				assert.Error(t, err)
+				assert.Nil(t, got)
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tt.wantResp, got)
+			}
+		})
+	}
+}