test: [Coda] enhance prompt snippet coverage

(LogID: 2025102221342501009110401665250F9)

Co-Authored-By: Coda <[email protected]>

Author: kasarolzzw

backend/modules/prompt/application/convertor/prompt_test.go

@@ -394,6 +394,123 @@ func mockPromptCases() []promptTestCase {
 				},
 			},
 		},
+		{
+			name: "snippet prompt with snippets",
+			dto: &prompt.Prompt{
+				ID:          ptr.Of(int64(789)),
+				WorkspaceID: ptr.Of(int64(321)),
+				PromptKey:   ptr.Of("snippet_prompt"),
+				PromptBasic: &prompt.PromptBasic{
+					PromptType:    ptr.Of(prompt.PromptTypeSnippet),
+					DisplayName:   ptr.Of("Snippet Prompt"),
+					Description:   ptr.Of("Snippet description"),
+					LatestVersion: ptr.Of("2.0.0"),
+					CreatedBy:     ptr.Of("snippet_creator"),
+					UpdatedBy:     ptr.Of("snippet_updater"),
+					CreatedAt:     ptr.Of(nowMilli),
+					UpdatedAt:     ptr.Of(nowMilli),
+				},
+				PromptCommit: &prompt.PromptCommit{
+					CommitInfo: &prompt.CommitInfo{
+						Version:     ptr.Of("2.0.0"),
+						BaseVersion: ptr.Of("1.0.0"),
+						Description: ptr.Of("Snippet version"),
+						CommittedBy: ptr.Of("snippet_creator"),
+						CommittedAt: ptr.Of(nowMilli),
+					},
+					Detail: &prompt.PromptDetail{
+						PromptTemplate: &prompt.PromptTemplate{
+							TemplateType: ptr.Of(prompt.TemplateTypeNormal),
+							HasSnippet:   ptr.Of(true),
+							Messages: []*prompt.Message{
+								{
+									Role:    ptr.Of(prompt.RoleSystem),
+									Content: ptr.Of("Snippet content"),
+								},
+							},
+						},
+					},
+				},
+				PromptDraft: &prompt.PromptDraft{
+					DraftInfo: &prompt.DraftInfo{
+						UserID:      ptr.Of("snippet_creator"),
+						BaseVersion: ptr.Of("2.0.0"),
+						IsModified:  ptr.Of(false),
+						CreatedAt:   ptr.Of(nowMilli),
+						UpdatedAt:   ptr.Of(nowMilli),
+					},
+					Detail: &prompt.PromptDetail{
+						PromptTemplate: &prompt.PromptTemplate{
+							TemplateType: ptr.Of(prompt.TemplateTypeNormal),
+							HasSnippet:   ptr.Of(true),
+							Messages: []*prompt.Message{
+								{
+									Role:    ptr.Of(prompt.RoleUser),
+									Content: ptr.Of("Draft snippet content"),
+								},
+							},
+						},
+					},
+				},
+			},
+			do: &entity.Prompt{
+				ID:        789,
+				SpaceID:   321,
+				PromptKey: "snippet_prompt",
+				PromptBasic: &entity.PromptBasic{
+					PromptType:    entity.PromptTypeSnippet,
+					DisplayName:   "Snippet Prompt",
+					Description:   "Snippet description",
+					LatestVersion: "2.0.0",
+					CreatedBy:     "snippet_creator",
+					UpdatedBy:     "snippet_updater",
+					CreatedAt:     time.UnixMilli(nowMilli),
+					UpdatedAt:     time.UnixMilli(nowMilli),
+				},
+				PromptCommit: &entity.PromptCommit{
+					CommitInfo: &entity.CommitInfo{
+						Version:     "2.0.0",
+						BaseVersion: "1.0.0",
+						Description: "Snippet version",
+						CommittedBy: "snippet_creator",
+						CommittedAt: time.UnixMilli(nowMilli),
+					},
+					PromptDetail: &entity.PromptDetail{
+						PromptTemplate: &entity.PromptTemplate{
+							TemplateType: entity.TemplateTypeNormal,
+							HasSnippets:  true,
+							Messages: []*entity.Message{
+								{
+									Role:    entity.RoleSystem,
+									Content: ptr.Of("Snippet content"),
+								},
+							},
+						},
+					},
+				},
+				PromptDraft: &entity.PromptDraft{
+					DraftInfo: &entity.DraftInfo{
+						UserID:      "snippet_creator",
+						BaseVersion: "2.0.0",
+						IsModified:  false,
+						CreatedAt:   time.UnixMilli(nowMilli),
+						UpdatedAt:   time.UnixMilli(nowMilli),
+					},
+					PromptDetail: &entity.PromptDetail{
+						PromptTemplate: &entity.PromptTemplate{
+							TemplateType: entity.TemplateTypeNormal,
+							HasSnippets:  true,
+							Messages: []*entity.Message{
+								{
+									Role:    entity.RoleUser,
+									Content: ptr.Of("Draft snippet content"),
+								},
+							},
+						},
+					},
+				},
+			},
+		},
 	}
 }
 

backend/modules/prompt/application/debug_test.go

@@ -182,6 +182,45 @@ func TestPromptDebugApplicationImpl_DebugStreaming(t *testing.T) {
 			},
 			wantErr: nil,
 		},
+		{
+			name: "expand snippets error",
+			fieldsGetter: func(ctrl *gomock.Controller) fields {
+				mockPromptSvc := servicemocks.NewMockIPromptService(ctrl)
+				mockPromptSvc.EXPECT().ExpandSnippets(gomock.Any(), gomock.Any()).Return(errorx.New("expand error"))
+
+				mockBenefitSvc := benefitmocks.NewMockIBenefitService(ctrl)
+				mockBenefitSvc.EXPECT().CheckPromptBenefit(gomock.Any(), gomock.Any()).Return(&benefit.CheckPromptBenefitResult{}, nil)
+
+				mockAuth := rpcmocks.NewMockIAuthProvider(ctrl)
+				mockAuth.EXPECT().MCheckPromptPermission(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil)
+
+				return fields{
+					promptService:  mockPromptSvc,
+					benefitService: mockBenefitSvc,
+					auth:           mockAuth,
+				}
+			},
+			args: args{
+				ctx: session.WithCtxUser(context.Background(), mockUser),
+				req: &debug.DebugStreamingRequest{
+					Prompt: &prompt.Prompt{
+						ID:          ptr.Of(int64(123456)),
+						WorkspaceID: ptr.Of(int64(123456)),
+						PromptDraft: &prompt.PromptDraft{
+							Detail: &prompt.PromptDetail{
+								PromptTemplate: &prompt.PromptTemplate{
+									TemplateType: ptr.Of(prompt.TemplateTypeNormal),
+								},
+								ModelConfig: &prompt.ModelConfig{},
+							},
+						},
+					},
+					SingleStepDebug: ptr.Of(true),
+				},
+				stream: localstream.NewInMemStream(context.Background(), make(chan *debug.DebugStreamingResponse), make(chan error)),
+			},
+			wantErr: errorx.NewByCode(prompterr.CommonInternalErrorCode),
+		},
 		{
 			name:         "invalid param: prompt is nil",
 			fieldsGetter: func(ctrl *gomock.Controller) fields { return fields{} },

backend/modules/prompt/application/execute_test.go

@@ -20,8 +20,7 @@ import (
 	"github.com/coze-dev/coze-loop/backend/modules/prompt/domain/service"
 	servicemocks "github.com/coze-dev/coze-loop/backend/modules/prompt/domain/service/mocks"
 
-	// prompterr "github.com/coze-dev/coze-loop/backend/modules/prompt/pkg/errno"
-	// "github.com/coze-dev/coze-loop/backend/pkg/errorx"
+	"github.com/coze-dev/coze-loop/backend/pkg/errorx"
 	"github.com/coze-dev/coze-loop/backend/pkg/lang/ptr"
 	"github.com/coze-dev/coze-loop/backend/pkg/unittest"
 )
@@ -190,6 +189,31 @@ func TestPromptExecuteApplicationImpl_ExecuteInternal(t *testing.T) {
 			wantR:   execute.NewExecuteInternalResponse(),
 			wantErr: errors.New("execution error"),
 		},
+		{
+			name: "expand snippets error",
+			fieldsGetter: func(ctrl *gomock.Controller) fields {
+				mockManageRepo := repomocks.NewMockIManageRepo(ctrl)
+				mockManageRepo.EXPECT().GetPrompt(gomock.Any(), gomock.Any()).Return(createMockPrompt(), nil)
+
+				mockPromptService := servicemocks.NewMockIPromptService(ctrl)
+				mockPromptService.EXPECT().ExpandSnippets(gomock.Any(), gomock.Any()).Return(errorx.New("expand error"))
+
+				return fields{
+					promptService: mockPromptService,
+					manageRepo:    mockManageRepo,
+				}
+			},
+			args: args{
+				ctx: context.Background(),
+				req: &execute.ExecuteInternalRequest{
+					PromptID:    ptr.Of(int64(123)),
+					WorkspaceID: ptr.Of(int64(123456)),
+					Version:     ptr.Of("1.0.0"),
+				},
+			},
+			wantR:   execute.NewExecuteInternalResponse(),
+			wantErr: errorx.New("expand error"),
+		},
 		{
 			name: "success with override params",
 			fieldsGetter: func(ctrl *gomock.Controller) fields {