Switch to a reducer for tracking update state fixing flicker issues due to continuous renders (#10280)

packages/cli/src/ui/commands/extensionsCommand.test.ts

@@ -26,6 +26,7 @@ import { ExtensionUpdateState } from '../state/extensions.js';
 vi.mock('../../config/extensions/update.js', () => ({
   updateExtension: vi.fn(),
   updateAllUpdatableExtensions: vi.fn(),
+  checkForAllExtensionUpdates: vi.fn(),
 }));
 
 const mockUpdateExtension = updateExtension as MockedFunction<
@@ -51,6 +52,9 @@ describe('extensionsCommand', () => {
           getWorkingDir: () => '/test/dir',
         },
       },
+      ui: {
+        dispatchExtensionStateUpdate: vi.fn(),
+      },
     });
   });
 
@@ -168,10 +172,10 @@ describe('extensionsCommand', () => {
         updatedVersion: '1.0.1',
       });
       mockGetExtensions.mockReturnValue([extension]);
-      mockContext.ui.extensionsUpdateState.set(
-        extension.name,
-        ExtensionUpdateState.UPDATE_AVAILABLE,
-      );
+      mockContext.ui.extensionsUpdateState.set(extension.name, {
+        status: ExtensionUpdateState.UPDATE_AVAILABLE,
+        processed: false,
+      });
       await updateAction(mockContext, 'ext-one');
       expect(mockUpdateExtension).toHaveBeenCalledWith(
         extension,