fix(manager-react-components): useNotifications prevent early clears

Signed-off-by: Florian Renaut <[email protected]>
Signed-off-by: Tristan WAGNER <[email protected]>

packages/manager-react-components/src/components/notifications/notifications.spec.tsx

@@ -1,6 +1,6 @@
-import { vitest } from 'vitest';
+import { vi, vitest } from 'vitest';
 import React, { useEffect } from 'react';
-import { render } from '@testing-library/react';
+import { render, waitFor } from '@testing-library/react';
 import { useNotifications, NotificationType } from './useNotifications';
 import { Notifications } from './notifications.component';
 
@@ -35,11 +35,29 @@ describe('notifications component', () => {
     container = await render(<Notifications />).container;
     expect(container.children.length).toBe(2);
   });
-  it('should clear notifications', async () => {
+  it('should not clear notifications created within the last 1000ms', async () => {
     let { container } = render(<Notifications />);
-    expect(container.children.length).not.toBe(0);
+    expect(container.children.length).toBe(2);
+
     render(<ClearNotifications />);
     container = render(<Notifications />).container;
+
+    expect(container.children.length).toBe(2);
+  });
+
+  it('should clear notifications older than 1000ms', async () => {
+    vi.useFakeTimers();
+
+    let { container } = render(<Notifications />);
+    expect(container.children.length).toBe(2);
+
+    vi.advanceTimersByTime(2000);
+
+    render(<ClearNotifications />);
+    container = render(<Notifications />).container;
+
     expect(container.children.length).toBe(0);
+
+    vi.useRealTimers();
   });
 });

packages/manager-react-components/src/components/notifications/useNotifications.ts

@@ -14,6 +14,7 @@ export interface Notification {
   content: ReactNode;
   type: NotificationType;
   dismissable?: boolean;
+  creationTimestamp?: number;
 }
 
 export interface NotificationState {
@@ -44,7 +45,13 @@ export const useNotifications = create<NotificationState>((set, get) => ({
       uid: state.uid + 1,
       notifications: [
         ...state.notifications,
-        { uid: state.uid, content, type, dismissable },
+        {
+          uid: state.uid,
+          content,
+          type,
+          dismissable,
+          creationTimestamp: Date.now(),
+        },
       ],
     })),
   addSuccess: (content: ReactNode, dismissable = false) =>
@@ -61,7 +68,12 @@ export const useNotifications = create<NotificationState>((set, get) => ({
         ({ uid }) => uid !== toRemoveUid,
       ),
     })),
-  clearNotifications: () => set(() => ({ notifications: [] })),
+  clearNotifications: () =>
+    set((state) => ({
+      notifications: state.notifications.filter(
+        (notification) => Date.now() - notification.creationTimestamp < 1000,
+      ),
+    })),
 }));
 
 export default useNotifications;