mirror of
https://github.com/stablyai/orca.git
synced 2026-10-27 08:02:51 +00:00
502 lines
16 KiB
TypeScript
502 lines
16 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
|
|
|
type RateLimitGuardResult =
|
|
| { blocked: false }
|
|
| { blocked: true; remaining: number; limit: number; resetAt: number }
|
|
|
|
const {
|
|
ghExecFileAsyncMock,
|
|
getOwnerRepoMock,
|
|
getIssueOwnerRepoMock,
|
|
getWorkItemMock,
|
|
getPRChecksMock,
|
|
getPRCommentsMock,
|
|
rateLimitGuardMock,
|
|
noteRateLimitSpendMock,
|
|
ghRepoExecOptionsMock,
|
|
githubRepoContextMock,
|
|
acquireMock,
|
|
releaseMock
|
|
} = vi.hoisted(() => ({
|
|
ghExecFileAsyncMock: vi.fn(),
|
|
getOwnerRepoMock: vi.fn(),
|
|
getIssueOwnerRepoMock: vi.fn(),
|
|
getWorkItemMock: vi.fn(),
|
|
getPRChecksMock: vi.fn(),
|
|
getPRCommentsMock: vi.fn(),
|
|
rateLimitGuardMock: vi.fn<() => RateLimitGuardResult>(() => ({ blocked: false })),
|
|
noteRateLimitSpendMock: vi.fn(),
|
|
ghRepoExecOptionsMock: vi.fn((context) =>
|
|
context.connectionId
|
|
? {}
|
|
: { cwd: context.repoPath, ...(context.wslDistro ? { wslDistro: context.wslDistro } : {}) }
|
|
),
|
|
githubRepoContextMock: vi.fn((repoPath, connectionId, localGitOptions) => ({
|
|
repoPath,
|
|
connectionId: connectionId ?? null,
|
|
...localGitOptions
|
|
})),
|
|
acquireMock: vi.fn(),
|
|
releaseMock: vi.fn()
|
|
}))
|
|
|
|
vi.mock('./gh-utils', () => ({
|
|
ghExecFileAsync: ghExecFileAsyncMock,
|
|
getOwnerRepo: getOwnerRepoMock,
|
|
getIssueOwnerRepo: getIssueOwnerRepoMock,
|
|
ghRepoExecOptions: ghRepoExecOptionsMock,
|
|
githubRepoContext: githubRepoContextMock,
|
|
acquire: acquireMock,
|
|
release: releaseMock
|
|
}))
|
|
|
|
vi.mock('./client', () => ({
|
|
getWorkItem: getWorkItemMock,
|
|
getPRChecks: getPRChecksMock,
|
|
getPRComments: getPRCommentsMock
|
|
}))
|
|
|
|
vi.mock('./rate-limit', () => ({
|
|
rateLimitGuard: rateLimitGuardMock,
|
|
noteRateLimitSpend: noteRateLimitSpendMock
|
|
}))
|
|
|
|
import { getWorkItemDetails } from './work-item-details'
|
|
|
|
describe('getWorkItemDetails', () => {
|
|
beforeEach(() => {
|
|
ghExecFileAsyncMock.mockReset()
|
|
getOwnerRepoMock.mockReset()
|
|
getIssueOwnerRepoMock.mockReset()
|
|
getWorkItemMock.mockReset()
|
|
getPRChecksMock.mockReset()
|
|
getPRCommentsMock.mockReset()
|
|
rateLimitGuardMock.mockReset()
|
|
rateLimitGuardMock.mockReturnValue({ blocked: false })
|
|
noteRateLimitSpendMock.mockReset()
|
|
ghRepoExecOptionsMock.mockClear()
|
|
githubRepoContextMock.mockClear()
|
|
acquireMock.mockReset()
|
|
releaseMock.mockReset()
|
|
acquireMock.mockResolvedValue(undefined)
|
|
})
|
|
|
|
it('uses the collapsed GraphQL issue query with timeline activity enrichment', async () => {
|
|
getWorkItemMock.mockResolvedValueOnce({
|
|
id: 'issue:923',
|
|
type: 'issue',
|
|
number: 923,
|
|
title: 'Use upstream issues',
|
|
state: 'open',
|
|
url: 'https://github.com/acme/widgets/issues/923',
|
|
labels: [],
|
|
updatedAt: '2026-04-01T00:00:00Z',
|
|
author: 'issue-author'
|
|
})
|
|
getIssueOwnerRepoMock.mockResolvedValue({ owner: 'acme', repo: 'widgets' })
|
|
const timelineEvents = [
|
|
{
|
|
id: 101,
|
|
event: 'assigned',
|
|
actor: { login: 'timeline-actor', avatar_url: 'https://x/timeline-actor' },
|
|
assignee: { login: 'assigned-user' },
|
|
created_at: '2026-04-01T01:00:00Z'
|
|
},
|
|
{
|
|
id: 102,
|
|
event: 'cross-referenced',
|
|
actor: { login: 'timeline-actor', avatar_url: 'https://x/timeline-actor' },
|
|
created_at: '2026-04-01T02:00:00Z',
|
|
source: {
|
|
issue: {
|
|
number: 6180,
|
|
title: 'Synthetic reference PR',
|
|
html_url: 'https://github.com/acme/widgets/pull/6180',
|
|
repository: { owner: { login: 'acme' }, name: 'widgets' },
|
|
pull_request: {}
|
|
}
|
|
}
|
|
},
|
|
{
|
|
id: 103,
|
|
event: 'moved_columns_in_project',
|
|
actor: { login: 'github-project-automation', avatar_url: 'https://x/bot' },
|
|
created_at: '2026-04-01T03:00:00Z',
|
|
previous_column_name: 'Doing',
|
|
project_column_name: 'Complete',
|
|
project: { name: 'Example Project' }
|
|
},
|
|
{
|
|
id: 104,
|
|
event: 'closed',
|
|
actor: { login: 'timeline-actor', avatar_url: 'https://x/timeline-actor' },
|
|
created_at: '2026-04-01T04:00:00Z',
|
|
state_reason: 'completed',
|
|
closer: {
|
|
number: 6180,
|
|
title: 'Synthetic reference PR',
|
|
html_url: 'https://github.com/acme/widgets/pull/6180',
|
|
repository: { owner: { login: 'acme' }, name: 'widgets' },
|
|
pull_request: {}
|
|
}
|
|
}
|
|
]
|
|
ghExecFileAsyncMock
|
|
.mockResolvedValueOnce({
|
|
stdout: JSON.stringify({
|
|
data: {
|
|
repository: {
|
|
issue: {
|
|
body: 'Issue body',
|
|
assignees: { nodes: [{ login: 'assigned-user' }] },
|
|
participants: {
|
|
nodes: [{ login: 'issue-author', avatarUrl: 'https://x/y', name: 'Issue Author' }]
|
|
},
|
|
comments: {
|
|
nodes: [
|
|
{
|
|
databaseId: 7,
|
|
body: 'first',
|
|
createdAt: '2026-04-01T00:00:00Z',
|
|
url: 'https://github.com/acme/widgets/issues/923#issuecomment-7',
|
|
author: { login: 'issue-author', avatarUrl: 'https://x/y' }
|
|
}
|
|
]
|
|
}
|
|
}
|
|
}
|
|
}
|
|
})
|
|
})
|
|
.mockResolvedValueOnce({
|
|
stdout: timelineEvents.map((event) => JSON.stringify(event)).join('\n')
|
|
})
|
|
|
|
const details = await getWorkItemDetails('/repo-root', 923, 'issue')
|
|
|
|
expect(getWorkItemMock).toHaveBeenCalledWith('/repo-root', 923, 'issue', undefined)
|
|
expect(ghExecFileAsyncMock).toHaveBeenCalledTimes(2)
|
|
expect(ghExecFileAsyncMock.mock.calls[0][0][0]).toBe('api')
|
|
expect(ghExecFileAsyncMock.mock.calls[0][0][1]).toBe('graphql')
|
|
expect(ghExecFileAsyncMock.mock.calls[1][0]).toEqual([
|
|
'api',
|
|
'--cache',
|
|
'60s',
|
|
'repos/acme/widgets/issues/923/timeline?per_page=100&page=1',
|
|
'--jq',
|
|
'.[] | @json'
|
|
])
|
|
expect(details?.body).toBe('Issue body')
|
|
expect(details?.assignees).toEqual(['assigned-user'])
|
|
expect(details?.comments).toHaveLength(1)
|
|
expect(details?.comments[0].id).toBe(7)
|
|
expect(details?.timelineItems).toMatchObject([
|
|
{ event: 'assigned', actor: 'timeline-actor', assignee: 'assigned-user' },
|
|
{
|
|
event: 'cross-referenced',
|
|
source: {
|
|
type: 'pr',
|
|
number: 6180,
|
|
repository: 'acme/widgets'
|
|
}
|
|
},
|
|
{
|
|
event: 'moved_columns_in_project',
|
|
actor: 'github-project-automation',
|
|
previousColumnName: 'Doing',
|
|
columnName: 'Complete',
|
|
projectName: 'Example Project'
|
|
},
|
|
{
|
|
event: 'closed',
|
|
stateReason: 'completed',
|
|
closer: { type: 'pr', number: 6180 }
|
|
}
|
|
])
|
|
expect(details?.participants?.[0]?.login).toBe('issue-author')
|
|
})
|
|
|
|
it('caps issue timeline pagination by supported activity items', async () => {
|
|
getWorkItemMock.mockResolvedValueOnce({
|
|
id: 'issue:923',
|
|
type: 'issue',
|
|
number: 923,
|
|
title: 'Use upstream issues',
|
|
state: 'open',
|
|
url: 'https://github.com/acme/widgets/issues/923',
|
|
labels: [],
|
|
updatedAt: '2026-04-01T00:00:00Z',
|
|
author: 'issue-author'
|
|
})
|
|
getIssueOwnerRepoMock.mockResolvedValue({ owner: 'acme', repo: 'widgets' })
|
|
const makeTimelineEvent = (page: number, index: number, event: string): string =>
|
|
JSON.stringify({
|
|
id: `${page}:${index}`,
|
|
event,
|
|
actor: { login: 'issue-author', avatar_url: 'https://x/y' },
|
|
assignee: { login: `assignee-${page}-${index}` },
|
|
created_at: '2026-04-01T00:00:00Z'
|
|
})
|
|
const makeTimelinePage = (page: number, supportedCount: number): string =>
|
|
Array.from({ length: 100 }, (_, index) =>
|
|
makeTimelineEvent(page, index, index < supportedCount ? 'assigned' : 'subscribed')
|
|
).join('\n')
|
|
ghExecFileAsyncMock
|
|
.mockResolvedValueOnce({
|
|
stdout: JSON.stringify({
|
|
data: {
|
|
repository: {
|
|
issue: {
|
|
body: 'Issue body',
|
|
assignees: { nodes: [] },
|
|
participants: { nodes: [] },
|
|
comments: { nodes: [] }
|
|
}
|
|
}
|
|
}
|
|
})
|
|
})
|
|
.mockResolvedValueOnce({ stdout: makeTimelinePage(1, 0) })
|
|
.mockResolvedValueOnce({ stdout: makeTimelinePage(2, 0) })
|
|
.mockResolvedValueOnce({ stdout: makeTimelinePage(3, 10) })
|
|
.mockResolvedValueOnce({ stdout: makeTimelinePage(4, 100) })
|
|
.mockResolvedValueOnce({ stdout: makeTimelinePage(5, 100) })
|
|
.mockResolvedValueOnce({ stdout: makeTimelinePage(6, 100) })
|
|
|
|
const details = await getWorkItemDetails('/repo-root', 923, 'issue')
|
|
|
|
expect(ghExecFileAsyncMock).toHaveBeenCalledTimes(7)
|
|
expect(ghExecFileAsyncMock.mock.calls[1][0]).toContain(
|
|
'repos/acme/widgets/issues/923/timeline?per_page=100&page=1'
|
|
)
|
|
expect(ghExecFileAsyncMock.mock.calls[2][0]).toContain(
|
|
'repos/acme/widgets/issues/923/timeline?per_page=100&page=2'
|
|
)
|
|
expect(ghExecFileAsyncMock.mock.calls[3][0]).toContain(
|
|
'repos/acme/widgets/issues/923/timeline?per_page=100&page=3'
|
|
)
|
|
expect(ghExecFileAsyncMock.mock.calls[6][0]).toContain(
|
|
'repos/acme/widgets/issues/923/timeline?per_page=100&page=6'
|
|
)
|
|
expect(
|
|
ghExecFileAsyncMock.mock.calls.some((call) =>
|
|
call[0].includes('repos/acme/widgets/issues/923/timeline?per_page=100&page=7')
|
|
)
|
|
).toBe(false)
|
|
const timelineItems = details?.timelineItems
|
|
if (!timelineItems) {
|
|
throw new Error('Expected timeline items to be present')
|
|
}
|
|
expect(timelineItems).toHaveLength(300)
|
|
expect(timelineItems.at(0)).toMatchObject({ assignee: 'assignee-3-0' })
|
|
expect(timelineItems.at(-1)).toMatchObject({ assignee: 'assignee-6-89' })
|
|
})
|
|
|
|
it('falls back to REST + GraphQL when the collapsed issue query fails', async () => {
|
|
getWorkItemMock.mockResolvedValueOnce({
|
|
id: 'issue:923',
|
|
type: 'issue',
|
|
number: 923,
|
|
title: 'Use upstream issues',
|
|
state: 'open',
|
|
url: 'https://github.com/acme/widgets/issues/923',
|
|
labels: [],
|
|
updatedAt: '2026-04-01T00:00:00Z',
|
|
author: 'issue-author'
|
|
})
|
|
getIssueOwnerRepoMock.mockResolvedValue({ owner: 'acme', repo: 'widgets' })
|
|
// Collapsed GraphQL throws → fallback path picks up.
|
|
ghExecFileAsyncMock
|
|
.mockRejectedValueOnce(new Error('GraphQL error'))
|
|
.mockResolvedValueOnce({ stdout: JSON.stringify({ body: 'Issue body' }) })
|
|
.mockResolvedValueOnce({ stdout: '[]' })
|
|
.mockResolvedValueOnce({ stdout: '[]' })
|
|
.mockResolvedValueOnce({
|
|
stdout: JSON.stringify({
|
|
data: { repository: { issue: { participants: { nodes: [] } } } }
|
|
})
|
|
})
|
|
.mockResolvedValueOnce({
|
|
stdout: JSON.stringify({ data: {} })
|
|
})
|
|
|
|
const details = await getWorkItemDetails('/repo-root', 923, 'issue')
|
|
|
|
expect(ghExecFileAsyncMock).toHaveBeenNthCalledWith(
|
|
2,
|
|
['api', '--cache', '60s', 'repos/acme/widgets/issues/923'],
|
|
{ cwd: '/repo-root' }
|
|
)
|
|
expect(ghExecFileAsyncMock).toHaveBeenNthCalledWith(
|
|
3,
|
|
['api', '--cache', '60s', 'repos/acme/widgets/issues/923/comments?per_page=100'],
|
|
{ cwd: '/repo-root' }
|
|
)
|
|
expect(ghExecFileAsyncMock).toHaveBeenNthCalledWith(
|
|
4,
|
|
[
|
|
'api',
|
|
'--cache',
|
|
'60s',
|
|
'repos/acme/widgets/issues/923/timeline?per_page=100&page=1',
|
|
'--jq',
|
|
'.[] | @json'
|
|
],
|
|
{ cwd: '/repo-root' }
|
|
)
|
|
expect(details?.body).toBe('Issue body')
|
|
})
|
|
|
|
it('skips optional GraphQL issue detail calls when the cached GraphQL budget is low', async () => {
|
|
rateLimitGuardMock.mockReturnValue({
|
|
blocked: true,
|
|
remaining: 3,
|
|
limit: 5000,
|
|
resetAt: 1_800_000_000
|
|
})
|
|
getWorkItemMock.mockResolvedValueOnce({
|
|
id: 'issue:923',
|
|
type: 'issue',
|
|
number: 923,
|
|
title: 'Use upstream issues',
|
|
state: 'open',
|
|
url: 'https://github.com/acme/widgets/issues/923',
|
|
labels: [],
|
|
updatedAt: '2026-04-01T00:00:00Z',
|
|
author: 'issue-author'
|
|
})
|
|
getIssueOwnerRepoMock.mockResolvedValue({ owner: 'acme', repo: 'widgets' })
|
|
ghExecFileAsyncMock
|
|
.mockResolvedValueOnce({ stdout: JSON.stringify({ body: 'Issue body', assignees: [] }) })
|
|
.mockResolvedValueOnce({ stdout: '[]' })
|
|
.mockResolvedValueOnce({ stdout: '[]' })
|
|
|
|
const details = await getWorkItemDetails('/repo-root', 923, 'issue')
|
|
|
|
expect(ghExecFileAsyncMock).toHaveBeenCalledTimes(3)
|
|
expect(ghExecFileAsyncMock.mock.calls.some((call) => call[0][1] === 'graphql')).toBe(false)
|
|
expect(noteRateLimitSpendMock).not.toHaveBeenCalled()
|
|
expect(details?.body).toBe('Issue body')
|
|
expect(details?.participants).toEqual([])
|
|
})
|
|
|
|
it('uses SSH connection context for issue details without local cwd', async () => {
|
|
getWorkItemMock.mockResolvedValueOnce({
|
|
id: 'issue:923',
|
|
type: 'issue',
|
|
number: 923,
|
|
title: 'Use upstream issues',
|
|
state: 'open',
|
|
url: 'https://github.com/acme/widgets/issues/923',
|
|
labels: [],
|
|
updatedAt: '2026-04-01T00:00:00Z',
|
|
author: 'issue-author'
|
|
})
|
|
getIssueOwnerRepoMock.mockResolvedValue({ owner: 'acme', repo: 'widgets' })
|
|
ghExecFileAsyncMock
|
|
.mockResolvedValueOnce({
|
|
stdout: JSON.stringify({
|
|
data: {
|
|
repository: {
|
|
issue: {
|
|
body: 'Remote issue body',
|
|
assignees: { nodes: [] },
|
|
participants: { nodes: [] },
|
|
comments: { nodes: [] }
|
|
}
|
|
}
|
|
}
|
|
})
|
|
})
|
|
.mockResolvedValueOnce({ stdout: '[]' })
|
|
|
|
const details = await getWorkItemDetails('/home/tester/widgets', 923, 'issue', 'ssh-test-1')
|
|
|
|
expect(getWorkItemMock).toHaveBeenCalledWith('/home/tester/widgets', 923, 'issue', 'ssh-test-1')
|
|
expect(getIssueOwnerRepoMock).toHaveBeenCalledWith('/home/tester/widgets', 'ssh-test-1')
|
|
expect(ghExecFileAsyncMock.mock.calls[0][1]).toEqual({})
|
|
expect(details?.body).toBe('Remote issue body')
|
|
})
|
|
|
|
it('routes local WSL PR detail fan-out through the selected distro', async () => {
|
|
const localGitOptions = { wslDistro: 'Ubuntu' }
|
|
getWorkItemMock.mockResolvedValueOnce({
|
|
id: 'pr:42',
|
|
type: 'pr',
|
|
number: 42,
|
|
title: 'Review drawer WSL',
|
|
state: 'open',
|
|
url: 'https://github.com/acme/widgets/pull/42',
|
|
labels: [],
|
|
updatedAt: '2026-04-01T00:00:00Z',
|
|
author: 'pr-author'
|
|
})
|
|
getOwnerRepoMock.mockResolvedValue({ owner: 'acme', repo: 'widgets' })
|
|
getPRCommentsMock.mockResolvedValue([])
|
|
getPRChecksMock.mockResolvedValue([])
|
|
ghExecFileAsyncMock.mockImplementation(async (args: string[]) => {
|
|
const target = args.at(-1)
|
|
if (target === 'repos/acme/widgets/pulls/42') {
|
|
return {
|
|
stdout: JSON.stringify({
|
|
body: 'PR body',
|
|
head: { sha: 'head-sha' },
|
|
base: { sha: 'base-sha' }
|
|
})
|
|
}
|
|
}
|
|
if (target === 'repos/acme/widgets/pulls/42/files?per_page=100') {
|
|
return { stdout: '[]' }
|
|
}
|
|
const query = args.find((arg) => arg.startsWith('query=')) ?? ''
|
|
if (query.includes('viewerViewedState')) {
|
|
return {
|
|
stdout: JSON.stringify({
|
|
data: {
|
|
repository: {
|
|
pullRequest: {
|
|
id: 'PR_kwDO123',
|
|
files: { pageInfo: { hasNextPage: false }, nodes: [] }
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
if (query.includes('participants(first: 100)')) {
|
|
return {
|
|
stdout: JSON.stringify({
|
|
data: { repository: { pullRequest: { participants: { nodes: [] } } } }
|
|
})
|
|
}
|
|
}
|
|
return { stdout: JSON.stringify({ data: {} }) }
|
|
})
|
|
|
|
const details = await getWorkItemDetails('/repo-root', 42, 'pr', null, localGitOptions)
|
|
|
|
expect(details?.body).toBe('PR body')
|
|
expect(getWorkItemMock).toHaveBeenCalledWith('/repo-root', 42, 'pr', null, localGitOptions)
|
|
expect(getOwnerRepoMock).toHaveBeenCalledWith('/repo-root', null, localGitOptions)
|
|
expect(getPRCommentsMock).toHaveBeenCalledWith(
|
|
'/repo-root',
|
|
42,
|
|
undefined,
|
|
null,
|
|
localGitOptions
|
|
)
|
|
expect(getPRChecksMock).toHaveBeenCalledWith(
|
|
'/repo-root',
|
|
42,
|
|
'head-sha',
|
|
null,
|
|
undefined,
|
|
null,
|
|
localGitOptions
|
|
)
|
|
expect(ghExecFileAsyncMock.mock.calls.every((call) => call[1]?.wslDistro === 'Ubuntu')).toBe(
|
|
true
|
|
)
|
|
})
|
|
})
|