Function setupE2eMocks

sdk/e2e/utils/e2e-mocks.ts:423–457

()

let mocksApplied = false

export function setupE2eMocks(): void {
  if (mocksApplied) {
    return
  }
  mocksApplied = true

  spyOn(databaseModule, 'getUserInfoFromApiKey').mockImplementation(
    async ({ fields }) =>
      Object.fromEntries(
        fields.map((field) => [field, MOCK_USER[field]]),
      ) as unknown as Awaited<
        ReturnType<typeof databaseModule.getUserInfoFromApiKey>
      >,
  )
  spyOn(databaseModule, 'fetchAgentFromDatabase').mockImplementation(
    async ({ parsedAgentId }) => buildMockAgentTemplate(parsedAgentId),
  )
  spyOn(databaseModule, 'startAgentRun').mockImplementation(
    async () => `mock-run-${Math.random().toString(36).slice(2, 10)}`,
  )
  spyOn(databaseModule, 'finishAgentRun').mockImplementation(async () => {})
  spyOn(databaseModule, 'addAgentStep').mockImplementation(
    async () => `mock-step-${Math.random().toString(36).slice(2, 10)}`,
  )

  spyOn(llmModule, 'promptAiSdkStream').mockImplementation(
    promptAiSdkStreamMock,
  )
  spyOn(llmModule, 'promptAiSdk').mockImplementation(promptAiSdkMock)
  spyOn(llmModule, 'promptAiSdkStructured').mockImplementation(
    promptAiSdkStructuredMock as typeof llmModule.promptAiSdkStructured,
  )

  spyOn(CodebuffClient.prototype, 'checkConnection').mockResolvedValue(true)
}