Function TestPromoteQueuedFallsThroughOnStaleHeartbeat

coderd/x/chatd/chatd_test.go:10972–11041

TestPromoteQueuedFallsThroughOnStaleHeartbeat asserts a stale heartbeat takes the synchronous path so the chat does not strand in Waiting waiting on a worker that will not return.

(t *testing.T)

// heartbeat takes the synchronous path so the chat does not strand
// in Waiting waiting on a worker that will not return.
func TestPromoteQueuedFallsThroughOnStaleHeartbeat(t *testing.T) {
	t.Parallel()

	db, ps := dbtestutil.NewDB(t)
	ctx := testutil.Context(t, testutil.WaitLong)

	staleAfter := 100 * time.Millisecond
	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true})
	server := chatd.New(chatd.Config{
		Logger:                     logger,
		Database:                   db,
		ReplicaID:                  uuid.New(),
		Pubsub:                     ps,
		PendingChatAcquireInterval: testutil.WaitLong,
		InFlightChatStaleAfter:     staleAfter,
	})
	t.Cleanup(func() { require.NoError(t, server.Close()) })

	user, org, model := seedChatDependencies(t, db)

	chat, err := db.InsertChat(ctx, database.InsertChatParams{
		OrganizationID:    org.ID,
		Status:            database.ChatStatusWaiting,
		ClientType:        database.ChatClientTypeUi,
		OwnerID:           user.ID,
		Title:             "stale-heartbeat-promote-fallthrough",
		LastModelConfigID: model.ID,
	})
	require.NoError(t, err)

	// Place the chat in Running with a stale heartbeat. We do not
	// start the server's run loop, so no worker will ever pick this
	// chat up; the test isolates the fall-through decision in
	// PromoteQueued.
	deadWorker := uuid.New()
	staleTime := time.Now().Add(-2 * staleAfter)
	_, err = db.UpdateChatStatus(ctx, database.UpdateChatStatusParams{
		ID:          chat.ID,
		Status:      database.ChatStatusRunning,
		WorkerID:    uuid.NullUUID{UUID: deadWorker, Valid: true},
		StartedAt:   sql.NullTime{Time: staleTime, Valid: true},
		HeartbeatAt: sql.NullTime{Time: staleTime, Valid: true},
	})
	require.NoError(t, err)

	queued, err := server.SendMessage(ctx, chatd.SendMessageOptions{
		ChatID:       chat.ID,
		Content:      []codersdk.ChatMessagePart{codersdk.ChatMessageText("promote me")},
		BusyBehavior: chatd.SendMessageBusyBehaviorQueue,
	})
	require.NoError(t, err)
	require.True(t, queued.Queued)
	require.NotNil(t, queued.QueuedMessage)

	result, err := server.PromoteQueued(ctx, chatd.PromoteQueuedOptions{
		ChatID:          chat.ID,
		QueuedMessageID: queued.QueuedMessage.ID,
		CreatedBy:       user.ID,