@ledgerhq/coin-tezos 7.12.0 → 8.0.0

package/src/api/index.test.ts

@@ -1,12 +1,22 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
+import type { TezosCoinConfig, TezosContext } from '../config'
 import type { APIAccount } from '../network/types'
 import type { BalanceOptions, Operation } from '@ledgerhq/coin-module-framework/api/types'
 import type { TransactionIntent } from '@ledgerhq/coin-module-framework/api/types'
-import networkApi from '../network/tzkt'
 import { createApi } from './index'
 
+// The tzkt factory `createTzktApi(config)` builds a fresh object on every call, so a spy on a
+// single instance would not intercept the calls the source makes internally. We mock the module
+// so `createTzktApi` always returns the SAME shared object, and expose its `getAccountByAddress`
+// jest.fn as `mockGetAccountByAddress` for the assertions below (which used to target
+// `mockGetAccountByAddress`).
+const mockGetAccountByAddress = jest.fn()
+jest.mock('../network/tzkt', () => ({
+  createTzktApi: () => ({ getAccountByAddress: mockGetAccountByAddress }),
+}))
+
 const DEFAULT_ESTIMATED_FEES = 300n
 const DEFAULT_GAS_LIMIT = 30n
 const DEFAULT_STORAGE_LIMIT = 40n
@@ -14,7 +24,7 @@ const DEFAULT_STORAGE_LIMIT = 40n
 const logicGetTransactions = jest.fn()
 const logicEstimateFees = jest.fn()
 const logicCraftTransactionMock = jest.fn(
-  (_account: unknown, _transaction: { fee: { fees: string } }) => {
+  (_context: unknown, _account: unknown, _transaction: { fee: { fees: string } }) => {
     return { type: undefined, contents: undefined }
   }
 )
@@ -23,8 +33,8 @@ const logicCraftRawOperationsMock = jest.fn()
 jest.mock('../logic', () => ({
   listOperations: async () => logicGetTransactions(),
   estimateFees: (...args: unknown[]) => logicEstimateFees(...args),
-  craftTransaction: (account: unknown, transaction: { fee: { fees: string } }) =>
-    logicCraftTransactionMock(account, transaction),
+  craftTransaction: (context: unknown, account: unknown, transaction: { fee: { fees: string } }) =>
+    logicCraftTransactionMock(context, account, transaction),
   craftRawOperations: (...args: unknown[]) => logicCraftRawOperationsMock(...args),
   rawEncode: () => Promise.resolve('tz1heMGVHQnx7ALDcDKqez8fan64Eyicw4DJ'),
 }))
@@ -34,7 +44,7 @@ jest.mock('../logic/tezosToolkit', () => ({
   getTezosToolkit: () => mockGetTezosToolkit(),
 }))
 
-jest.spyOn(networkApi, 'getAccountByAddress').mockResolvedValue({
+mockGetAccountByAddress.mockResolvedValue({
   type: 'user',
   balance: 1000,
   revealed: true,
@@ -47,7 +57,17 @@ jest.spyOn(networkApi, 'getAccountByAddress').mockResolvedValue({
   firstActivityTime: '2021-01-01T00:00:00Z',
 } as APIAccount)
 
-const api = createApi({
+const api = createApi()
+
+/** Builds a TezosContext carrying the given coin config (see logic/getBalance.test.ts). */
+function makeContext(config: unknown): TezosContext {
+  return {
+    config: async () => config as TezosCoinConfig,
+    logger: () => undefined,
+  }
+}
+
+const context = makeContext({
   baker: {
     url: 'https://baker.example.com',
   },
@@ -76,9 +96,16 @@ describe('craftRawTransaction', () => {
     logicCraftRawOperationsMock.mockResolvedValue('03deadbeef')
     const rawTransaction = '[{"kind":"transaction","destination":"tz1recipient","amount":"1000"}]'
 
-    const result = await api.craftRawTransaction(rawTransaction, 'tz1sender', 'edpkpublickey', 5n)
+    const result = await api.craftRawTransaction(
+      context,
+      rawTransaction,
+      'tz1sender',
+      'edpkpublickey',
+      5n
+    )
 
     expect(logicCraftRawOperationsMock).toHaveBeenCalledWith(
+      expect.anything(),
       rawTransaction,
       'tz1sender',
       'edpkpublickey',
@@ -95,7 +122,7 @@ describe('get operations', () => {
 
   it('could return no operation', async () => {
     logicGetTransactions.mockResolvedValue([[], ''])
-    const { items: operations, next: token } = await api.listOperations('addr', {
+    const { items: operations, next: token } = await api.listOperations(context, 'addr', {
       minHeight: 100,
       order: 'asc',
     })
@@ -125,7 +152,7 @@ describe('get operations', () => {
 
   it('only does 1 iteration', async () => {
     logicGetTransactions.mockResolvedValue([[op], '888'])
-    const { items: operations, next: token } = await api.listOperations('addr', {
+    const { items: operations, next: token } = await api.listOperations(context, 'addr', {
       minHeight: 100,
       order: 'asc',
     })
@@ -151,7 +178,7 @@ describe('craftTransaction', () => {
           txFee: 500n,
         },
       })
-      await api.craftTransaction({
+      await api.craftTransaction(context, {
         intentType: 'transaction',
         type: 'send',
         sender: 'tz1test',
@@ -160,6 +187,7 @@ describe('craftTransaction', () => {
       } as TransactionIntent)
       expect(logicEstimateFees).toHaveBeenCalledTimes(1)
       expect(logicCraftTransactionMock).toHaveBeenCalledWith(
+        expect.anything(),
         expect.objectContaining({ address: 'tz1test' }),
         expect.objectContaining({
           fee: expect.objectContaining({
@@ -183,7 +211,7 @@ describe('craftTransaction', () => {
           txFee: 50n,
         },
       })
-      await api.craftTransaction({
+      await api.craftTransaction(context, {
         intentType: 'transaction',
         type: 'send',
         sender: 'tz1test',
@@ -191,6 +219,7 @@ describe('craftTransaction', () => {
         amount: 1000n,
       } as TransactionIntent)
       expect(logicCraftTransactionMock).toHaveBeenCalledWith(
+        expect.anything(),
         expect.any(Object),
         expect.objectContaining({
           fee: expect.objectContaining({ fees: estimatedFees.toString() }),
@@ -213,6 +242,7 @@ describe('craftTransaction', () => {
         },
       })
       await api.craftTransaction(
+        context,
         {
           intentType: 'transaction',
           type: 'send',
@@ -220,9 +250,10 @@ describe('craftTransaction', () => {
           recipient: 'tz1recipient',
           amount: 1000n,
         } as TransactionIntent,
-        { value: customFee }
+        { customFees: { value: customFee } }
       )
       expect(logicCraftTransactionMock).toHaveBeenCalledWith(
+        expect.anything(),
         expect.any(Object),
         expect.objectContaining({
           fee: expect.objectContaining({ fees: customFee.toString() }),
@@ -242,6 +273,7 @@ describe('craftTransaction', () => {
         },
       })
       await api.craftTransaction(
+        context,
         {
           intentType: 'transaction',
           type: 'send',
@@ -249,9 +281,10 @@ describe('craftTransaction', () => {
           recipient: 'tz1recipient',
           amount: 1000n,
         } as TransactionIntent,
-        { value: customFee }
+        { customFees: { value: customFee } }
       )
       expect(logicCraftTransactionMock).toHaveBeenCalledWith(
+        expect.anything(),
         expect.any(Object),
         expect.objectContaining({
           fee: expect.objectContaining({ fees: customFee.toString() }),
@@ -265,7 +298,7 @@ describe('craftTransaction', () => {
       const delegateRecipient = 'tz3Vq38qYD3GEbWcXHMLt5PaASZrkDtEiA8D'
       const totalCustomFee = BigInt(minFees * 2)
 
-      const apiMinFees1000 = createApi({
+      const contextMinFees1000 = makeContext({
         baker: { url: 'https://baker.example.com' },
         explorer: { url: 'foo', maxTxQuery: 1 },
         node: { url: 'bar' },
@@ -290,7 +323,7 @@ describe('craftTransaction', () => {
         numTransactions: 0,
         firstActivityTime: '2021-01-01T00:00:00Z',
       } as APIAccount
-      ;(networkApi.getAccountByAddress as jest.Mock)
+      mockGetAccountByAddress
         .mockResolvedValueOnce(unrevealedAccount)
         .mockResolvedValueOnce(unrevealedAccount)
 
@@ -306,7 +339,8 @@ describe('craftTransaction', () => {
         },
       })
 
-      await apiMinFees1000.craftTransaction(
+      await api.craftTransaction(
+        contextMinFees1000,
         {
           intentType: 'staking',
           type: 'delegate',
@@ -319,6 +353,7 @@ describe('craftTransaction', () => {
       )
 
       expect(logicCraftTransactionMock).toHaveBeenCalledWith(
+        expect.anything(),
         expect.objectContaining({ address: unrevealedSender }),
         expect.objectContaining({
           fee: expect.objectContaining({
@@ -340,7 +375,7 @@ describe('craftTransaction', () => {
         storageLimit: DEFAULT_STORAGE_LIMIT,
       })
 
-      await api.craftTransaction({
+      await api.craftTransaction(context, {
         intentType: 'staking',
         type,
         sender: 'tz1test',
@@ -349,6 +384,7 @@ describe('craftTransaction', () => {
       } as TransactionIntent)
 
       expect(logicCraftTransactionMock).toHaveBeenCalledWith(
+        expect.anything(),
         expect.objectContaining({ address: 'tz1test' }),
         expect.objectContaining({
           type,
@@ -366,7 +402,7 @@ describe('craftTransaction', () => {
       storageLimit: DEFAULT_STORAGE_LIMIT,
     })
 
-    await api.craftTransaction({
+    await api.craftTransaction(context, {
       intentType: 'staking',
       type: 'finalize_unstake',
       sender: 'tz1test',
@@ -375,6 +411,7 @@ describe('craftTransaction', () => {
     } as TransactionIntent)
 
     expect(logicCraftTransactionMock).toHaveBeenCalledWith(
+      expect.anything(),
       expect.objectContaining({ address: 'tz1test' }),
       expect.objectContaining({
         type: 'finalize_unstake',
@@ -395,7 +432,7 @@ describe('estimateFees', () => {
         gasLimit: DEFAULT_GAS_LIMIT,
         storageLimit: DEFAULT_STORAGE_LIMIT,
       })
-      const result = await api.estimateFees({
+      const result = await api.estimateFees(context, {
         intentType: 'transaction',
         type: 'send',
         sender: 'tz1test',
@@ -414,7 +451,7 @@ describe('estimateFees', () => {
         gasLimit: DEFAULT_GAS_LIMIT,
         storageLimit: DEFAULT_STORAGE_LIMIT,
       })
-      const result = await api.estimateFees({
+      const result = await api.estimateFees(context, {
         intentType: 'transaction',
         type: 'send',
         sender: 'tz1test',
@@ -433,7 +470,7 @@ describe('estimateFees', () => {
       gasLimit: DEFAULT_GAS_LIMIT,
       storageLimit: DEFAULT_STORAGE_LIMIT,
     })
-    const result = await api.estimateFees({
+    const result = await api.estimateFees(context, {
       intentType: 'transaction',
       type: 'send',
       sender: 'tz1test',
@@ -450,7 +487,7 @@ describe('estimateFees', () => {
   })
 
   it('forwards stakedBalance and unstakedBalance to the logic estimate for send-max', async () => {
-    ;(networkApi.getAccountByAddress as jest.Mock).mockResolvedValueOnce({
+    mockGetAccountByAddress.mockResolvedValueOnce({
       type: 'user',
       balance: 1000,
       revealed: true,
@@ -471,7 +508,7 @@ describe('estimateFees', () => {
       amount: 500n,
     })
 
-    await api.estimateFees({
+    await api.estimateFees(context, {
       intentType: 'transaction',
       type: 'send',
       sender: 'tz1test',
@@ -481,6 +518,7 @@ describe('estimateFees', () => {
     } as TransactionIntent)
 
     expect(logicEstimateFees).toHaveBeenCalledWith(
+      expect.anything(),
       expect.objectContaining({
         account: expect.objectContaining({ stakedBalance: 200n, unstakedBalance: 300n }),
       })
@@ -495,7 +533,7 @@ describe('estimateFees', () => {
       taquitoError: 'test',
     })
     await expect(
-      api.estimateFees({
+      api.estimateFees(context, {
         intentType: 'transaction',
         type: 'send',
         sender: 'tz1test',
@@ -512,7 +550,7 @@ describe('estimateFees', () => {
       storageLimit: DEFAULT_STORAGE_LIMIT,
       taquitoError: 'proto.024-PtTALLiN.michelson_v1.script_rejected',
     })
-    const result = await api.estimateFees({
+    const result = await api.estimateFees(context, {
       intentType: 'transaction',
       type: 'send',
       sender: 'tz1test',
@@ -532,7 +570,7 @@ describe('estimateFees', () => {
       storageLimit: DEFAULT_STORAGE_LIMIT,
       taquitoError: 'proto.022-PsRiotum.delegate.unchanged',
     })
-    const result = await api.estimateFees({
+    const result = await api.estimateFees(context, {
       intentType: 'staking',
       type: 'delegate',
       sender: 'tz1test',
@@ -557,7 +595,7 @@ describe('estimateFees', () => {
       taquitoError:
         'proto.alpha.cannot_stake_with_unfinalizable_unstake_requests_to_another_delegate',
     })
-    const result = await api.estimateFees({
+    const result = await api.estimateFees(context, {
       intentType: 'staking',
       type: 'stake',
       sender: 'tz1test',
@@ -576,7 +614,7 @@ describe('estimateFees', () => {
 
   it('returns total estimation for unrevealed delegate when minFees applied (composite)', async () => {
     const unrevealedSender = 'tz2TaTpo31sAiX2HBJUTLLdUnqVJR4QjLy1V'
-    ;(networkApi.getAccountByAddress as jest.Mock).mockResolvedValueOnce({
+    mockGetAccountByAddress.mockResolvedValueOnce({
       type: 'user',
       balance: 1000000,
       revealed: false,
@@ -597,7 +635,7 @@ describe('estimateFees', () => {
       storageLimit: DEFAULT_STORAGE_LIMIT,
     })
 
-    const result = await api.estimateFees({
+    const result = await api.estimateFees(context, {
       intentType: 'staking',
       type: 'delegate',
       sender: unrevealedSender,
@@ -622,7 +660,7 @@ describe('estimateFees', () => {
       storageLimit: DEFAULT_STORAGE_LIMIT,
     })
 
-    const result = await api.estimateFees({
+    const result = await api.estimateFees(context, {
       intentType: 'staking',
       type,
       sender: 'tz1test',
@@ -632,6 +670,7 @@ describe('estimateFees', () => {
 
     expect(result.value).toBe(DEFAULT_ESTIMATED_FEES)
     expect(logicEstimateFees).toHaveBeenCalledWith(
+      expect.anything(),
       expect.objectContaining({
         transaction: expect.objectContaining({
           mode: type,
@@ -646,7 +685,7 @@ describe('estimateFees', () => {
     const delegateRecipient = 'tz3Vq38qYD3GEbWcXHMLt5PaASZrkDtEiA8D'
     const minFees = 1000
 
-    const apiMinFees1000 = createApi({
+    const contextMinFees1000 = makeContext({
       baker: { url: 'https://baker.example.com' },
       explorer: { url: 'foo', maxTxQuery: 1 },
       node: { url: 'bar' },
@@ -695,7 +734,7 @@ describe('estimateFees', () => {
       numTransactions: 0,
       firstActivityTime: '2021-01-01T00:00:00Z',
     } as APIAccount
-    ;(networkApi.getAccountByAddress as jest.Mock)
+    mockGetAccountByAddress
       .mockImplementationOnce(() => Promise.resolve(defaultUserForTry))
       .mockImplementationOnce(() => Promise.resolve(recipientAccount))
       .mockImplementationOnce(() => Promise.resolve(senderUnrevealedAccount))
@@ -712,7 +751,7 @@ describe('estimateFees', () => {
       },
     })
 
-    const result = await apiMinFees1000.estimateFees({
+    const result = await api.estimateFees(contextMinFees1000, {
       intentType: 'staking',
       type: 'delegate',
       sender: unrevealedSender,
@@ -732,7 +771,7 @@ describe('estimateFees', () => {
     // SAFE_FALLBACK_GAS = 2500, FALLBACK_OP_SIZE_BYTES = 154
     // baseTxFee = max(minFees=1, ceil(120 + 0.1*2500 + 154)) = max(1, 524) = 524
     // createApi here to reset the global coinConfig to minFees=1 (other tests may leave it at a higher value)
-    const apiLowMinFees = createApi({
+    const contextLowMinFees = makeContext({
       baker: { url: 'https://baker.example.com' },
       explorer: { url: 'foo', maxTxQuery: 1 },
       node: { url: 'bar' },
@@ -772,7 +811,7 @@ describe('estimateFees', () => {
       firstActivityTime: '2021-01-01T00:00:00Z',
     } as APIAccount
 
-    ;(networkApi.getAccountByAddress as jest.Mock)
+    mockGetAccountByAddress
       .mockResolvedValueOnce(revealedAccount) // initial sender lookup
       .mockResolvedValueOnce(recipientAccount) // recipient lookup in catch block
       .mockResolvedValueOnce(revealedAccount) // sender lookup for needsReveal in catch block
@@ -785,7 +824,7 @@ describe('estimateFees', () => {
       },
     })
 
-    const result = await apiLowMinFees.estimateFees({
+    const result = await api.estimateFees(contextLowMinFees, {
       intentType: 'transaction',
       type: 'send',
       sender: revealedSender,
@@ -806,7 +845,7 @@ describe('estimateFees', () => {
     // baseTxFee = max(minFees=1, ceil(120 + 0.1*2500 + 154)) = 524
     // revealFee = max(0, getRevealFee(tz2address)) — tz2 addresses have a non-zero reveal fee
     // createApi here to reset the global coinConfig to minFees=1 (other tests may leave it at a higher value)
-    const apiLowMinFees = createApi({
+    const contextLowMinFees = makeContext({
       baker: { url: 'https://baker.example.com' },
       explorer: { url: 'foo', maxTxQuery: 1 },
       node: { url: 'bar' },
@@ -846,7 +885,7 @@ describe('estimateFees', () => {
       firstActivityTime: '2021-01-01T00:00:00Z',
     } as APIAccount
 
-    ;(networkApi.getAccountByAddress as jest.Mock)
+    mockGetAccountByAddress
       .mockResolvedValueOnce(unrevealedAccount) // initial sender lookup
       .mockResolvedValueOnce(recipientAccount) // recipient lookup in catch block
       .mockResolvedValueOnce(unrevealedAccount) // sender lookup for needsReveal in catch block
@@ -859,7 +898,7 @@ describe('estimateFees', () => {
       },
     })
 
-    const result = await apiLowMinFees.estimateFees({
+    const result = await api.estimateFees(contextLowMinFees, {
       intentType: 'transaction',
       type: 'send',
       sender: unrevealedSender,
@@ -880,7 +919,7 @@ describe('estimateFees', () => {
   it('uses minFees floor when it exceeds the static fallback calculation', async () => {
     // With minFees = 2000 > ceil(120 + 0.1*2500 + 154) = 524 → baseTxFee = 2000
     const minFees = 2000
-    const apiHighMinFees = createApi({
+    const contextHighMinFees = makeContext({
       baker: { url: 'https://baker.example.com' },
       explorer: { url: 'foo', maxTxQuery: 1 },
       node: { url: 'bar' },
@@ -918,7 +957,7 @@ describe('estimateFees', () => {
       firstActivityTime: '2021-01-01T00:00:00Z',
     } as APIAccount
 
-    ;(networkApi.getAccountByAddress as jest.Mock)
+    mockGetAccountByAddress
       .mockResolvedValueOnce(revealedAccount)
       .mockResolvedValueOnce(recipientAccount)
       .mockResolvedValueOnce(revealedAccount)
@@ -931,7 +970,7 @@ describe('estimateFees', () => {
       },
     })
 
-    const result = await apiHighMinFees.estimateFees({
+    const result = await api.estimateFees(contextHighMinFees, {
       intentType: 'transaction',
       type: 'send',
       sender: 'tz1test',
@@ -950,18 +989,18 @@ describe('estimateFees', () => {
 describe('getBalance', () => {
   it('should throw an exception when options is provided', async () => {
     await expect(
-      api.getBalance('random address', {} as unknown as BalanceOptions)
+      api.getBalance(context, 'random address', {} as unknown as BalanceOptions)
     ).rejects.toMatchObject({ name: 'InvalidParameterError' })
   })
 })
 
 describe('getAccountInfo', () => {
   afterEach(() => {
-    ;(networkApi.getAccountByAddress as jest.Mock).mockClear()
+    mockGetAccountByAddress.mockClear()
   })
 
   it('returns revealed:true for a revealed user account', async () => {
-    ;(networkApi.getAccountByAddress as jest.Mock).mockResolvedValueOnce({
+    mockGetAccountByAddress.mockResolvedValueOnce({
       type: 'user',
       balance: 1000,
       revealed: true,
@@ -974,15 +1013,15 @@ describe('getAccountInfo', () => {
       firstActivityTime: '2021-01-01T00:00:00Z',
     } as APIAccount)
 
-    await expect(api.getAccountInfo('tz1test')).resolves.toEqual({
+    await expect(api.getAccountInfo(context, 'tz1test')).resolves.toEqual({
       type: 'tezos',
       revealed: true,
     })
-    expect(networkApi.getAccountByAddress).toHaveBeenCalledWith('tz1test')
+    expect(mockGetAccountByAddress).toHaveBeenCalledWith('tz1test')
   })
 
   it('returns revealed:false for an unrevealed user account', async () => {
-    ;(networkApi.getAccountByAddress as jest.Mock).mockResolvedValueOnce({
+    mockGetAccountByAddress.mockResolvedValueOnce({
       type: 'user',
       balance: 1000,
       revealed: false,
@@ -995,17 +1034,17 @@ describe('getAccountInfo', () => {
       firstActivityTime: '2021-01-01T00:00:00Z',
     } as APIAccount)
 
-    await expect(api.getAccountInfo('tz2unrevealed')).resolves.toEqual({
+    await expect(api.getAccountInfo(context, 'tz2unrevealed')).resolves.toEqual({
       type: 'tezos',
       revealed: false,
     })
-    expect(networkApi.getAccountByAddress).toHaveBeenCalledWith('tz2unrevealed')
+    expect(mockGetAccountByAddress).toHaveBeenCalledWith('tz2unrevealed')
   })
 
   it('returns revealed:true for a registered baker (delegate) account', async () => {
     // tzkt reports bakers with type "delegate" (not "user"), still carrying revealed:true.
     // Regression test for LIVE-34256: bakers must not be reported as unrevealed.
-    ;(networkApi.getAccountByAddress as jest.Mock).mockResolvedValueOnce({
+    mockGetAccountByAddress.mockResolvedValueOnce({
       type: 'delegate',
       balance: 616109,
       revealed: true,
@@ -1014,32 +1053,32 @@ describe('getAccountInfo', () => {
       counter: 18,
     } as APIAccount)
 
-    await expect(api.getAccountInfo('tz3RDC3Jdn4j15J7bBHZd29EUee9gVB1CxD9')).resolves.toEqual({
+    await expect(
+      api.getAccountInfo(context, 'tz3RDC3Jdn4j15J7bBHZd29EUee9gVB1CxD9')
+    ).resolves.toEqual({
       type: 'tezos',
       revealed: true,
     })
-    expect(networkApi.getAccountByAddress).toHaveBeenCalledWith(
-      'tz3RDC3Jdn4j15J7bBHZd29EUee9gVB1CxD9'
-    )
+    expect(mockGetAccountByAddress).toHaveBeenCalledWith('tz3RDC3Jdn4j15J7bBHZd29EUee9gVB1CxD9')
   })
 
   it('returns revealed:false for an empty / non-existent account', async () => {
-    ;(networkApi.getAccountByAddress as jest.Mock).mockResolvedValueOnce({
+    mockGetAccountByAddress.mockResolvedValueOnce({
       type: 'empty',
       address: 'tz1empty',
       counter: 0,
     } as APIAccount)
 
-    await expect(api.getAccountInfo('tz1empty')).resolves.toEqual({
+    await expect(api.getAccountInfo(context, 'tz1empty')).resolves.toEqual({
       type: 'tezos',
       revealed: false,
     })
-    expect(networkApi.getAccountByAddress).toHaveBeenCalledWith('tz1empty')
+    expect(mockGetAccountByAddress).toHaveBeenCalledWith('tz1empty')
   })
 
   it('propagates errors from getAccountByAddress', async () => {
-    ;(networkApi.getAccountByAddress as jest.Mock).mockRejectedValueOnce(new Error('boom'))
+    mockGetAccountByAddress.mockRejectedValueOnce(new Error('boom'))
 
-    await expect(api.getAccountInfo('tz1boom')).rejects.toThrow('boom')
+    await expect(api.getAccountInfo(context, 'tz1boom')).rejects.toThrow('boom')
   })
 })