packages/components/hooks/useInboxDesktopBadgeCount.test.ts

   1 import { renderHook } from '@testing-library/react-hooks';
   2
   3 import useInboxDesktopBadgeCount from '@proton/components/hooks/useInboxDesktopBadgeCount';
   4 import { useConversationCounts } from '@proton/mail/counts/conversationCounts';
   5 import { useMessageCounts } from '@proton/mail/counts/messageCounts';
   6 import { useMailSettings } from '@proton/mail/mailSettings/hooks';
   7 import { MAILBOX_LABEL_IDS } from '@proton/shared/lib/constants';
   8 import * as desktopHelpers from '@proton/shared/lib/helpers/desktop';
   9 import { VIEW_MODE } from '@proton/shared/lib/mail/mailSettings';
  10
  11 jest.mock('@proton/mail/mailSettings/hooks');
  12 jest.mock('@proton/mail/counts/conversationCounts');
  13 jest.mock('@proton/mail/counts/messageCounts');
  14 jest.mock('@proton/shared/lib/helpers/desktop');
  15 const desktopHelpersMock = desktopHelpers as jest.MockedObject<typeof desktopHelpers>;
  16
  17 declare const global: {
  18     ipcInboxMessageBroker?: any;
  19 };
  20
  21 const originalWindow = { ...window };
  22
  23 describe('useInboxDesktopBadgeCount', () => {
  24     const useConversationCountsMock = useConversationCounts as jest.Mock;
  25     const useMessageCountsMock = useMessageCounts as jest.Mock;
  26     const useMailSettingsMock = useMailSettings as jest.Mock;
  27     const ipcInboxMessageBrokerMock = {
  28         send: jest.fn(),
  29     };
  30
  31     beforeEach(() => {
  32         // @ts-ignore
  33         const windowSpy = jest.spyOn(global, 'window', 'get');
  34         windowSpy.mockReturnValue({ ...originalWindow, ipcInboxMessageBroker: ipcInboxMessageBrokerMock });
  35     });
  36
  37     afterEach(() => {
  38         jest.clearAllMocks();
  39     });
  40
  41     it('should not call when not on desktop', () => {
  42         desktopHelpersMock.isElectronMail = false;
  43         useConversationCountsMock.mockReturnValue([]);
  44         useMessageCountsMock.mockReturnValue([]);
  45
  46         renderHook(() => useInboxDesktopBadgeCount());
  47         expect(ipcInboxMessageBrokerMock.send).not.toHaveBeenCalled();
  48     });
  49
  50     it('should call with 0 when no count', () => {
  51         desktopHelpersMock.isElectronMail = true;
  52         useMailSettingsMock.mockReturnValue([{ ViewMode: VIEW_MODE.GROUP }]);
  53         useConversationCountsMock.mockReturnValue([[]]);
  54         useMessageCountsMock.mockReturnValue([]);
  55
  56         renderHook(() => useInboxDesktopBadgeCount());
  57         expect(ipcInboxMessageBrokerMock.send).toHaveBeenCalledWith('updateNotification', 0);
  58     });
  59
  60     it('should call with 0 when negative count', () => {
  61         desktopHelpersMock.isElectronMail = true;
  62         useMailSettingsMock.mockReturnValue([{ ViewMode: VIEW_MODE.GROUP }]);
  63         useConversationCountsMock.mockReturnValue([[{ LabelID: MAILBOX_LABEL_IDS.INBOX, Unread: -1, Total: 1 }]]);
  64         useMessageCountsMock.mockReturnValue([]);
  65
  66         renderHook(() => useInboxDesktopBadgeCount());
  67         expect(ipcInboxMessageBrokerMock.send).toHaveBeenCalledWith('updateNotification', 0);
  68     });
  69
  70     it('should call with 1 when 1 unread conversation', () => {
  71         desktopHelpersMock.isElectronMail = true;
  72         useMailSettingsMock.mockReturnValue([{ ViewMode: VIEW_MODE.GROUP }]);
  73         useConversationCountsMock.mockReturnValue([[{ LabelID: MAILBOX_LABEL_IDS.INBOX, Unread: 1, Total: 1 }]]);
  74         useMessageCountsMock.mockReturnValue([]);
  75
  76         renderHook(() => useInboxDesktopBadgeCount());
  77         expect(ipcInboxMessageBrokerMock.send).toHaveBeenCalledWith('updateNotification', 1);
  78     });
  79
  80     it('should call with 100 when 100 unread conversation', () => {
  81         desktopHelpersMock.isElectronMail = true;
  82         useMailSettingsMock.mockReturnValue([{ ViewMode: VIEW_MODE.GROUP }]);
  83         useConversationCountsMock.mockReturnValue([[{ LabelID: MAILBOX_LABEL_IDS.INBOX, Unread: 100, Total: 100 }]]);
  84         useMessageCountsMock.mockReturnValue([]);
  85
  86         renderHook(() => useInboxDesktopBadgeCount());
  87         expect(ipcInboxMessageBrokerMock.send).toHaveBeenCalledWith('updateNotification', 100);
  88     });
  89
  90     it('should call with 1 when 1 unread message', () => {
  91         desktopHelpersMock.isElectronMail = true;
  92         useMailSettingsMock.mockReturnValue([{ ViewMode: VIEW_MODE.SINGLE }]);
  93         useConversationCountsMock.mockReturnValue([]);
  94         useMessageCountsMock.mockReturnValue([[{ LabelID: MAILBOX_LABEL_IDS.INBOX, Unread: 1, Total: 1 }]]);
  95
  96         renderHook(() => useInboxDesktopBadgeCount());
  97         expect(ipcInboxMessageBrokerMock.send).toHaveBeenCalledWith('updateNotification', 1);
  98     });
  99
 100     it('should call with 100 when 100 unread message', () => {
 101         desktopHelpersMock.isElectronMail = true;
 102         useMailSettingsMock.mockReturnValue([{ ViewMode: VIEW_MODE.SINGLE }]);
 103         useConversationCountsMock.mockReturnValue([]);
 104         useMessageCountsMock.mockReturnValue([[{ LabelID: MAILBOX_LABEL_IDS.INBOX, Unread: 100, Total: 100 }]]);
 105
 106         renderHook(() => useInboxDesktopBadgeCount());
 107         expect(ipcInboxMessageBrokerMock.send).toHaveBeenCalledWith('updateNotification', 100);
 108     });
 109 });