github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-14 14:25:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

tests(ui): check for error instance instead of message

Browse Source
This commit is contained in:
psychedelicious
2025-06-29 00:00:43 +10:00
parent 38d80c9ce5
commit 42d577e65a
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
invokeai/frontend/web/src/services/api/run-graph.test.ts
View File

@@ -6,7 +6,9 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
import {
IterateNodeFoundError,
NodeNotFoundError,
OutputNodeNotFoundError,
ResultNotFoundError,
runGraph,
SessionAbortedError,
SessionCancelationError,
@@ -576,7 +578,7 @@ describe('runGraph', () => {
});
describe('output extraction', () => {
it('should throw NodeNotFoundError if output node not in session mapping', async () => {
it('should reject with NodeNotFoundError if output node not in session mapping', async () => {
const mockQueueItem = createMockQueueItem('completed', {
session: {
source_prepared_mapping: {
@@ -607,10 +609,10 @@ describe('runGraph', () => {
origin: TEST_ORIGIN,
} as S['QueueItemStatusChangedEvent']);
await expect(promise).rejects.toThrow("Node 'output-node' not found in session");
await expect(promise).rejects.toThrow(NodeNotFoundError);
});
it('should throw ResultNotFoundError if result not found for prepared node', async () => {
it('should reject with ResultNotFoundError if result not found for prepared node', async () => {
const mockQueueItem = createMockQueueItem('completed', {
session: {
source_prepared_mapping: {
@@ -641,7 +643,7 @@ describe('runGraph', () => {
origin: TEST_ORIGIN,
} as S['QueueItemStatusChangedEvent']);
await expect(promise).rejects.toThrow("Result for node 'output-node' not found in session");
await expect(promise).rejects.toThrow(ResultNotFoundError);
});
});