fix(prompt-editor): fix flushSync warning in PromptEditor

lib/prompt-editor/src/PromptEditor.tsx

@@ -110,7 +110,11 @@ export const PromptEditor: FunctionComponent<Props> = ({
             setEditorState(state: SerializedPromptEditorState): void {
                 const editor = editorRef.current
                 if (editor) {
-                    editor.setEditorState(editor.parseEditorState(state.lexicalEditorState))
+                    // Use requestAnimationFrame to ensure the editor state is set outside of React's render cycle
+                    // This prevents the flushSync warning
+                    requestAnimationFrame(() => {
+                        editor.setEditorState(editor.parseEditorState(state.lexicalEditorState))
+                    })
                 }
             },
             getSerializedValue(): SerializedPromptEditorValue {
@@ -325,7 +329,11 @@ export const PromptEditor: FunctionComponent<Props> = ({
                 const currentEditorState = normalizeEditorStateJSON(editor.getEditorState().toJSON())
                 const newEditorState = initialEditorState.lexicalEditorState
                 if (!isEqual(currentEditorState, newEditorState)) {
-                    editor.setEditorState(editor.parseEditorState(newEditorState))
+                    // Use requestAnimationFrame to ensure the editor state is set outside of React's render cycle
+                    // NOTE: This addresses the flushSync warning.
+                    requestAnimationFrame(() => {
+                        editor.setEditorState(editor.parseEditorState(newEditorState))
+                    })
                 }
             }
         }

vscode/test/e2e/initial-context.test.ts

@@ -19,9 +19,9 @@
         await sidebarSignin(page, sidebar)
         const chatFrame = getChatSidebarPanel(page)
         const lastChatInput = getChatInputs(chatFrame).last()
-
+        await lastChatInput.click({ delay: 100 })
         // The current repository should be initially present in the chat input.
         await expect(chatInputMentions(lastChatInput)).toHaveText(['myrepo'])
     }
 )
 

vscode/webviews/chat/Transcript.test.tsx

@@ -1,7 +1,7 @@
 import { type ChatMessage, FIXTURE_MODELS, errorToChatError, ps } from '@sourcegraph/cody-shared'
 import { fireEvent, getQueriesForElement, render as render_, screen } from '@testing-library/react'
 import type { ComponentProps } from 'react'
-import { describe, expect, test, vi } from 'vitest'
+import { afterEach, beforeEach, describe, expect, test, vi } from 'vitest'
 import { URI } from 'vscode-uri'
 import { AppWrapperForTest } from '../AppWrapperForTest'
 import { MockNoGuardrails } from '../utils/guardrails'
@@ -20,6 +20,21 @@ const PROPS: Omit<ComponentProps<typeof Transcript>, 'transcript'> = {
     guardrails: new MockNoGuardrails(),
 }
 
+// Mock requestAnimationFrame to execute callback immediately in test environment
+const originalRAF = global.requestAnimationFrame
+beforeEach(() => {
+    global.requestAnimationFrame = (cb: FrameRequestCallback): number => {
+        // Execute the callback immediately
+        cb(0)
+        return 0
+    }
+})
+
+afterEach(() => {
+    // Restore the original requestAnimationFrame
+    global.requestAnimationFrame = originalRAF
+})
+
 vi.mock('../utils/VSCodeApi', () => ({
     getVSCodeAPI: vi.fn().mockReturnValue({
         onMessage: () => {},