Li Liu
27 lines
897 B
TypeScript
27 lines
897 B
TypeScript
import { describe, it, expect } from 'vitest'
|
|
import { existsSync } from 'fs'
|
|
import { preparePromptForCli } from '../../src/utils/prompt-file.js'
|
|
|
|
describe('preparePromptForCli', () => {
|
|
it('should return original prompt when under threshold', () => {
|
|
const result = preparePromptForCli('small prompt')
|
|
expect(result.prompt).toBe('small prompt')
|
|
result.cleanup()
|
|
})
|
|
|
|
it('should write large prompts to temp file and cleanup', () => {
|
|
const largePrompt = 'x'.repeat(200 * 1024)
|
|
const result = preparePromptForCli(largePrompt)
|
|
expect(result.prompt).toContain('file')
|
|
expect(result.prompt).not.toBe(largePrompt)
|
|
|
|
const pathMatch = result.prompt.match(/\/.*magpie_prompt_\S+/)
|
|
expect(pathMatch).toBeTruthy()
|
|
const tmpPath = pathMatch![0]
|
|
expect(existsSync(tmpPath)).toBe(true)
|
|
|
|
result.cleanup()
|
|
expect(existsSync(tmpPath)).toBe(false)
|
|
})
|
|
})
|