155 lines
6.9 KiB
JavaScript
155 lines
6.9 KiB
JavaScript
import test from 'node:test'
|
|
import assert from 'node:assert/strict'
|
|
import { makeHarness, registerBuiltins } from './harness.mjs'
|
|
import * as plugin from '../index.js'
|
|
|
|
async function setup(config = {}, model) {
|
|
const h = await makeHarness()
|
|
registerBuiltins(h)
|
|
plugin.apply(h.ctx, config)
|
|
const agent = h.newAgent(model)
|
|
return { h, agent }
|
|
}
|
|
|
|
const tagOf = (text, file) => new RegExp(`\\[${file.replace('.', '\\.')}#([0-9A-F]{4})\\]`).exec(text)?.[1]
|
|
|
|
test('read is tagged; hashline edit applies, returns fresh tag + preview + diff', async () => {
|
|
const { h, agent } = await setup()
|
|
h.write('app.py', 'def f():\n return 1\n\nprint(f())\n')
|
|
const r = await h.call(agent, 'read', { file_path: 'app.py' })
|
|
const tag = tagOf(r.text, 'app.py')
|
|
assert.ok(tag, r.text)
|
|
assert.match(r.text, /^2: return 1$/m)
|
|
const e = await h.call(agent, 'edit', { input: `[app.py#${tag}]\nPUT 2.=2:\n+ return 2` })
|
|
assert.equal(h.read('app.py'), 'def f():\n return 2\n\nprint(f())\n')
|
|
const newTag = tagOf(e.text, 'app.py')
|
|
assert.ok(newTag && newTag !== tag, e.text)
|
|
assert.match(e.text, /2: return 2/)
|
|
assert.ok(e.meta.diffs.length >= 1)
|
|
h.cleanup()
|
|
})
|
|
|
|
test('edit without read is refused (DSH observation policy)', async () => {
|
|
const { h, agent } = await setup()
|
|
h.write('x.txt', 'a\n')
|
|
await assert.rejects(h.call(agent, 'edit', { input: '[x.txt#0000]\nPUT 1.=1:\n+b' }), /never issued|read/)
|
|
h.cleanup()
|
|
})
|
|
|
|
test('stale tag after external change is recovered when anchors survive', async () => {
|
|
const { h, agent } = await setup()
|
|
h.write('s.txt', 'alpha\nbeta\ngamma\n')
|
|
const tag = tagOf((await h.call(agent, 'read', { file_path: 's.txt' })).text, 's.txt')
|
|
h.write('s.txt', 'HEADER\nalpha\nbeta\ngamma\n') // external edit
|
|
const e = await h.call(agent, 'edit', { input: `[s.txt#${tag}]\nPUT 3.=3:\n+GAMMA` })
|
|
assert.equal(h.read('s.txt'), 'HEADER\nalpha\nbeta\nGAMMA\n')
|
|
assert.match(e.text, /recovered stale tag/)
|
|
h.cleanup()
|
|
})
|
|
|
|
test('stale tag with changed anchor is rejected, nothing written', async () => {
|
|
const { h, agent } = await setup()
|
|
h.write('s.txt', 'alpha\nbeta\ngamma\n')
|
|
const tag = tagOf((await h.call(agent, 'read', { file_path: 's.txt' })).text, 's.txt')
|
|
h.write('s.txt', 'alpha\nBETA\ngamma\n')
|
|
await assert.rejects(h.call(agent, 'edit', { input: `[s.txt#${tag}]\nPUT 2.=2:\n+x` }), /anchored lines moved or changed/)
|
|
assert.equal(h.read('s.txt'), 'alpha\nBETA\ngamma\n')
|
|
h.cleanup()
|
|
})
|
|
|
|
test('multi-file move with named register, then chained edit using returned tag', async () => {
|
|
const { h, agent } = await setup()
|
|
h.write('a.js', 'function helper() {\n return 1\n}\n\nhelper()\n')
|
|
h.write('b.js', '// lib\n')
|
|
const ta = tagOf((await h.call(agent, 'read', { file_path: 'a.js' })).text, 'a.js')
|
|
const tb = tagOf((await h.call(agent, 'read', { file_path: 'b.js' })).text, 'b.js')
|
|
const e = await h.call(agent, 'edit', { input: `[a.js#${ta}]\nCUT 1* @fn\nCUT 4.=4\n[b.js#${tb}]\nPUT >$ @fn` })
|
|
assert.equal(h.read('a.js'), 'helper()\n')
|
|
assert.equal(h.read('b.js'), '// lib\nfunction helper() {\n return 1\n}\n')
|
|
const tb2 = tagOf(e.text, 'b.js')
|
|
await h.call(agent, 'edit', { input: `[b.js#${tb2}]\nPUT 3.=3:\n+ return 2` })
|
|
assert.match(h.read('b.js'), /return 2/)
|
|
h.cleanup()
|
|
})
|
|
|
|
test('grep output is tagged and counts as an observation', async () => {
|
|
const { h, agent } = await setup()
|
|
h.write('g.txt', 'one\nneedle here\nthree\n')
|
|
const g = await h.call(agent, 'grep', { pattern: 'needle' })
|
|
const tag = tagOf(g.text, 'g.txt')
|
|
assert.ok(tag, g.text)
|
|
assert.match(g.text, /^2:needle here$/m)
|
|
await h.call(agent, 'edit', { input: `[g.txt#${tag}]\nPUT 2.=2:\n+found` })
|
|
assert.equal(h.read('g.txt'), 'one\nfound\nthree\n')
|
|
h.cleanup()
|
|
})
|
|
|
|
test('REM and MV', async () => {
|
|
const { h, agent } = await setup()
|
|
h.write('old.txt', 'x\n')
|
|
h.write('gone.txt', 'y\n')
|
|
const t1 = tagOf((await h.call(agent, 'read', { file_path: 'old.txt' })).text, 'old.txt')
|
|
const t2 = tagOf((await h.call(agent, 'read', { file_path: 'gone.txt' })).text, 'gone.txt')
|
|
await h.call(agent, 'edit', { input: `[old.txt#${t1}]\nPUT 1.=1:\n+z\nMV new.txt\n[gone.txt#${t2}]\nREM` })
|
|
assert.equal(h.read('new.txt'), 'z\n')
|
|
assert.throws(() => h.read('old.txt'))
|
|
assert.throws(() => h.read('gone.txt'))
|
|
h.cleanup()
|
|
})
|
|
|
|
test('CRLF files keep CRLF', async () => {
|
|
const { h, agent } = await setup()
|
|
h.write('w.txt', 'a\r\nb\r\n')
|
|
const tag = tagOf((await h.call(agent, 'read', { file_path: 'w.txt' })).text, 'w.txt')
|
|
await h.call(agent, 'edit', { input: `[w.txt#${tag}]\nPUT 2.=2:\n+c` })
|
|
assert.equal(h.read('w.txt'), 'a\r\nc\r\n')
|
|
h.cleanup()
|
|
})
|
|
|
|
test('DeepSeek model falls back to replace mode; strict keeps hashline', async () => {
|
|
const { h, agent } = await setup({}, 'deepseek-v4')
|
|
const t = h.ctx.tools.get('edit', agent)
|
|
assert.ok(t.parameters.properties.old_string, 'replace schema')
|
|
h.write('r.txt', 'hello world\n')
|
|
await h.call(agent, 'read', { file_path: 'r.txt' })
|
|
await h.call(agent, 'edit', { file_path: 'r.txt', edits: [{ old_string: 'hello', new_string: 'goodbye' }, { old_string: 'world', new_string: 'moon' }] })
|
|
assert.equal(h.read('r.txt'), 'goodbye moon\n')
|
|
h.cleanup()
|
|
const s = await setup({ strict: true }, 'deepseek-v4')
|
|
assert.ok(s.h.ctx.tools.get('edit', s.agent).parameters.properties.input)
|
|
s.h.cleanup()
|
|
})
|
|
|
|
test('replace mode refuses when file changed since read', async () => {
|
|
const { h, agent } = await setup({ mode: 'replace' })
|
|
h.write('c.txt', 'v1\n')
|
|
await h.call(agent, 'read', { file_path: 'c.txt' })
|
|
h.write('c.txt', 'v2\n')
|
|
await assert.rejects(h.call(agent, 'edit', { file_path: 'c.txt', old_string: 'v2', new_string: 'v3' }), /changed since it was read/)
|
|
h.cleanup()
|
|
})
|
|
|
|
test('sloppy and apply_patch modes', async () => {
|
|
const s = await setup({ mode: 'sloppy' })
|
|
s.h.write('u.ts', 'const a = 1;\nrun(a);\n')
|
|
await s.h.call(s.agent, 'read', { file_path: 'u.ts' })
|
|
await s.h.call(s.agent, 'edit', { input: '*** Edit File: u.ts\n*** Find\nrun(a);\n*** Insert Before\nlog(a);' })
|
|
assert.equal(s.h.read('u.ts'), 'const a = 1;\nlog(a);\nrun(a);\n')
|
|
s.h.cleanup()
|
|
const p = await setup({ mode: 'apply_patch' })
|
|
p.h.write('m.py', 'def f():\n return 1\n')
|
|
await p.h.call(p.agent, 'read', { file_path: 'm.py' })
|
|
await p.h.call(p.agent, 'edit', { input: '*** Begin Patch\n*** Update File: m.py\n def f():\n- return 1\n+ return 2\n*** Add File: n.py\n+x = 1\n*** End Patch' })
|
|
assert.equal(p.h.read('m.py'), 'def f():\n return 2\n')
|
|
assert.equal(p.h.read('n.py'), 'x = 1\n')
|
|
p.h.cleanup()
|
|
})
|
|
|
|
test('model switch re-registers the edit tool in the agent scope', async () => {
|
|
const { h, agent } = await setup()
|
|
assert.ok(h.ctx.tools.get('edit', agent).parameters.properties.input)
|
|
await h.ctx.waterfall('agent/request', {}, async () => ({ provider: 'omni', model: 'deepseek-chat' }))
|
|
assert.ok(h.ctx.tools.get('edit', agent).parameters.properties.old_string)
|
|
h.cleanup()
|
|
})
|