Update on tab change

__tests__/components/brain/left-sidebar/waves/BrainLeftSidebarWaveDropTime.test.tsx

@@ -1,19 +1,68 @@
-import { render } from '@testing-library/react';
-import React from 'react';
-import BrainLeftSidebarWaveDropTime from '@/components/brain/left-sidebar/waves/BrainLeftSidebarWaveDropTime';
-import { getTimeAgoShort } from '@/helpers/Helpers';
-
-jest.mock('@/helpers/Helpers');
-
-describe('BrainLeftSidebarWaveDropTime', () => {
-  it('renders time using helper and sets interval', () => {
-    (getTimeAgoShort as jest.Mock).mockReturnValue('1m');
-    const setSpy = jest.spyOn(global, 'setInterval');
-    const clearSpy = jest.spyOn(global, 'clearInterval');
-    const { unmount, getByText } = render(<BrainLeftSidebarWaveDropTime time={10} />);
-    expect(getByText('1m')).toBeInTheDocument();
+import { act, render } from "@testing-library/react";
+import React from "react";
+import BrainLeftSidebarWaveDropTime from "@/components/brain/left-sidebar/waves/BrainLeftSidebarWaveDropTime";
+import { getTimeAgoShort } from "@/helpers/Helpers";
+
+jest.mock("@/helpers/Helpers");
+
+describe("BrainLeftSidebarWaveDropTime", () => {
+  const setDocumentVisibilityState = (state: DocumentVisibilityState) => {
+    Object.defineProperty(document, "visibilityState", {
+      configurable: true,
+      value: state,
+    });
+  };
+
+  beforeEach(() => {
+    jest.clearAllMocks();
+    setDocumentVisibilityState("visible");
+  });
+
+  it("renders time using helper and sets interval", () => {
+    (getTimeAgoShort as jest.Mock).mockReturnValue("1m");
+    const setSpy = jest.spyOn(global, "setInterval");
+    const clearSpy = jest.spyOn(global, "clearInterval");
+    const { unmount, getByText } = render(
+      <BrainLeftSidebarWaveDropTime time={10} />
+    );
+    expect(getByText("1m")).toBeInTheDocument();
     expect(setSpy).toHaveBeenCalledWith(expect.any(Function), 60000);
     unmount();
     expect(clearSpy).toHaveBeenCalled();
   });
+
+  it("refreshes immediately when the tab becomes visible again or gains focus", () => {
+    jest.useFakeTimers();
+    jest.setSystemTime(new Date("2026-04-07T10:00:00.000Z"));
+    (getTimeAgoShort as jest.Mock).mockImplementation(
+      (_time: number, now?: number) => `${now}`
+    );
+
+    render(<BrainLeftSidebarWaveDropTime time={10} />);
+    expect(getTimeAgoShort).toHaveBeenLastCalledWith(
+      10,
+      new Date("2026-04-07T10:00:00.000Z").getTime()
+    );
+
+    jest.setSystemTime(new Date("2026-04-07T10:01:00.000Z"));
+    act(() => {
+      setDocumentVisibilityState("visible");
+      document.dispatchEvent(new Event("visibilitychange"));
+    });
+    expect(getTimeAgoShort).toHaveBeenLastCalledWith(
+      10,
+      new Date("2026-04-07T10:01:00.000Z").getTime()
+    );
+
+    jest.setSystemTime(new Date("2026-04-07T10:02:00.000Z"));
+    act(() => {
+      window.dispatchEvent(new Event("focus"));
+    });
+    expect(getTimeAgoShort).toHaveBeenLastCalledWith(
+      10,
+      new Date("2026-04-07T10:02:00.000Z").getTime()
+    );
+
+    jest.useRealTimers();
+  });
 });

__tests__/services/websocket/WebSocketProvider.test.tsx

@@ -680,7 +680,9 @@
       consoleSpy.mockRestore();
     });
 
-    it("closes existing connection before creating new one", () => {
+    it("closes existing connection before creating new one and ignores stale close events", () => {
+      jest.useFakeTimers();
+
       const wrapper = createWrapper({ url: "ws://test" });
       const { result } = renderHook(() => React.useContext(WebSocketContext)!, {
         wrapper,
@@ -702,11 +704,69 @@
         result.current.connect("token2");
       });
 
+      const ws2 = (global.WebSocket as jest.MockedFunction<typeof WebSocket>)
+        .mock.results[1]?.value as MockWebSocket;
+
       expect(ws1.close).toHaveBeenCalled();
       expect(global.WebSocket).toHaveBeenCalledTimes(2);
       expect(global.WebSocket).toHaveBeenLastCalledWith(
         "ws://test?token=token2"
       );
+
+      expect(result.current.status).toBe(WebSocketStatus.CONNECTING);
+
+      act(() => {
+        ws1.triggerClose(1006, "Stale socket close");
+      });
+
+      expect(result.current.status).toBe(WebSocketStatus.CONNECTING);
+
+      act(() => {
+        jest.advanceTimersByTime(5000);
+      });
+
+      expect(global.WebSocket).toHaveBeenCalledTimes(2);
+
+      act(() => {
+        ws2.triggerOpen();
+      });
+
+      expect(result.current.status).toBe(WebSocketStatus.CONNECTED);
+    });
+
+    it("ignores open events from a replaced socket", () => {
+      const wrapper = createWrapper({ url: "ws://test" });
+      const { result } = renderHook(() => React.useContext(WebSocketContext)!, {
+        wrapper,
+      });
+
+      act(() => {
+        result.current.connect("token1");
+      });
+
+      const ws1 = (global.WebSocket as jest.MockedFunction<typeof WebSocket>)
+        .mock.results[0]?.value as MockWebSocket;
+
+      act(() => {
+        result.current.connect("token2");
+      });
+
+      const ws2 = (global.WebSocket as jest.MockedFunction<typeof WebSocket>)
+        .mock.results[1]?.value as MockWebSocket;
+
+      expect(result.current.status).toBe(WebSocketStatus.CONNECTING);
+
+      act(() => {
+        ws1.triggerOpen();
+      });
+
+      expect(result.current.status).toBe(WebSocketStatus.CONNECTING);
+
+      act(() => {
+        ws2.triggerOpen();
+      });
+
+      expect(result.current.status).toBe(WebSocketStatus.CONNECTED);
     });
   });