packages/components/hooks/useMozillaCheck.spec.ts

   1 import { renderHook } from '@testing-library/react-hooks';
   2
   3 import { useSubscription } from '@proton/account/subscription/hooks';
   4 import { FREE_SUBSCRIPTION } from '@proton/payments';
   5
   6 import useMozillaCheck from './useMozillaCheck';
   7
   8 jest.mock('@proton/account/subscription/hooks');
   9
  10 it('should return false if subscription is still loading', () => {
  11     jest.mocked(useSubscription).mockReturnValue([undefined as any, true]);
  12
  13     const { result } = renderHook(() => useMozillaCheck());
  14
  15     expect(result.current).toEqual([false, true]);
  16 });
  17
  18 it('should return false if not managed by Mozilla', () => {
  19     jest.mocked(useSubscription).mockReturnValue([{ isManagedByMozilla: false } as any, false]);
  20
  21     const { result } = renderHook(() => useMozillaCheck());
  22
  23     expect(result.current).toEqual([false, false]);
  24 });
  25
  26 it('should return true if managed by Mozilla', () => {
  27     jest.mocked(useSubscription).mockReturnValue([{ isManagedByMozilla: true } as any, false]);
  28
  29     const { result } = renderHook(() => useMozillaCheck());
  30
  31     expect(result.current).toEqual([true, false]);
  32 });
  33
  34 it('should return false if subscription is free', () => {
  35     jest.mocked(useSubscription).mockReturnValue([FREE_SUBSCRIPTION as any, false]);
  36
  37     const { result } = renderHook(() => useMozillaCheck());
  38
  39     expect(result.current).toEqual([false, false]);
  40 });