fix: chat.test.ts

Author: khushal87

package/src/components/Chat/__tests__/Chat.test.js

@@ -222,79 +222,77 @@ describe('TranslationContext', () => {
       expect(context.tDateTimeParser).not.toBe(i18nInstance.tDateTimeParser);
       expect(context.tDateTimeParser).toBe(newI18nInstance.tDateTimeParser);
     });
+  });
+
+  it('makes sure DBSyncManager listeners are cleaned up after Chat remount', async () => {
+    const chatClientWithUser = await getTestClientWithUser({ id: 'testID' });
+    jest.spyOn(DBSyncManager, 'init');
+
+    // initial mount and render
+    const { rerender } = render(<Chat client={chatClientWithUser} enableOfflineSupport key={1} />);
+
+    // the unsubscribe fn changes during init(), so we keep a reference to the spy
+    const unsubscribeSpy = jest.spyOn(DBSyncManager.connectionChangedListener, 'unsubscribe');
+    const listenersAfterInitialMount = chatClientWithUser.listeners['connection.changed'];
 
-    it('makes sure DBSyncManager listeners are cleaned up after Chat remount', async () => {
-      const chatClientWithUser = await getTestClientWithUser({ id: 'testID' });
-      jest.spyOn(DBSyncManager, 'init');
+    // remount
+    rerender(<Chat client={chatClientWithUser} enableOfflineSupport key={2} />);
 
-      // initial mount and render
-      const { rerender } = render(
-        <Chat client={chatClientWithUser} enableOfflineSupport key={1} />,
+    await waitFor(() => {
+      expect(DBSyncManager.init).toHaveBeenCalledTimes(2);
+      expect(unsubscribeSpy).toHaveBeenCalledTimes(2);
+      expect(chatClientWithUser.listeners['connection.changed'].length).toBe(
+        listenersAfterInitialMount.length,
       );
+    });
+  });
 
-      // the unsubscribe fn changes during init(), so we keep a reference to the spy
-      const unsubscribeSpy = jest.spyOn(DBSyncManager.connectionChangedListener, 'unsubscribe');
-      const listenersAfterInitialMount = chatClientWithUser.listeners['connection.changed'];
+  it('makes sure DBSyncManager listeners are cleaned up if the user changes', async () => {
+    const chatClientWithUser = await getTestClientWithUser({ id: 'testID1' });
+    jest.spyOn(DBSyncManager, 'init');
 
-      // remount
-      rerender(<Chat client={chatClientWithUser} enableOfflineSupport key={2} />);
+    // initial render
+    const { rerender } = render(<Chat client={chatClientWithUser} enableOfflineSupport />);
 
-      await waitFor(() => {
-        expect(DBSyncManager.init).toHaveBeenCalledTimes(2);
-        expect(unsubscribeSpy).toHaveBeenCalledTimes(2);
-        expect(chatClientWithUser.listeners['connection.changed'].length).toBe(
-          listenersAfterInitialMount.length,
-        );
-      });
+    // the unsubscribe fn changes during init(), so we keep a reference to the spy
+    const unsubscribeSpy = jest.spyOn(DBSyncManager.connectionChangedListener, 'unsubscribe');
+    await act(async () => {
+      await setUser(chatClientWithUser, { id: 'testID2' });
     });
+    const listenersAfterInitialMount = chatClientWithUser.listeners['connection.changed'];
 
-    it('makes sure DBSyncManager listeners are cleaned up if the user changes', async () => {
-      const chatClientWithUser = await getTestClientWithUser({ id: 'testID1' });
-      jest.spyOn(DBSyncManager, 'init');
-
-      // initial render
-      const { rerender } = render(<Chat client={chatClientWithUser} enableOfflineSupport />);
-
-      // the unsubscribe fn changes during init(), so we keep a reference to the spy
-      const unsubscribeSpy = jest.spyOn(DBSyncManager.connectionChangedListener, 'unsubscribe');
-      await act(async () => {
-        await setUser(chatClientWithUser, { id: 'testID2' });
-      });
-      const listenersAfterInitialMount = chatClientWithUser.listeners['connection.changed'];
-
-      // rerender with different user ID
-      rerender(<Chat client={chatClientWithUser} enableOfflineSupport />);
-
-      await waitFor(() => {
-        expect(DBSyncManager.init).toHaveBeenCalledTimes(2);
-        expect(unsubscribeSpy).toHaveBeenCalledTimes(1);
-        expect(chatClientWithUser.listeners['connection.changed'].length).toBe(
-          listenersAfterInitialMount.length,
-        );
-      });
+    // rerender with different user ID
+    rerender(<Chat client={chatClientWithUser} enableOfflineSupport />);
+
+    await waitFor(() => {
+      expect(DBSyncManager.init).toHaveBeenCalledTimes(2);
+      expect(unsubscribeSpy).toHaveBeenCalledTimes(1);
+      expect(chatClientWithUser.listeners['connection.changed'].length).toBe(
+        listenersAfterInitialMount.length,
+      );
     });
+  });
 
-    it('makes sure DBSyncManager state stays intact during normal rerenders', async () => {
-      const chatClientWithUser = await getTestClientWithUser({ id: 'testID' });
-      jest.spyOn(DBSyncManager, 'init');
+  it('makes sure DBSyncManager state stays intact during normal rerenders', async () => {
+    const chatClientWithUser = await getTestClientWithUser({ id: 'testID' });
+    jest.spyOn(DBSyncManager, 'init');
 
-      // initial render
-      const { rerender } = render(<Chat client={chatClientWithUser} enableOfflineSupport />);
+    // initial render
+    const { rerender } = render(<Chat client={chatClientWithUser} enableOfflineSupport />);
 
-      // the unsubscribe fn changes during init(), so we keep a reference to the spy
-      const unsubscribeSpy = jest.spyOn(DBSyncManager.connectionChangedListener, 'unsubscribe');
-      const listenersAfterInitialMount = chatClientWithUser.listeners['connection.changed'];
+    // the unsubscribe fn changes during init(), so we keep a reference to the spy
+    const unsubscribeSpy = jest.spyOn(DBSyncManager.connectionChangedListener, 'unsubscribe');
+    const listenersAfterInitialMount = chatClientWithUser.listeners['connection.changed'];
 
-      // rerender
-      rerender(<Chat client={chatClientWithUser} enableOfflineSupport />);
+    // rerender
+    rerender(<Chat client={chatClientWithUser} enableOfflineSupport />);
 
-      await waitFor(() => {
-        expect(DBSyncManager.init).toHaveBeenCalledTimes(1);
-        expect(unsubscribeSpy).toHaveBeenCalledTimes(0);
-        expect(chatClientWithUser.listeners['connection.changed'].length).toBe(
-          listenersAfterInitialMount.length,
-        );
-      });
+    await waitFor(() => {
+      expect(DBSyncManager.init).toHaveBeenCalledTimes(1);
+      expect(unsubscribeSpy).toHaveBeenCalledTimes(0);
+      expect(chatClientWithUser.listeners['connection.changed'].length).toBe(
+        listenersAfterInitialMount.length,
+      );
     });
   });
 });