60 lines
2.4 KiB
TypeScript
60 lines
2.4 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from 'vitest'
|
|
import { access, mkdir, mkdtemp, rm, writeFile } from 'node:fs/promises'
|
|
import * as path from 'node:path'
|
|
import { tmpdir } from 'node:os'
|
|
import { removeSafeUntrackedDiscardTarget } from './git-discard-path-safety'
|
|
|
|
const tempRoots: string[] = []
|
|
|
|
async function createWorktree(): Promise<string> {
|
|
const worktreePath = await mkdtemp(path.join(tmpdir(), 'orca-discard-safety-'))
|
|
tempRoots.push(worktreePath)
|
|
return worktreePath
|
|
}
|
|
|
|
afterEach(async () => {
|
|
await Promise.all(tempRoots.splice(0).map((root) => rm(root, { recursive: true, force: true })))
|
|
})
|
|
|
|
describe('removeSafeUntrackedDiscardTarget', () => {
|
|
it('rejects empty and root targets before removing anything', async () => {
|
|
const worktreePath = await createWorktree()
|
|
const keepFile = path.join(worktreePath, 'keep.txt')
|
|
const removePath = vi.fn(async () => undefined)
|
|
await writeFile(keepFile, 'keep')
|
|
|
|
await expect(removeSafeUntrackedDiscardTarget(worktreePath, '', removePath)).rejects.toThrow(
|
|
'resolves outside the worktree'
|
|
)
|
|
await expect(removeSafeUntrackedDiscardTarget(worktreePath, '.', removePath)).rejects.toThrow(
|
|
'resolves outside the worktree'
|
|
)
|
|
await expect(
|
|
removeSafeUntrackedDiscardTarget(worktreePath, 'child/..', removePath)
|
|
).rejects.toThrow('resolves outside the worktree')
|
|
|
|
await expect(access(keepFile)).resolves.toBeUndefined()
|
|
expect(removePath).not.toHaveBeenCalled()
|
|
})
|
|
|
|
it('allows missing child paths whose nearest existing parent is the worktree', async () => {
|
|
const worktreePath = await createWorktree()
|
|
const removePath = vi.fn(async () => undefined)
|
|
await mkdir(path.join(worktreePath, 'existing'))
|
|
|
|
await expect(
|
|
removeSafeUntrackedDiscardTarget(worktreePath, 'existing/missing.txt', removePath)
|
|
).resolves.toBeUndefined()
|
|
await expect(
|
|
removeSafeUntrackedDiscardTarget(worktreePath, 'new-dir/missing.txt', removePath)
|
|
).resolves.toBeUndefined()
|
|
await expect(
|
|
removeSafeUntrackedDiscardTarget(worktreePath, 'missing.txt', removePath)
|
|
).resolves.toBeUndefined()
|
|
await expect(access(worktreePath)).resolves.toBeUndefined()
|
|
expect(removePath).toHaveBeenNthCalledWith(1, 'existing/missing.txt')
|
|
expect(removePath).toHaveBeenNthCalledWith(2, 'new-dir/missing.txt')
|
|
expect(removePath).toHaveBeenNthCalledWith(3, 'missing.txt')
|
|
})
|
|
})
|