feat: Add auto update functionality (#4686)

1 files changed, 9 insertions, 5 deletions

diff --git a/packages/cli/src/ui/utils/updateCheck.test.ts b/packages/cli/src/ui/utils/updateCheck.test.ts
index 975c320d..4985afe8 100644
--- a/packages/cli/src/ui/utils/updateCheck.test.ts
+++ b/packages/cli/src/ui/utils/updateCheck.test.ts
@@ -50,7 +50,9 @@ describe('checkForUpdates', () => {
       name: 'test-package',
       version: '1.0.0',
     });
-    updateNotifier.mockReturnValue({ update: null });
+    updateNotifier.mockReturnValue({
+      fetchInfo: vi.fn(async () => null),
+    });
     const result = await checkForUpdates();
     expect(result).toBeNull();
   });
@@ -61,10 +63,12 @@ describe('checkForUpdates', () => {
       version: '1.0.0',
     });
     updateNotifier.mockReturnValue({
-      update: { current: '1.0.0', latest: '1.1.0' },
+      fetchInfo: vi.fn(async () => ({ current: '1.0.0', latest: '1.1.0' })),
     });
+
     const result = await checkForUpdates();
-    expect(result).toContain('1.0.0 → 1.1.0');
+    expect(result?.message).toContain('1.0.0 → 1.1.0');
+    expect(result?.update).toEqual({ current: '1.0.0', latest: '1.1.0' });
   });
 
   it('should return null if the latest version is the same as the current version', async () => {
@@ -73,7 +77,7 @@ describe('checkForUpdates', () => {
       version: '1.0.0',
     });
     updateNotifier.mockReturnValue({
-      update: { current: '1.0.0', latest: '1.0.0' },
+      fetchInfo: vi.fn(async () => ({ current: '1.0.0', latest: '1.0.0' })),
     });
     const result = await checkForUpdates();
     expect(result).toBeNull();
@@ -85,7 +89,7 @@ describe('checkForUpdates', () => {
       version: '1.1.0',
     });
     updateNotifier.mockReturnValue({
-      update: { current: '1.1.0', latest: '1.0.0' },
+      fetchInfo: vi.fn(async () => ({ current: '1.0.0', latest: '0.09' })),
     });
     const result = await checkForUpdates();
     expect(result).toBeNull();