Files
orca/src/main/text-generation/pull-request-context.test.ts
T

366 lines
13 KiB
TypeScript

/* eslint-disable max-lines */
// Why: PR context generation depends on command order across remote-state
// variants; keeping the table of git command mocks together makes regressions
// easier to audit than splitting the suite by helper.
import { afterEach, describe, expect, it, vi } from 'vitest'
import { getPullRequestDraftContext } from './pull-request-context'
type GitExec = Parameters<typeof getPullRequestDraftContext>[0]
afterEach(() => {
vi.restoreAllMocks()
})
function createContextInput(base = 'main') {
return {
base,
currentTitle: 'Existing title',
currentBody: 'Existing body',
currentDraft: false
}
}
describe('getPullRequestDraftContext', () => {
it('fetches the resolved remote base before collecting PR context without mutating HEAD', async () => {
const execGit = vi.fn<GitExec>(async (args) => {
if (args[0] === 'fetch') {
return { stdout: '', stderr: '' }
}
if (args[0] === 'remote') {
return { stdout: 'origin\nupstream\n', stderr: '' }
}
if (args[0] === 'for-each-ref') {
return { stdout: 'origin/HEAD\norigin/main\nupstream/main\n', stderr: '' }
}
if (args[0] === 'branch') {
return { stdout: 'feature/pr-details\n', stderr: '' }
}
if (args[0] === 'merge-base') {
return { stdout: 'abc123\n', stderr: '' }
}
if (args[0] === 'log') {
return { stdout: '- feat: summarize branch\n', stderr: '' }
}
if (args[0] === 'diff' && args[1] === '--name-status') {
return { stdout: 'M\tsrc/file.ts\n', stderr: '' }
}
if (args[0] === 'diff') {
return { stdout: 'diff --git a/src/file.ts b/src/file.ts\n+change\n', stderr: '' }
}
throw new Error(`Unexpected git args: ${args.join(' ')}`)
})
const context = await getPullRequestDraftContext(execGit, createContextInput())
expect(context).toMatchObject({
branch: 'feature/pr-details',
base: 'main',
branchChangedByPreparation: false,
commitSummary: '- feat: summarize branch',
changeSummary: 'M\tsrc/file.ts'
})
expect(execGit).toHaveBeenCalledWith(
['fetch', '--no-tags', 'origin', '+refs/heads/main:refs/remotes/origin/main'],
expect.any(Object)
)
expect(execGit).not.toHaveBeenCalledWith(expect.arrayContaining(['rebase']), expect.anything())
expect(execGit).not.toHaveBeenCalledWith(
expect.arrayContaining(['rev-parse']),
expect.anything()
)
expect(execGit).toHaveBeenCalledWith(['merge-base', 'origin/main', 'HEAD'], expect.any(Object))
const commandNames = execGit.mock.calls.map(([args]) => args[0])
expect(commandNames.indexOf('fetch')).toBeLessThan(commandNames.indexOf('merge-base'))
})
it('fetches the preferred remote base even when the tracking ref is absent locally', async () => {
const execGit = vi.fn<GitExec>(async (args) => {
if (args[0] === 'fetch') {
return { stdout: '', stderr: '' }
}
if (args[0] === 'remote') {
return { stdout: 'origin\n', stderr: '' }
}
if (args[0] === 'for-each-ref') {
return { stdout: '', stderr: '' }
}
if (args[0] === 'branch') {
return { stdout: 'feature/pr-details\n', stderr: '' }
}
if (args[0] === 'merge-base') {
return { stdout: 'abc123\n', stderr: '' }
}
if (args[0] === 'log') {
return { stdout: '- feat: summarize branch\n', stderr: '' }
}
if (args[0] === 'diff') {
return { stdout: 'M\tREADME.md\n', stderr: '' }
}
throw new Error(`Unexpected git args: ${args.join(' ')}`)
})
await getPullRequestDraftContext(execGit, createContextInput())
expect(execGit).toHaveBeenCalledWith(
['fetch', '--no-tags', 'origin', '+refs/heads/main:refs/remotes/origin/main'],
expect.any(Object)
)
expect(execGit).not.toHaveBeenCalledWith(expect.arrayContaining(['rebase']), expect.anything())
})
it('does not fetch unrelated fork remotes before generating PR context', async () => {
const execGit = vi.fn<GitExec>(async (args) => {
if (args[0] === 'fetch') {
expect(args).not.toContain('--all')
expect(args[2]).toBe('origin')
return { stdout: '', stderr: '' }
}
if (args[0] === 'remote') {
return { stdout: 'origin\nstale-fork\n', stderr: '' }
}
if (args[0] === 'for-each-ref') {
return {
stdout: 'origin/main\nstale-fork/feature/from-stale-fork\n',
stderr: ''
}
}
if (args[0] === 'branch') {
return { stdout: 'feature/pr-details\n', stderr: '' }
}
if (args[0] === 'merge-base') {
return { stdout: 'abc123\n', stderr: '' }
}
if (args[0] === 'log') {
return { stdout: '- feat: change\n', stderr: '' }
}
if (args[0] === 'diff') {
return { stdout: 'M\tREADME.md\n', stderr: '' }
}
throw new Error(`Unexpected git args: ${args.join(' ')}`)
})
await expect(getPullRequestDraftContext(execGit, createContextInput())).resolves.toMatchObject({
branch: 'feature/pr-details'
})
expect(execGit).not.toHaveBeenCalledWith(['fetch', '--all', '--prune'], expect.any(Object))
expect(execGit).not.toHaveBeenCalledWith(
expect.arrayContaining(['stale-fork']),
expect.any(Object)
)
})
it('does not guess between multiple non-preferred remote bases for a bare base name', async () => {
const execGit = vi.fn<GitExec>(async (args) => {
if (args[0] === 'fetch') {
throw new Error(`Unexpected fetch: ${args.join(' ')}`)
}
if (args[0] === 'remote') {
return { stdout: 'contributor-a\ncontributor-b\n', stderr: '' }
}
if (args[0] === 'for-each-ref') {
return { stdout: 'contributor-a/main\ncontributor-b/main\n', stderr: '' }
}
if (args[0] === 'branch') {
return { stdout: 'feature\n', stderr: '' }
}
if (args[0] === 'merge-base') {
expect(args[1]).toBe('main')
return { stdout: 'abc123\n', stderr: '' }
}
if (args[0] === 'log') {
return { stdout: '- feat: change\n', stderr: '' }
}
if (args[0] === 'diff') {
return { stdout: 'M\tREADME.md\n', stderr: '' }
}
throw new Error(`Unexpected git args: ${args.join(' ')}`)
})
await getPullRequestDraftContext(execGit, createContextInput())
expect(execGit).not.toHaveBeenCalledWith(
expect.arrayContaining(['contributor-a']),
expect.any(Object)
)
expect(execGit).not.toHaveBeenCalledWith(
expect.arrayContaining(['contributor-b']),
expect.any(Object)
)
expect(execGit).not.toHaveBeenCalledWith(expect.arrayContaining(['rebase']), expect.anything())
})
it('reports no branch change because PR context preparation is read-only', async () => {
const execGit = vi.fn<GitExec>(async (args) => {
if (args[0] === 'fetch') {
return { stdout: '', stderr: '' }
}
if (args[0] === 'remote') {
return { stdout: 'origin\n', stderr: '' }
}
if (args[0] === 'for-each-ref') {
return { stdout: 'origin/main\n', stderr: '' }
}
if (args[0] === 'branch') {
return { stdout: 'feature\n', stderr: '' }
}
if (args[0] === 'merge-base') {
return { stdout: 'abc123\n', stderr: '' }
}
if (args[0] === 'log') {
return { stdout: '- feat: change\n', stderr: '' }
}
if (args[0] === 'diff') {
return { stdout: 'M\tREADME.md\n', stderr: '' }
}
throw new Error(`Unexpected git args: ${args.join(' ')}`)
})
const context = await getPullRequestDraftContext(execGit, createContextInput())
expect(context?.branchChangedByPreparation).toBe(false)
expect(execGit).not.toHaveBeenCalledWith(expect.arrayContaining(['rebase']), expect.anything())
expect(execGit).not.toHaveBeenCalledWith(
expect.arrayContaining(['rev-parse']),
expect.anything()
)
})
it('keeps a remote-qualified base when the selected base includes the remote', async () => {
const execGit = vi.fn<GitExec>(async (args) => {
if (args[0] === 'fetch') {
return { stdout: '', stderr: '' }
}
if (args[0] === 'remote') {
return { stdout: 'origin\nupstream\n', stderr: '' }
}
if (args[0] === 'for-each-ref') {
return { stdout: 'origin/main\nupstream/main\n', stderr: '' }
}
if (args[0] === 'branch') {
return { stdout: 'feature\n', stderr: '' }
}
if (args[0] === 'merge-base') {
return { stdout: 'abc123\n', stderr: '' }
}
if (args[0] === 'log') {
return { stdout: '- feat: change\n', stderr: '' }
}
if (args[0] === 'diff') {
return { stdout: 'M\tREADME.md\n', stderr: '' }
}
return { stdout: '', stderr: '' }
})
await getPullRequestDraftContext(execGit, createContextInput('upstream/main'))
expect(execGit).toHaveBeenCalledWith(
['fetch', '--no-tags', 'upstream', '+refs/heads/main:refs/remotes/upstream/main'],
expect.any(Object)
)
expect(execGit).not.toHaveBeenCalledWith(expect.arrayContaining(['rebase']), expect.anything())
expect(execGit).toHaveBeenCalledWith(
['merge-base', 'upstream/main', 'HEAD'],
expect.any(Object)
)
})
it('does not run rebase before collecting PR context', async () => {
const execGit = vi.fn<GitExec>(async (args) => {
if (args[0] === 'fetch') {
return { stdout: '', stderr: '' }
}
if (args[0] === 'remote') {
return { stdout: 'origin\n', stderr: '' }
}
if (args[0] === 'for-each-ref') {
return { stdout: 'origin/main\n', stderr: '' }
}
if (args[0] === 'branch') {
return { stdout: 'feature\n', stderr: '' }
}
if (args[0] === 'rebase') {
throw new Error('Generate must not rebase the live worktree')
}
if (args[0] === 'merge-base') {
return { stdout: 'abc123\n', stderr: '' }
}
if (args[0] === 'log') {
return { stdout: '- feat: change\n', stderr: '' }
}
if (args[0] === 'diff') {
return { stdout: 'M\tREADME.md\n', stderr: '' }
}
throw new Error(`Unexpected git args: ${args.join(' ')}`)
})
await expect(getPullRequestDraftContext(execGit, createContextInput())).resolves.toMatchObject({
branch: 'feature'
})
expect(execGit).not.toHaveBeenCalledWith(expect.arrayContaining(['rebase']), expect.anything())
})
it('stops generation when the relevant base fetch fails', async () => {
const execGit = vi.fn<GitExec>(async (args) => {
if (args[0] === 'remote') {
return { stdout: 'origin\nstale-fork\n', stderr: '' }
}
if (args[0] === 'for-each-ref') {
return { stdout: 'origin/main\nstale-fork/main\n', stderr: '' }
}
if (args[0] === 'fetch') {
if (args[2] !== 'origin') {
throw new Error(`Fetched unrelated remote: ${args.join(' ')}`)
}
throw new Error(
'Command failed: git fetch --no-tags origin +refs/heads/main:refs/remotes/origin/main\nfatal: unable to access origin'
)
}
throw new Error(`Unexpected git args: ${args.join(' ')}`)
})
await expect(getPullRequestDraftContext(execGit, createContextInput())).rejects.toThrow(
'Fetch before generating PR details failed: fatal: unable to access origin'
)
})
it('handles newline-heavy remote state and fetch errors without line-array splitting', async () => {
const splitSpy = vi.spyOn(String.prototype, 'split')
const execGit = vi.fn<GitExec>(async (args) => {
if (args[0] === 'remote') {
return { stdout: `${'\r\n'.repeat(10_000)}origin\r\n`, stderr: '' }
}
if (args[0] === 'for-each-ref') {
return { stdout: `${'\r\n'.repeat(10_000)}origin/main\r\n`, stderr: '' }
}
if (args[0] === 'fetch') {
throw new Error(
`Command failed: git fetch\r\n${'remote: progress\r\n'.repeat(10_000)}fatal: unable to access origin\r\n`
)
}
throw new Error(`Unexpected git args: ${args.join(' ')}`)
})
await expect(getPullRequestDraftContext(execGit, createContextInput())).rejects.toThrow(
'Fetch before generating PR details failed: fatal: unable to access origin'
)
const usedLineSplit = splitSpy.mock.calls.some(
([separator]) =>
(typeof separator === 'string' && separator === '\n') ||
(separator instanceof RegExp && separator.source === '\\r?\\n')
)
expect(usedLineSplit).toBe(false)
})
it('returns null without running git when the base is invalid', async () => {
const execGit = vi.fn<GitExec>()
await expect(getPullRequestDraftContext(execGit, createContextInput('--main'))).resolves.toBe(
null
)
expect(execGit).not.toHaveBeenCalled()
})
})