Function TestProposeChatTitle_DebugRun

coderd/x/chatd/chatd_test.go:5924–6051

(t *testing.T)

}

func TestProposeChatTitle_DebugRun(t *testing.T) {
	t.Parallel()

	wantTitle := "Debug proposal title"
	tests := []struct {
		name                    string
		alwaysEnableDebugLogs   bool
		response                func() chattest.OpenAIResponse
		wantErr                 bool
		wantTitle               string
		wantTitleGenerationRuns int
		wantDebugStatus         codersdk.ChatDebugStatus
	}{
		{
			name:                  "Enabled",
			alwaysEnableDebugLogs: true,
			response: func() chattest.OpenAIResponse {
				return chattest.OpenAINonStreamingResponse(
					"{\"title\":\"" + wantTitle + "\"}",
				)
			},
			wantTitle:               wantTitle,
			wantTitleGenerationRuns: 1,
			wantDebugStatus:         codersdk.ChatDebugStatusCompleted,
		},
		{
			name:                  "Disabled",
			alwaysEnableDebugLogs: false,
			response: func() chattest.OpenAIResponse {
				return chattest.OpenAINonStreamingResponse(
					"{\"title\":\"" + wantTitle + "\"}",
				)
			},
			wantTitle: wantTitle,
		},
		{
			name:                  "GenerationErrorFinalizesDebugRun",
			alwaysEnableDebugLogs: true,
			response: func() chattest.OpenAIResponse {
				return chattest.OpenAINonStreamingResponse("not json")
			},
			wantErr:                 true,
			wantTitleGenerationRuns: 1,
			wantDebugStatus:         codersdk.ChatDebugStatusError,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			ctx := testutil.Context(t, testutil.WaitLong)
			db, ps, rawDB := dbtestutil.NewDBWithSQLDB(t)
			openAIURL := chattest.NewOpenAI(t, func(req *chattest.OpenAIRequest) chattest.OpenAIResponse {
				require.False(t, req.Stream)
				return tt.response()
			})
			user, org, model := seedChatDependenciesWithProvider(