github-mirror/n8n
1
0
Fork 0
mirror of https://github.com/n8n-io/n8n.git synced 2026-10-17 13:39:44 +02:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity Graph

test(core): Cover Agent foreground cancellation routes (no-changelog)

Browse Source
This commit is contained in:
Eugene Molodkin 2026-10-16 18:13:42 +02:00
parent f474fcee42
commit f1deb15d2b
No known key found for this signature in database
1 changed files with 55 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
packages/cli/src/modules/agents/__tests__/agent-chat.controller.test.ts
View File

@ -825,30 +825,64 @@ describe('AgentChatController budget notice', () => {
});
});
describe('AgentChatController execution cancellation', () => {
it.each([undefined, 'all', 'foreground'] as const)(
'forwards scope %s for the current preview execution',
async (scope) => {
const { controller, chatExecutionService } = makeController();
chatExecutionService.requestCancel.mockResolvedValue(true);
await expect(
controller.cancelChatExecution(
{ params: { projectId: 'project-1' }, user: { id: 'user-1' } } as never,
{} as never,
'agent-1',
'thread-1',
'execution-1',
scope ? { scope } : undefined,
),
).resolves.toEqual({ cancelRequested: true });
expect(chatExecutionService.requestCancel).toHaveBeenCalledWith({
projectId: 'project-1',
agentId: 'agent-1',
threadId: 'thread-1',
executionId: 'execution-1',
userId: 'user-1',
surface: 'preview',
...(scope ? { scope } : {}),
});
},
);
});
describe('AgentChatController HITL cancellation', () => {
it('cancels a suspended run for the current preview user', async () => {
const { controller, agentExecutionOrchestratorService, agentsService } = makeController();
agentsService.findById.mockResolvedValue({ id: 'agent-1' } as never);
agentExecutionOrchestratorService.cancelChatRun.mockResolvedValue(true);
it.each([undefined, 'all', 'foreground'] as const)(
'cancels a suspended run for the current preview user with scope %s',
async (scope) => {
const { controller, agentExecutionOrchestratorService, agentsService } = makeController();
agentsService.findById.mockResolvedValue({ id: 'agent-1' } as never);
agentExecutionOrchestratorService.cancelChatRun.mockResolvedValue(true);
await expect(
controller.cancelChatRun(
{
params: { projectId: 'project-1' },
user: { id: 'user-1' },
} as never,
{} as never,
'agent-1',
'run-1',
),
).resolves.toEqual({ cancelled: true });
await expect(
controller.cancelChatRun(
{
params: { projectId: 'project-1' },
user: { id: 'user-1' },
} as never,
{} as never,
'agent-1',
'run-1',
scope ? { scope } : undefined,
),
).resolves.toEqual({ cancelled: true });
expect(agentExecutionOrchestratorService.cancelChatRun).toHaveBeenCalledWith({
agentId: 'agent-1',
runId: 'run-1',
resourceId: 'draft-chat:user-1',
});
});
expect(agentExecutionOrchestratorService.cancelChatRun).toHaveBeenCalledWith({
agentId: 'agent-1',
runId: 'run-1',
resourceId: 'draft-chat:user-1',
...(scope === 'foreground' ? { cancelBackgroundJobs: false } : {}),
});
},
);
});
describe('AgentChatController attachment cleanup on failed turns', () => {