From 42d577e65a854baa0b3bf11dbb21b0e54a6d8cbf Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Sun, 29 Jun 2025 00:00:43 +1000
Subject: [PATCH] tests(ui): check for error instance instead of message

---
 .../frontend/web/src/services/api/run-graph.test.ts    | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/invokeai/frontend/web/src/services/api/run-graph.test.ts b/invokeai/frontend/web/src/services/api/run-graph.test.ts
index 3d18d9a214..b610a587e0 100644
--- a/invokeai/frontend/web/src/services/api/run-graph.test.ts
+++ b/invokeai/frontend/web/src/services/api/run-graph.test.ts
@@ -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);
     });
   });