Function TestEditMessageRejectsUnknownModelConfig

coderd/x/chatd/chatd_test.go:9041–9087

TestEditMessageRejectsUnknownModelConfig verifies the edit handler returns ErrInvalidModelConfigID when the requested model does not exist, mirroring SendMessage's validation.

(t *testing.T)

// returns ErrInvalidModelConfigID when the requested model does not
// exist, mirroring SendMessage's validation.
func TestEditMessageRejectsUnknownModelConfig(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, modelA := seedChatDependencies(t, db)

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

	initial, err := db.GetChatMessagesByChatID(ctx, database.GetChatMessagesByChatIDParams{
		ChatID:  chat.ID,
		AfterID: 0,
	})
	require.NoError(t, err)
	require.Len(t, initial, 1)

	_, err = replica.EditMessage(ctx, chatd.EditMessageOptions{
		ChatID:          chat.ID,
		EditedMessageID: initial[0].ID,
		Content:         []codersdk.ChatMessagePart{codersdk.ChatMessageText("edited")},
		ModelConfigID:   uuid.New(),
	})
	require.ErrorIs(t, err, chatd.ErrInvalidModelConfigID)

	// The edit must roll back: the original message should still be
	// present and the chat's LastModelConfigID unchanged.
	stillThere, err := db.GetChatMessagesByChatID(ctx, database.GetChatMessagesByChatIDParams{
		ChatID:  chat.ID,
		AfterID: 0,
	})
	require.NoError(t, err)
	require.Len(t, stillThere, 1)
	require.Equal(t, initial[0].ID, stillThere[0].ID)

	storedChat, err := db.GetChatByID(ctx, chat.ID)
	require.NoError(t, err)
	require.Equal(t, modelA.ID, storedChat.LastModelConfigID)
}

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