@ledgerhq/coin-tezos 7.12.0 → 9.0.0

package/src/logic/estimateFees.test.ts

@@ -1,17 +1,13 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
+import type { TezosCoinConfig, TezosContext } from '../config'
 import { getRevealFee } from '@taquito/taquito'
-import coinConfig from '../config'
 import { normalizePublicKeyForAddress, STAKE_USE_ALL_RESERVE_MUTEZ } from '../utils'
 import { estimateFees } from './estimateFees'
 import { getTezosToolkit } from './tezosToolkit'
 
 jest.mock('./tezosToolkit')
-jest.mock('../config', () => ({
-  __esModule: true,
-  default: { getCoinConfig: jest.fn() },
-}))
 
 const defaultFeesConfig = {
   minGasLimit: 600,
@@ -58,12 +54,19 @@ describe('estimateFees', () => {
     setProvider: jest.fn(),
   }
 
+  let currentConfig: TezosCoinConfig
+  const context: TezosContext = {
+    config: async () => currentConfig,
+    logger: () => undefined,
+  }
+  const mkConfig = (cfg: unknown): TezosCoinConfig => cfg as unknown as TezosCoinConfig
+
   beforeEach(() => {
     jest.clearAllMocks()
     ;(getTezosToolkit as jest.Mock).mockReturnValue(mockTezosToolkit)
-    ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+    currentConfig = {
       fees: defaultFeesConfig,
-    })
+    } as unknown as TezosCoinConfig
   })
 
   describe('signer public key from account.xpub', () => {
@@ -77,7 +80,7 @@ describe('estimateFees', () => {
         opSize: 100,
       })
 
-      await estimateFees({
+      await estimateFees(context, {
         account: { ...unrevealedAccount },
         transaction: {
           mode: 'delegate',
@@ -109,11 +112,11 @@ describe('estimateFees', () => {
         burnFeeMutez: 0,
         opSize: 100,
       })
-      ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+      currentConfig = mkConfig({
         fees: { ...defaultFeesConfig, minFees },
       })
 
-      const result = await estimateFees({
+      const result = await estimateFees(context, {
         account: revealedAccount,
         transaction: {
           mode: 'delegate',
@@ -135,11 +138,11 @@ describe('estimateFees', () => {
         burnFeeMutez: 0,
         opSize: 100,
       })
-      ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+      currentConfig = mkConfig({
         fees: { ...defaultFeesConfig, minFees },
       })
 
-      const result = await estimateFees({
+      const result = await estimateFees(context, {
         account: revealedAccount,
         transaction: {
           mode: 'delegate',
@@ -161,7 +164,7 @@ describe('estimateFees', () => {
         burnFeeMutez: 0,
         opSize: 100,
       })
-      ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+      currentConfig = mkConfig({
         fees: { ...defaultFeesConfig, minFees },
       })
 
@@ -172,7 +175,7 @@ describe('estimateFees', () => {
         amount: BigInt(0),
       }
 
-      const result = await estimateFees({ account, transaction })
+      const result = await estimateFees(context, { account, transaction })
 
       expect(result.fees).toBe(BigInt(minFees))
       expect(result.estimatedFees).toBe(BigInt(minFees * 2))
@@ -188,7 +191,7 @@ describe('estimateFees', () => {
       burnFeeMutez: 0,
       opSize: 100,
     })
-    ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+    currentConfig = mkConfig({
       fees: { ...defaultFeesConfig, minFees: 0, minEstimatedFees: 0 },
     })
 
@@ -199,7 +202,7 @@ describe('estimateFees', () => {
       amount: BigInt(0),
     }
 
-    const result = await estimateFees({ account, transaction })
+    const result = await estimateFees(context, { account, transaction })
 
     const expectedRevealFee = getRevealFee(account.address)
     expect(result.estimatedFees).toBe(BigInt(delegationFee + expectedRevealFee))
@@ -215,7 +218,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: revealedAccount,
       transaction: {
         mode: 'undelegate',
@@ -240,7 +243,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: revealedAccount,
       transaction: {
         mode: 'stake',
@@ -269,7 +272,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: revealedAccount,
       transaction: {
         mode: 'unstake',
@@ -298,7 +301,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: { ...revealedAccount, stakedBalance: 5000n },
       transaction: {
         mode: 'unstake',
@@ -322,7 +325,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: { ...revealedAccount, balance },
       transaction: {
         mode: 'stake',
@@ -351,7 +354,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: { ...unrevealedAccount, balance },
       transaction: {
         mode: 'stake',
@@ -377,7 +380,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: { ...revealedAccount, balance },
       transaction: {
         mode: 'stake',
@@ -402,7 +405,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: { ...revealedAccount, balance, stakedBalance },
       transaction: {
         mode: 'stake',
@@ -430,7 +433,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: { ...revealedAccount, balance, stakedBalance, unstakedBalance },
       transaction: {
         mode: 'stake',
@@ -454,7 +457,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: revealedAccount,
       transaction: {
         mode: 'unstake',
@@ -481,7 +484,7 @@ describe('estimateFees', () => {
       opSize: 100,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: revealedAccount,
       transaction: {
         mode: 'finalize_unstake',
@@ -508,11 +511,11 @@ describe('estimateFees', () => {
       burnFeeMutez: 0,
       opSize: 200,
     })
-    ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+    currentConfig = mkConfig({
       fees: { ...defaultFeesConfig, minFees },
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: revealedAccount,
       transaction: {
         mode: 'send',
@@ -538,7 +541,7 @@ describe('estimateFees', () => {
       opSize: 200,
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: unrevealedAccount,
       transaction: {
         mode: 'send',
@@ -553,7 +556,7 @@ describe('estimateFees', () => {
   })
 
   it('returns zero estimation when balance is 0 and not useAllAmount', async () => {
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: { ...revealedAccount, balance: 0n },
       transaction: {
         mode: 'delegate',
@@ -568,7 +571,7 @@ describe('estimateFees', () => {
   })
 
   it('returns estimation with amount 0 when balance is 0 and useAllAmount', async () => {
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: { ...revealedAccount, balance: 0n },
       transaction: {
         mode: 'send',
@@ -583,7 +586,7 @@ describe('estimateFees', () => {
   })
 
   it('returns fallback estimation when send has no recipient', async () => {
-    ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+    currentConfig = mkConfig({
       fees: {
         ...defaultFeesConfig,
         minEstimatedFees: 300,
@@ -592,7 +595,7 @@ describe('estimateFees', () => {
       },
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: revealedAccount,
       transaction: {
         mode: 'send',
@@ -610,7 +613,7 @@ describe('estimateFees', () => {
 
   it('uses fallback estimation and adds reveal fee when Taquito throws Public key not found (unrevealed)', async () => {
     const minEstimatedFees = 400
-    ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+    currentConfig = mkConfig({
       fees: {
         ...defaultFeesConfig,
         minEstimatedFees,
@@ -622,7 +625,7 @@ describe('estimateFees', () => {
       Object.assign(new Error('Fail'), { status: 404, message: 'Public key not found' })
     )
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: unrevealedAccount,
       transaction: {
         mode: 'send',
@@ -648,7 +651,7 @@ describe('estimateFees', () => {
     })
 
     const contractAddress = 'KT1CpeSQKdkhWi4pinYcseCFKmDhs5M74BkU'
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: revealedAccount,
       transaction: {
         mode: 'send_token',
@@ -685,7 +688,7 @@ describe('estimateFees', () => {
     })
 
     const contractAddress = 'KT1CpeSQKdkhWi4pinYcseCFKmDhs5M74BkU'
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: unrevealedAccount,
       transaction: {
         mode: 'send_token',
@@ -703,7 +706,7 @@ describe('estimateFees', () => {
 
   it('send_token uses fallback when Taquito throws Public key not found', async () => {
     const minEstimatedFees = 400
-    ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+    currentConfig = mkConfig({
       fees: {
         ...defaultFeesConfig,
         minEstimatedFees,
@@ -718,7 +721,7 @@ describe('estimateFees', () => {
     )
 
     const contractAddress = 'KT1CpeSQKdkhWi4pinYcseCFKmDhs5M74BkU'
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: unrevealedAccount,
       transaction: {
         mode: 'send_token',
@@ -739,7 +742,7 @@ describe('estimateFees', () => {
     const { transfer } = setupFa2ContractMock()
     mockTezosToolkit.contract.at.mockResolvedValue({ methods: { transfer } })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: revealedAccount,
       transaction: {
         mode: 'send_token',
@@ -759,7 +762,7 @@ describe('estimateFees', () => {
     mockTezosToolkit.contract.at.mockRejectedValue(new Error('Invalid contract'))
 
     await expect(
-      estimateFees({
+      estimateFees(context, {
         account: revealedAccount,
         transaction: {
           mode: 'send_token',
@@ -774,7 +777,7 @@ describe('estimateFees', () => {
 
   it('send_token throws when contractAddress or tokenId is missing', async () => {
     await expect(
-      estimateFees({
+      estimateFees(context, {
         account: revealedAccount,
         transaction: {
           mode: 'send_token',
@@ -796,11 +799,11 @@ describe('estimateFees', () => {
       burnFeeMutez: 0,
       opSize: 200,
     })
-    ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+    currentConfig = mkConfig({
       fees: { ...defaultFeesConfig, minFees },
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: { ...revealedAccount, balance: BigInt(balance) },
       transaction: {
         mode: 'send',
@@ -836,11 +839,11 @@ describe('estimateFees', () => {
         burnFeeMutez,
         opSize: 200,
       })
-      ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+      currentConfig = mkConfig({
         fees: { ...defaultFeesConfig, minFees },
       })
 
-      const result = await estimateFees({
+      const result = await estimateFees(context, {
         account: { ...revealedAccount, balance },
         transaction: {
           mode: 'send',
@@ -867,11 +870,11 @@ describe('estimateFees', () => {
       burnFeeMutez: 0,
       opSize: 200,
     })
-    ;(coinConfig.getCoinConfig as jest.Mock).mockReturnValue({
+    currentConfig = mkConfig({
       fees: { ...defaultFeesConfig, minFees: 0 },
     })
 
-    const result = await estimateFees({
+    const result = await estimateFees(context, {
       account: { ...revealedAccount, balance, stakedBalance, unstakedBalance },
       transaction: {
         mode: 'send',

package/src/logic/estimateFees.ts

@@ -1,12 +1,12 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
+import type { TezosCoinConfig, TezosContext } from '../config'
 import type { TezosOperationMode } from '../types/model'
 import type { Estimate } from '@taquito/taquito'
 import { log } from '@ledgerhq/logs'
 import { COST_PER_BYTE, getRevealFee, ORIGINATION_SIZE } from '@taquito/taquito'
 import { validatePublicKey, ValidationResult } from '@taquito/utils'
-import coinConfig from '../config'
 import { UnsupportedTransactionMode } from '../types/errors'
 import {
   computeMaxStakeAmount,
@@ -54,20 +54,24 @@ export type EstimatedFees = {
  * @param {Account} account
  * @param {Transaction} transaction
  */
-export async function estimateFees({
-  account,
-  transaction,
-}: {
-  account: CoreAccountInfo
-  transaction: CoreTransactionInfo
-}): Promise<EstimatedFees> {
+export async function estimateFees(
+  context: TezosContext,
+  {
+    account,
+    transaction,
+  }: {
+    account: CoreAccountInfo
+    transaction: CoreTransactionInfo
+  }
+): Promise<EstimatedFees> {
+  const config = await context.config()
   // Normalize public key (hex -> base58) when provided (may be undefined for unrevealed accounts)
   // before the device is connected
   const encodedPubKey = account.xpub
     ? normalizePublicKeyForAddress(account.xpub, account.address)
     : undefined
 
-  const tezosToolkit = getTezosToolkit()
+  const tezosToolkit = getTezosToolkit(config)
   if (encodedPubKey && validatePublicKey(encodedPubKey) === ValidationResult.VALID) {
     tezosToolkit.setProvider({ signer: createMockSigner(account.address, encodedPubKey) })
   } else {
@@ -112,7 +116,7 @@ export async function estimateFees({
     ) {
       return {
         ...estimation,
-        ...createFallbackEstimation(),
+        ...createFallbackEstimation(config),
       }
     }
     let estimate: Estimate
@@ -181,9 +185,11 @@ export async function estimateFees({
         throw new UnsupportedTransactionMode('unsupported mode', { mode: transaction.mode })
     }
 
-    const minFees = coinConfig.getCoinConfig().fees.minFees ?? 0
+    const minFees = config.fees.minFees ?? 0
     const mainOpFee = Math.max(minFees, estimate.suggestedFeeMutez)
-    const revealFee = account.revealed ? 0n : BigInt(getRevealFeeForEstimation(account.address))
+    const revealFee = account.revealed
+      ? 0n
+      : BigInt(getRevealFeeForEstimation(config, account.address))
 
     // NOTE: send-max only applies to native XTZ transfer, not FA2
     if (transaction.useAllAmount && transaction.mode === 'send') {
@@ -248,14 +254,14 @@ export async function estimateFees({
             transaction: transaction,
           }
         )
-        const fallback = createFallbackEstimation()
+        const fallback = createFallbackEstimation(config)
         estimation.fees = fallback.fees
         estimation.gasLimit = fallback.gasLimit
         estimation.storageLimit = fallback.storageLimit
         estimation.estimatedFees = fallback.fees
         if (!account.revealed) {
           estimation.estimatedFees =
-            estimation.estimatedFees + BigInt(getRevealFeeForEstimation(account.address))
+            estimation.estimatedFees + BigInt(getRevealFeeForEstimation(config, account.address))
         }
         // Handle useAllAmount also for send mode when estimation falls back
         if (transaction.useAllAmount && transaction.mode === 'send') {
@@ -269,7 +275,8 @@ export async function estimateFees({
             estimation.estimatedFees = BigInt(suggestedFee)
             if (!account.revealed) {
               estimation.estimatedFees =
-                estimation.estimatedFees + BigInt(getRevealFeeForEstimation(account.address))
+                estimation.estimatedFees +
+                BigInt(getRevealFeeForEstimation(config, account.address))
             }
           }
 
@@ -277,7 +284,9 @@ export async function estimateFees({
           const totalFees =
             suggestedFee + (burnFeeMutez > 0 ? burnFeeMutez - 20 * COST_PER_BYTE : 0)
 
-          const revealFee = account.revealed ? 0 : getRevealFeeForEstimation(account.address)
+          const revealFee = account.revealed
+            ? 0
+            : getRevealFeeForEstimation(config, account.address)
           const maxAmount = spendableForMax - (totalFees + revealFee)
 
           const MINIMAL_FEE_PER_GAS_MUTEZ = 0.1
@@ -297,14 +306,14 @@ export async function estimateFees({
     } else {
       const msg = String((e as any).message || '')
       if (msg.includes('No signer has been configured')) {
-        const fallback = createFallbackEstimation()
+        const fallback = createFallbackEstimation(config)
         estimation.fees = fallback.fees
         estimation.gasLimit = fallback.gasLimit
         estimation.storageLimit = fallback.storageLimit
         estimation.estimatedFees = fallback.estimatedFees
         if (!account.revealed) {
           estimation.estimatedFees =
-            estimation.estimatedFees + BigInt(getRevealFeeForEstimation(account.address))
+            estimation.estimatedFees + BigInt(getRevealFeeForEstimation(config, account.address))
         }
         if (transaction.useAllAmount && transaction.mode === 'send') {
           const suggestedFee =
@@ -315,14 +324,17 @@ export async function estimateFees({
             estimation.estimatedFees = BigInt(suggestedFee)
             if (!account.revealed) {
               estimation.estimatedFees =
-                estimation.estimatedFees + BigInt(getRevealFeeForEstimation(account.address))
+                estimation.estimatedFees +
+                BigInt(getRevealFeeForEstimation(config, account.address))
             }
           }
 
           const burnFeeMutez = Number(estimation.storageLimit) * COST_PER_BYTE
           const totalFees =
             suggestedFee + (burnFeeMutez > 0 ? burnFeeMutez - 20 * COST_PER_BYTE : 0)
-          const revealFee = account.revealed ? 0 : getRevealFeeForEstimation(account.address)
+          const revealFee = account.revealed
+            ? 0
+            : getRevealFeeForEstimation(config, account.address)
           const maxAmount = spendableForMax - (totalFees + revealFee)
           const MINIMAL_FEE_PER_GAS_MUTEZ = 0.1
           const incr = OP_SIZE_XTZ_TRANSFER + DUST_MARGIN_MUTEZ * MINIMAL_FEE_PER_GAS_MUTEZ
@@ -340,7 +352,7 @@ export async function estimateFees({
   return estimation
 }
 
-function getRevealFeeForEstimation(address: string): number {
-  const minFees = coinConfig.getCoinConfig().fees.minFees ?? 0
+function getRevealFeeForEstimation(config: TezosCoinConfig, address: string): number {
+  const minFees = config.fees.minFees ?? 0
   return Math.max(minFees, getRevealFee(address))
 }

package/src/logic/getBalance.test.ts

@@ -1,11 +1,10 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
-import type { TezosCoinConfig } from '../config'
+import type { TezosCoinConfig, TezosContext } from '../config'
 import type { APITokenBalance } from '../network/types'
 import { HttpResponse, http } from 'msw'
 import { setupServer } from 'msw/node'
-import coinConfig from '../config'
 import { getBalance } from './getBalance'
 
 const mockTokensBalancesWithAllInfo: APITokenBalance[] = [
@@ -62,13 +61,14 @@ describe('getBalance', () => {
     mockServer.close()
   })
 
-  coinConfig.setCoinConfig(
-    () =>
+  const context: TezosContext = {
+    config: async () =>
       ({
         status: { type: 'active' },
         explorer: { url: 'http://tezos.explorer.com' },
-      }) as unknown as TezosCoinConfig
-  )
+      }) as unknown as TezosCoinConfig,
+    logger: () => undefined,
+  }
   it('gets the balance of a Tezos account', async () => {
     mockServer.use(
       http.get('http://tezos.explorer.com/v1/accounts/tz1WvvbEGpBXGeTVbLiR6DYBe1izmgiYuZbq', () =>
@@ -100,13 +100,13 @@ describe('getBalance', () => {
       })
     )
 
-    expect(await getBalance('tz1WvvbEGpBXGeTVbLiR6DYBe1izmgiYuZbq')).toEqual([
+    expect(await getBalance(context, 'tz1WvvbEGpBXGeTVbLiR6DYBe1izmgiYuZbq')).toEqual([
       {
         value: 0n,
         asset: { type: 'native' },
       },
     ])
-    expect(await getBalance('tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oSM8')).toEqual([
+    expect(await getBalance(context, 'tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oSM8')).toEqual([
       {
         value: BigInt(25),
         asset: { type: 'native' },
@@ -122,7 +122,7 @@ describe('getBalance', () => {
         },
       },
     ])
-    expect(await getBalance('tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oMMM')).toEqual([
+    expect(await getBalance(context, 'tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oMMM')).toEqual([
       {
         value: 15n,
         asset: { type: 'native' },
@@ -178,7 +178,7 @@ describe('getBalance', () => {
     it('attaches a delegation Stake when only delegate is set', async () => {
       mockAccount({ balance: 100, delegate: { address: delegateAddress } })
 
-      expect(await getBalance(address)).toEqual([
+      expect(await getBalance(context, address)).toEqual([
         { value: 100n, asset: { type: 'native' } },
         {
           value: 100n,
@@ -203,7 +203,7 @@ describe('getBalance', () => {
     it('attaches a stake Stake when stakedBalance > 0 (no delegate)', async () => {
       mockAccount({ balance: 100, stakedBalance: 30 })
 
-      expect(await getBalance(address)).toEqual([
+      expect(await getBalance(context, address)).toEqual([
         { value: 100n, asset: { type: 'native' }, locked: 30n },
         {
           value: 30n,
@@ -231,7 +231,7 @@ describe('getBalance', () => {
         delegate: { address: delegateAddress },
       })
 
-      expect(await getBalance(address)).toEqual([
+      expect(await getBalance(context, address)).toEqual([
         { value: 100n, asset: { type: 'native' }, locked: 30n },
         {
           value: 70n,
@@ -297,7 +297,7 @@ describe('getBalance', () => {
         http.get('http://tezos.explorer.com/v1/tokens/balances', () => HttpResponse.json([]))
       )
 
-      const result = await getBalance(address)
+      const result = await getBalance(context, address)
 
       expect(result).toHaveLength(4)
       expect(result[0]).toEqual({ value: 100n, asset: { type: 'native' }, locked: 40n })
@@ -352,7 +352,7 @@ describe('getBalance', () => {
         http.get('http://tezos.explorer.com/v1/tokens/balances', () => HttpResponse.json([]))
       )
 
-      const result = await getBalance(address)
+      const result = await getBalance(context, address)
 
       expect(result[0]).toEqual({ value: 100n, asset: { type: 'native' }, locked: 50n })
 
@@ -421,7 +421,7 @@ describe('getBalance', () => {
         )
       )
 
-      expect(await getBalance(address)).toEqual([
+      expect(await getBalance(context, address)).toEqual([
         { value: 100n, asset: { type: 'native' }, locked: 40n },
         {
           value: 30n,
@@ -455,7 +455,9 @@ describe('getBalance', () => {
     it('returns only the primary native Balance when no staking activity', async () => {
       mockAccount({ balance: 50 })
 
-      expect(await getBalance(address)).toEqual([{ value: 50n, asset: { type: 'native' } }])
+      expect(await getBalance(context, address)).toEqual([
+        { value: 50n, asset: { type: 'native' } },
+      ])
     })
 
     it('still excludes unstaked-frozen funds from spendable when the unstake_requests endpoint fails', async () => {
@@ -474,7 +476,7 @@ describe('getBalance', () => {
         http.get('http://tezos.explorer.com/v1/tokens/balances', () => HttpResponse.json([]))
       )
 
-      const result = await getBalance(address)
+      const result = await getBalance(context, address)
 
       // Breakdown rows are lost, but locked/delegated still derive from the account-level unstakedBalance.
       expect(result).toHaveLength(2)