Function TestClaimCommittedParts

coderd/x/chatd/chatd_internal_test.go:6019–6130

TestClaimCommittedParts covers the per-role behavior of claimCommittedParts: - assistant messages claim every in-progress part with the committed message ID. - tool / user messages do not claim parts. - parts already claimed by an earlier assistant message are not re-claimed. - a chat with no live s

(t *testing.T)

//     re-claimed.
//   - a chat with no live state is a no-op (does not panic).
func TestClaimCommittedParts(t *testing.T) {
	t.Parallel()

	t.Run("AssistantClaimsAllInProgressParts", func(t *testing.T) {
		t.Parallel()

		server := &Server{
			logger: slogtest.Make(t, nil),
			clock:  quartz.NewMock(t),
		}
		chatID := uuid.New()
		state := server.getOrCreateStreamState(chatID)
		state.mu.Lock()
		state.buffer = []bufferedStreamPart{
			makeCommittedPart(100, "old-1"),
			makeInProgressPart("new-1"),
			makeInProgressPart("new-2"),
		}
		state.mu.Unlock()

		server.claimCommittedParts(chatID, database.ChatMessage{
			ID:   200,
			Role: database.ChatMessageRoleAssistant,
		})

		state.mu.Lock()
		defer state.mu.Unlock()
		require.Equal(t, int64(100), state.buffer[0].committedMessageID,
			"already-claimed parts must keep their original message ID")
		require.Equal(t, int64(200), state.buffer[1].committedMessageID,
			"in-progress parts must be claimed by the new message ID")
		require.Equal(t, int64(200), state.buffer[2].committedMessageID,
			"in-progress parts must be claimed by the new message ID")
	})

	t.Run("ToolMessageIsNoOp", func(t *testing.T) {
		t.Parallel()

		server := &Server{
			logger: slogtest.Make(t, nil),
			clock:  quartz.NewMock(t),
		}
		chatID := uuid.New()
		state := server.getOrCreateStreamState(chatID)
		state.mu.Lock()
		state.buffer = []bufferedStreamPart{
			makeInProgressPart("a"),
			makeInProgressPart("b"),
		}
		state.mu.Unlock()

		server.claimCommittedParts(chatID, database.ChatMessage{
			ID:   300,
			Role: database.ChatMessageRoleTool,
		})

		state.mu.Lock()
		defer state.mu.Unlock()