packages/wallet/tests/mocks/useBitcoinNetwork.ts

   1 import { WasmNetwork } from '@proton/andromeda';
   2
   3 import * as useBitcoinNetworkModule from '../../store/hooks/useBitcoinNetwork';
   4
   5 export const mockUseBitcoinNetwork = (
   6     mockedValue?: Partial<ReturnType<typeof useBitcoinNetworkModule.useBitcoinNetwork>>
   7 ) => {
   8     const spy = vi.spyOn(useBitcoinNetworkModule, 'useBitcoinNetwork');
   9
  10     spy.mockReturnValue([mockedValue?.[0] ?? WasmNetwork.Testnet, mockedValue?.[1] ?? false]);
  11
  12     return spy;
  13 };
  14
  15 export const mockUseGetBitcoinNetwork = (mockedValue?: WasmNetwork) => {
  16     const spy = vi.spyOn(useBitcoinNetworkModule, 'useGetBitcoinNetwork');
  17
  18     spy.mockImplementation(() => async () => mockedValue ?? WasmNetwork.Testnet);
  19
  20     return spy;
  21 };