packages/components/containers/addresses/AddressesWithUser.test.tsx

   1 import type { ComponentPropsWithoutRef } from 'react';
   2
   3 import { fireEvent, render } from '@testing-library/react';
   4
   5 import { useAddressesKeys } from '@proton/account/addressKeys/hooks';
   6 import { useAddresses } from '@proton/account/addresses/hooks';
   7 import { useOrganizationKey } from '@proton/account/organizationKey/hooks';
   8 import { useUser } from '@proton/account/user/hooks';
   9 import { useApi, useKTVerifier, useNotifications } from '@proton/components';
  10 import OrderableTable from '@proton/components/components/orderableTable/OrderableTable';
  11 import { orderAddress } from '@proton/shared/lib/api/addresses';
  12 import { ADDRESS_TYPE } from '@proton/shared/lib/constants';
  13 import type { Address, UserModel } from '@proton/shared/lib/interfaces';
  14 import { mockUseFeature } from '@proton/testing/lib/mockUseFeature';
  15 import useFlag from '@proton/unleash/useFlag';
  16
  17 import AddressesWithUser from './AddressesWithUser';
  18
  19 jest.mock('@proton/components/hooks/useEventManager', () => () => ({}));
  20 jest.mock('@proton/components/components/upsell/useOneDollarPromo.tsx', () => ({
  21     __esModule: true,
  22     default: () => null,
  23 }));
  24
  25 jest.mock('@proton/components/components/upsell/useUpsellConfig.ts', () => ({
  26     __esModule: true,
  27     default: () => null,
  28 }));
  29
  30 jest.mock('@proton/shared/lib/helpers/upsell.ts', () => ({
  31     __esModule: true,
  32     useNewUpsellModalVariant: () => true,
  33     getUpsellRef: () => '',
  34 }));
  35
  36 jest.mock('@proton/components/components/orderableTable/OrderableTable');
  37 const ActualOrderableTable = jest.requireActual('@proton/components/components/orderableTable/OrderableTable').default;
  38 const mockedOrderableTable = OrderableTable as jest.MockedFunction<typeof OrderableTable>;
  39
  40 jest.mock('@proton/account/addresses/hooks');
  41 const mockedUseAddresses = useAddresses as jest.MockedFunction<typeof useAddresses>;
  42
  43 jest.mock('@proton/components/hooks/useApi');
  44 const mockedUseApi = useApi as jest.MockedFunction<typeof useApi>;
  45
  46 jest.mock('@proton/components/hooks/useNotifications');
  47 const mockedUseNotifications = useNotifications as jest.MockedFunction<typeof useNotifications>;
  48
  49 jest.mock('@proton/account/user/hooks');
  50 const mockedUseUser = useUser as jest.MockedFunction<typeof useUser>;
  51
  52 jest.mock('@proton/account/addressKeys/hooks');
  53 const mockedUseAddressesKeys = useAddressesKeys as jest.MockedFunction<typeof useAddressesKeys>;
  54
  55 jest.mock('@proton/components/containers/keyTransparency/useKTVerifier');
  56 const mockedUseKTVerifier = useKTVerifier as jest.MockedFunction<typeof useKTVerifier>;
  57
  58 jest.mock('@proton/unleash/useFlag');
  59 const mockedUseFlag = useFlag as jest.MockedFunction<any>;
  60
  61 jest.mock('@proton/account/organizationKey/hooks');
  62 const mockedUseOrganizationKey = useOrganizationKey as jest.MockedFunction<typeof useOrganizationKey>;
  63
  64 jest.mock('@proton/redux-shared-store');
  65
  66 describe('addresses with user', () => {
  67     const user = {
  68         ID: 'abc',
  69     } as UserModel;
  70
  71     const addresses = [
  72         {
  73             ID: '1',
  74             Email: 'a@proton.me',
  75             Type: ADDRESS_TYPE.TYPE_ORIGINAL,
  76             Status: 1,
  77             Receive: 1,
  78             Send: 1,
  79             HasKeys: 1,
  80         },
  81         {
  82             ID: '2',
  83             Email: 'a@foo.bar',
  84             Type: ADDRESS_TYPE.TYPE_EXTERNAL,
  85             Status: 1,
  86             Receive: 0,
  87             Send: 0,
  88             HasKeys: 1,
  89         },
  90         {
  91             ID: '3',
  92             Email: 'a1@proton.me',
  93             Type: ADDRESS_TYPE.TYPE_ALIAS,
  94             Status: 1,
  95             Receive: 1,
  96             Send: 1,
  97             HasKeys: 1,
  98         },
  99         {
 100             ID: '4',
 101             Email: 'a2@pm.me',
 102             Type: ADDRESS_TYPE.TYPE_PREMIUM,
 103             Status: 0,
 104             Receive: 0,
 105             Send: 0,
 106             HasKeys: 1,
 107         },
 108     ] as Address[];
 109
 110     mockedUseAddresses.mockReturnValue([addresses, false]);
 111     mockedOrderableTable.mockImplementation(ActualOrderableTable);
 112     mockedUseNotifications.mockReturnValue({} as any);
 113     mockedUseUser.mockReturnValue([{}] as any);
 114     mockedUseAddressesKeys.mockReturnValue([{}] as any);
 115     mockedUseKTVerifier.mockReturnValue({} as any);
 116     mockedUseOrganizationKey.mockReturnValue([{}] as any);
 117     mockedUseFlag.mockReturnValue(true);
 118     mockUseFeature({ feature: { Value: true } as any });
 119
 120     const getFirstAddress = (container: HTMLElement) => {
 121         return container.querySelector('[title]');
 122     };
 123
 124     it('should be able to set an alias as default', async () => {
 125         const mockApi = jest.fn();
 126         mockedUseApi.mockReturnValue(mockApi);
 127
 128         const { getByTestId, getByText, container } = render(
 129             <AddressesWithUser user={user} allowAddressDeletion={false} />
 130         );
 131
 132         // Assumes that "Make default" is displayed on the address we're interested in
 133         fireEvent.click(getByTestId('dropdownActions:dropdown'));
 134         fireEvent.click(getByText('Set as default'));
 135         expect(mockApi).toHaveBeenCalledWith(orderAddress(['3', '1', '2', '4']));
 136         expect(getFirstAddress(container)?.innerHTML).toBe('a1@proton.me');
 137     });
 138
 139     describe('set as default', () => {
 140         const setup = () => {
 141             const createNotification = jest.fn();
 142             mockedUseNotifications.mockReturnValue({ createNotification } as any);
 143             let onSortEnd: ComponentPropsWithoutRef<typeof OrderableTable>['onSortEnd'];
 144             mockedOrderableTable.mockImplementation((props) => {
 145                 onSortEnd = props.onSortEnd;
 146                 return <ActualOrderableTable {...props} />;
 147             });
 148             const handleSortEnd: typeof onSortEnd = (...args) => {
 149                 onSortEnd?.(...args);
 150             };
 151             return { onSortEnd: handleSortEnd, createNotification };
 152         };
 153
 154         it('should not be able to set an external address as default', () => {
 155             const { onSortEnd, createNotification } = setup();
 156             const { container } = render(<AddressesWithUser user={user} allowAddressDeletion={false} />);
 157
 158             onSortEnd({ newIndex: 0, oldIndex: 2 } as any, {} as any);
 159             expect(createNotification).toHaveBeenCalledWith({
 160                 type: 'error',
 161                 text: 'An external address cannot be default',
 162             });
 163             expect(getFirstAddress(container)?.innerHTML).toBe('a@proton.me');
 164         });
 165
 166         it('should not be able to set a disabled address as default', () => {
 167             const { onSortEnd, createNotification } = setup();
 168             const { container } = render(<AddressesWithUser user={user} allowAddressDeletion={false} />);
 169
 170             onSortEnd({ newIndex: 0, oldIndex: 3 } as any, {} as any);
 171             expect(createNotification).toHaveBeenCalledWith({
 172                 type: 'error',
 173                 text: 'A disabled address cannot be default',
 174             });
 175             expect(getFirstAddress(container)?.innerHTML).toBe('a@proton.me');
 176         });
 177     });
 178 });