| import assert from 'node:assert/strict' |
| import test from 'node:test' |
| import { applyPatchSetToCode, parsePatchResponse } from '../code-retry/utils' |
|
|
| test('parses multiple hunks inside one patch block', () => { |
| const result = parsePatchResponse(`[[PATCH]] |
| [[SEARCH]] |
| const a = 1 |
| [[REPLACE]] |
| const a = 2 |
| [[SEARCH]] |
| const b = 1 |
| [[REPLACE]] |
| const b = 2 |
| [[END]]`) |
|
|
| assert.equal(result.patches.length, 2) |
| assert.equal( |
| applyPatchSetToCode('const a = 1\nconst b = 1', result), |
| 'const a = 2\nconst b = 2' |
| ) |
| }) |
|
|
| test('parses multiple patch blocks in one response', () => { |
| const result = parsePatchResponse(`[[PATCH]] |
| [[SEARCH]] |
| const a = 1 |
| [[REPLACE]] |
| const a = 2 |
| [[END]] |
| [[PATCH]] |
| [[SEARCH]] |
| const b = 1 |
| [[REPLACE]] |
| const b = 2 |
| [[END]]`) |
|
|
| assert.equal(result.patches.length, 2) |
| assert.equal( |
| applyPatchSetToCode('const a = 1\nconst b = 1', result), |
| 'const a = 2\nconst b = 2' |
| ) |
| }) |
|
|
| test('rejects a control marker nested inside SEARCH content', () => { |
| assert.throws( |
| () => parsePatchResponse(`[[PATCH]] |
| [[SEARCH]] |
| const a = 1 |
| [[PATCH]] |
| [[REPLACE]] |
| const a = 2 |
| [[END]]`), |
| /nested \[\[PATCH\]\] marker/ |
| ) |
| }) |
|
|
| test('rejects a response without a PATCH block', () => { |
| assert.throws( |
| () => parsePatchResponse(`[[SEARCH]] |
| const a = 1 |
| [[REPLACE]] |
| const a = 2 |
| [[END]]`), |
| /did not contain any \[\[PATCH\]\] blocks/ |
| ) |
| }) |
|
|
| test('rejects a patch that changes nothing', () => { |
| assert.throws( |
| () => parsePatchResponse(`[[PATCH]] |
| [[SEARCH]] |
| const a = 1 |
| [[REPLACE]] |
| const a = 1 |
| [[END]]`), |
| /produced no change/ |
| ) |
| }) |
|
|
