Function testAutoPromoteQueuedMessageFallback

coderd/x/chatd/chatd_test.go:2915–3013

(t *testing.T, queuedModelConfigID uuid.NullUUID)

}

func testAutoPromoteQueuedMessageFallback(t *testing.T, queuedModelConfigID uuid.NullUUID) {
	db, ps := dbtestutil.NewDB(t)
	ctx := testutil.Context(t, testutil.WaitSuperLong)

	firstRunStarted := make(chan struct{})
	secondRunStarted := make(chan struct{}, 1)
	allowFirstRunFinish := make(chan struct{})
	var requestCount atomic.Int32
	openAIURL := chattest.NewOpenAI(t, func(req *chattest.OpenAIRequest) chattest.OpenAIResponse {
		if !req.Stream {
			return chattest.OpenAINonStreamingResponse("title")
		}

		switch requestCount.Add(1) {
		case 1:
			chunks := make(chan chattest.OpenAIChunk, 1)
			go func() {
				defer close(chunks)
				chunks <- chattest.OpenAITextChunks("first run partial")[0]
				select {
				case <-firstRunStarted:
				default:
					close(firstRunStarted)
				}
				<-allowFirstRunFinish
			}()
			return chattest.OpenAIResponse{StreamingChunks: chunks}
		default:
			select {
			case secondRunStarted <- struct{}{}:
			default:
			}
			return chattest.OpenAIStreamingResponse(chattest.OpenAITextChunks("fallback run done")...)
		}
	})

	server := newActiveTestServer(t, db, ps, func(cfg *chatd.Config) {
		// Disable periodic polling so only signalWake can
		// trigger the next processing run.
		cfg.PendingChatAcquireInterval = time.Hour
	})
	user, org, modelConfig := seedChatDependenciesWithProvider(t, db, "openai-compat", openAIURL)
	chat, err := server.CreateChat(ctx, chatd.CreateOptions{
		OrganizationID:     org.ID,
		OwnerID:            user.ID,
		Title:              "auto-promote queued fallback",
		ModelConfigID:      modelConfig.ID,
		InitialUserContent: []codersdk.ChatMessagePart{codersdk.ChatMessageText("hello")},
	})
	require.NoError(t, err)

	testutil.TryReceive(ctx, t, firstRunStarted)

	queuedContent, err := json.Marshal([]codersdk.ChatMessagePart{codersdk.ChatMessageText("legacy queued row")})
	require.NoError(t, err)
	_, err = db.InsertChatQueuedMessage(ctx, database.InsertChatQueuedMessageParams{
		ChatID:        chat.ID,
		Content:       queuedContent,