Function TestSubmitToolResultsRejectsArchivedChat

coderd/x/chatd/chatd_test.go:9563–9602

(t *testing.T)

}

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

	db, ps := dbtestutil.NewDB(t)
	replica := newTestServer(t, db, ps, uuid.New())

	ctx := testutil.Context(t, testutil.WaitLong)
	user, org, model := seedChatDependencies(t, db)

	chat, err := replica.CreateChat(ctx, chatd.CreateOptions{
		OwnerID:            user.ID,
		OrganizationID:     org.ID,
		Title:              "submit-tool-archived",
		ModelConfigID:      model.ID,
		InitialUserContent: []codersdk.ChatMessagePart{codersdk.ChatMessageText("hello")},
	})
	require.NoError(t, err)

	err = replica.ArchiveChat(ctx, chat)
	require.NoError(t, err)

	// Set requires_action so the test exercises a realistic
	// scenario where SubmitToolResults would be called.
	_, err = db.UpdateChatStatus(ctx, database.UpdateChatStatusParams{
		ID:     chat.ID,
		Status: database.ChatStatusRequiresAction,
	})
	require.NoError(t, err)

	err = replica.SubmitToolResults(ctx, chatd.SubmitToolResultsOptions{
		ChatID:        chat.ID,
		UserID:        user.ID,
		ModelConfigID: model.ID,
		Results: []codersdk.ToolResult{{
			ToolCallID: "fake-tool-call-id",
			Output:     json.RawMessage(`{"result":"ignored"}`),
		}},
	})
	require.ErrorIs(t, err, chatd.ErrChatArchived)
}

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