From f44c06f16ba6f1a303220b353957738d8559119b Mon Sep 17 00:00:00 2001
From: bymyself <cbyrne@comfy.org>
Date: Tue, 20 Jan 2026 14:03:57 -0800
Subject: [PATCH] fix: restore mock cleanup in tests

mockRestore() calls are needed for these specific mocks
---
 .../services/workflowSessionStorageService.test.ts        | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/platform/workflow/persistence/services/workflowSessionStorageService.test.ts b/src/platform/workflow/persistence/services/workflowSessionStorageService.test.ts
index da2317d88..d6950560c 100644
--- a/src/platform/workflow/persistence/services/workflowSessionStorageService.test.ts
+++ b/src/platform/workflow/persistence/services/workflowSessionStorageService.test.ts
@@ -59,21 +59,25 @@ describe('workflowSessionStorageService', () => {
     })
 
     it('returns false after max eviction attempts', () => {
-      vi.spyOn(sessionStorage, 'setItem').mockImplementation(() => {
+      const spy = vi.spyOn(sessionStorage, 'setItem').mockImplementation(() => {
         throw new DOMException('Quota exceeded', 'QuotaExceededError')
       })
 
       const result = setWithEviction('key', { test: 'data' })
       expect(result).toBe(false)
+
+      spy.mockRestore()
     })
 
     it('returns false on unexpected errors', () => {
-      vi.spyOn(sessionStorage, 'setItem').mockImplementation(() => {
+      const spy = vi.spyOn(sessionStorage, 'setItem').mockImplementation(() => {
         throw new Error('Unexpected error')
       })
 
       const result = setWithEviction('key', { test: 'data' })
       expect(result).toBe(false)
+
+      spy.mockRestore()
     })
   })