From 548d38fe4b2dbb2521e8a900dc4d90591033c444 Mon Sep 17 00:00:00 2001
From: bymyself <cbyrne@comfy.org>
Date: Fri, 4 Jul 2025 16:22:11 -0700
Subject: [PATCH] Fix newUserService test after console.log removal

Update test to verify new user status without relying on console.log
---
 tests-ui/tests/services/newUserService.test.ts | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/tests-ui/tests/services/newUserService.test.ts b/tests-ui/tests/services/newUserService.test.ts
index 1e78efce5..e940a785b 100644
--- a/tests-ui/tests/services/newUserService.test.ts
+++ b/tests-ui/tests/services/newUserService.test.ts
@@ -323,9 +323,7 @@ describe('newUserService', () => {
       expect(mockSettingStore.set).toHaveBeenCalledTimes(1)
     })
 
-    it('should log new user status', async () => {
-      const consoleSpy = vi.spyOn(console, 'log').mockImplementation(() => {})
-
+    it('should correctly determine new user status', async () => {
       mockSettingStore.settingValues = {}
       mockSettingStore.get.mockImplementation((key: string) => {
         if (key === 'Comfy.TutorialCompleted') return undefined
@@ -333,12 +331,19 @@ describe('newUserService', () => {
       })
       mockLocalStorage.getItem.mockReturnValue(null)
 
+      // Before initialization, isNewUser should return null
+      expect(service.isNewUser()).toBeNull()
+
       await service.initializeIfNewUser(mockSettingStore)
 
-      expect(consoleSpy).toHaveBeenCalledWith(
-        'New user status determined: true'
+      // After initialization, isNewUser should return true for a new user
+      expect(service.isNewUser()).toBe(true)
+
+      // Should set the installed version for new users
+      expect(mockSettingStore.set).toHaveBeenCalledWith(
+        'Comfy.InstalledVersion',
+        expect.any(String)
       )
-      consoleSpy.mockRestore()
     })
   })