fix process-file-block.test.ts

Author: charleslien

backend/src/__tests__/process-file-block.test.ts

@@ -8,7 +8,7 @@ import {
   mockModule,
 } from '@codebuff/common/testing/mock-modules'
 import { cleanMarkdownCodeBlock } from '@codebuff/common/util/file'
-import { afterAll, beforeAll, describe, expect, it } from 'bun:test'
+import { afterAll, beforeAll, beforeEach, describe, expect, it } from 'bun:test'
 import { applyPatch } from 'diff'
 
 import { processFileBlock } from '../process-file-block'
@@ -18,10 +18,8 @@ import type {
   AgentRuntimeScopedDeps,
 } from '@codebuff/common/types/contracts/agent-runtime'
 
-const agentRuntimeImpl: AgentRuntimeDeps = { ...TEST_AGENT_RUNTIME_IMPL }
-const agentRuntimeScopedImpl: AgentRuntimeScopedDeps = {
-  ...TEST_AGENT_RUNTIME_SCOPED_IMPL,
-}
+let agentRuntimeImpl: AgentRuntimeDeps
+let agentRuntimeScopedImpl: AgentRuntimeScopedDeps
 
 describe('processFileBlockModule', () => {
   beforeAll(() => {
@@ -50,6 +48,13 @@ describe('processFileBlockModule', () => {
     clearMockedModules()
   })
 
+  beforeEach(() => {
+    agentRuntimeImpl = { ...TEST_AGENT_RUNTIME_IMPL }
+    agentRuntimeScopedImpl = {
+      ...TEST_AGENT_RUNTIME_SCOPED_IMPL,
+    }
+  })
+
   describe('cleanMarkdownCodeBlock', () => {
     it('should remove markdown code block syntax with language tag', () => {
       const input = '```typescript\nconst x = 1;\n```'
@@ -116,6 +121,17 @@ describe('processFileBlockModule', () => {
         '  return "See you later!";\r\n' +
         '}\r\n'
 
+      agentRuntimeImpl.promptAiSdk = async ({ messages }) => {
+        if (typeof messages[0].content !== 'string') {
+          throw new Error('Expected string prompt')
+        }
+        const m = messages[0].content.match(/<update>([\s\S]*)<\/update>/)
+        if (!m) {
+          return 'Test response'
+        }
+        return m[1].trim()
+      }
+
       const result = await processFileBlock({
         ...agentRuntimeImpl,
         ...agentRuntimeScopedImpl,
@@ -177,6 +193,17 @@ describe('processFileBlockModule', () => {
       const oldContent = 'const x = 1;\r\nconst y = 2;\r\n'
       const newContent = 'const x = 1;\r\nconst z = 3;\r\n'
 
+      agentRuntimeImpl.promptAiSdk = async ({ messages }) => {
+        if (typeof messages[0].content !== 'string') {
+          throw new Error('Expected string prompt')
+        }
+        const m = messages[0].content.match(/<update>([\s\S]*)<\/update>/)
+        if (!m) {
+          return 'Test response'
+        }
+        return m[1].trim()
+      }
+
       const result = await processFileBlock({
         ...agentRuntimeImpl,
         ...agentRuntimeScopedImpl,

backend/src/fast-rewrite.ts

@@ -22,7 +22,7 @@ export async function fastRewrite(
     filePath: string
     userMessage: string | undefined
     logger: Logger
-  } & ParamsExcluding<typeof promptRelaceAI, 'initialCode' | 'messageId'> &
+  } & ParamsExcluding<typeof promptRelaceAI, 'initialCode'> &
     ParamsExcluding<typeof rewriteWithOpenAI, 'oldContent'>,
 ) {
   const { initialContent, editSnippet, filePath, userMessage, logger } = params
@@ -31,7 +31,6 @@ export async function fastRewrite(
   let response = await promptRelaceAI({
     ...params,
     initialCode: initialContent,
-    messageId,
   })
   const relaceDuration = Date.now() - relaceStartTime