fix: broadcast channel state

Signed-off-by: Camila Belo <camilaibs@gmail.com>
2024-03-12 11:49:16 +01:00
parent c3656eaa02
commit fa34ff03ae
2 changed files with 30 additions and 1 deletions
@@ -53,7 +53,9 @@ export function TechDocsAuthProvider({ children }: { children: ReactNode }) {
  const app = useApp();
  const { Progress } = app.getComponents();

-  const [channel] = useState(new BroadcastChannel('techdocs-cookie-refresh'));
+  const [channel] = useState(
+    () => new BroadcastChannel('techdocs-cookie-refresh'),
+  );

  const { loading, error, value, retry, refresh } = useTechDocsCookie();

@@ -115,8 +115,35 @@ const mountedRoutes = {

 describe('<TechDocsReaderPage />', () => {
  beforeEach(() => {
+    type Listener = (event: { data: any }) => void;
+
+    global.BroadcastChannel = jest
+      .fn()
+      .mockImplementation((_channelName: string) => {
+        let listeners: Listener[] = [];
+        return {
+          postMessage: jest.fn((message: any) => {
+            listeners.forEach(listener => listener({ data: message }));
+          }),
+          addEventListener: jest.fn((event: string, listener: Listener) => {
+            if (event === 'message') {
+              listeners.push(listener);
+            }
+          }),
+          removeEventListener: jest.fn((event: string, listener: Listener) => {
+            if (event === 'message') {
+              listeners = listeners.filter(l => l !== listener);
+            }
+          }),
+        };
+      });
+
    getEntityMetadata.mockResolvedValue(mockEntityMetadata);
    getTechDocsMetadata.mockResolvedValue(mockTechDocsMetadata);
+    getCookie.mockResolvedValue({
+      // Expires in 10 minutes
+      expiresAt: new Date(Date.now() + 10 * 60 * 1000).toISOString(),
+    });
  });

  afterEach(() => {