@ledgerhq/coin-tezos 10.2.2 → 10.2.6

package/src/logic/listOperations.ts

@@ -1,7 +1,7 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
-import type { TezosCoinConfig, TezosContext } from '../config'
+import type { TezosContext } from '../config'
 import type { AccountsGetOperationsOptions } from '../network/types'
 import type {
   APIDelegationType,
@@ -14,7 +14,7 @@ import type {
   TokenTransfersGetOptions,
 } from '../network/types'
 import type { Operation } from '@ledgerhq/coin-module-framework/api/types'
-import { log } from '@ledgerhq/logs'
+import type { Logger } from '@ledgerhq/coin-module-framework/config'
 import { STAKING_ACTION_TO_OP_TYPE } from '../constants'
 import { createTzktApi } from '../network'
 import {
@@ -69,6 +69,7 @@ function maxLevel<T extends { level: number }>(items: T[]): number | undefined {
 
 /** When the API returned a full page, the trailing block level may be incomplete — drop it. */
 function trimPartialLastLevel<T extends { level: number; id: number }>(
+  logger: Logger,
   items: T[],
   full: boolean
 ): { trimmed: T[]; intraLevelLastId?: number } {
@@ -78,7 +79,7 @@ function trimPartialLastLevel<T extends { level: number; id: number }>(
   const lastLevel = last.level
   const filtered = items.filter((op) => op.level !== lastLevel)
   if (filtered.length === 0) {
-    log('coin:tezos', 'listOperations: full page single level — keeping all rows', {
+    logger('coin:tezos', 'listOperations: full page single level — keeping all rows', {
       lastLevel,
       count: items.length,
     })
@@ -251,6 +252,7 @@ function keepNativeOp(op: APIOperation, address: string): op is ConvertibleOpera
 }
 
 function convertNativeOps(
+  logger: Logger,
   ops: APIOperation[],
   address: string,
   stakingBlockHashes: Map<number, string>
@@ -282,7 +284,7 @@ function convertNativeOps(
   return ops
     .filter((op): op is ConvertibleOperation => keepNativeOp(op, address))
     .map((op) => {
-      const converted = convertOperation(address, op, stakingBlockHashes)
+      const converted = convertOperation(logger, address, op, stakingBlockHashes)
       const extraFee = internalFeesByHash.get(converted.tx.hash)
       if (extraFee) {
         // Add internal sub-tx fees to the parent operation, then clear to avoid
@@ -368,8 +370,8 @@ export async function listOperations(
   const nativePage = clampPage(nativeOpsRaw, limit)
   const tokenPage = clampPage(tokenTransfersRaw, limit)
 
-  const nativeTrimmed = trimPartialLastLevel(nativePage.sliced, nativePage.full)
-  const tokenTrimmed = trimPartialLastLevel(tokenPage.sliced, tokenPage.full)
+  const nativeTrimmed = trimPartialLastLevel(context.logger, nativePage.sliced, nativePage.full)
+  const tokenTrimmed = trimPartialLastLevel(context.logger, tokenPage.sliced, tokenPage.full)
 
   // when both streams are exhausted (neither hit the page limit),
   // we already have every op — alignment would silently discard ops
@@ -388,7 +390,7 @@ export async function listOperations(
       ? tokenTrimmed.trimmed
       : alignToBoundary(tokenTrimmed.trimmed, boundary, sort)
 
-  const stakingBlockHashes = await fetchMissingStakingBlockHashes(config, nativeAligned)
+  const stakingBlockHashes = await fetchMissingStakingBlockHashes(context, nativeAligned)
 
   const parentMap = buildParentMap(nativeAligned)
 
@@ -401,7 +403,12 @@ export async function listOperations(
     tokenTrimmed.intraLevelLastId
   )
 
-  const filteredNativeOps = convertNativeOps(nativeAligned, address, stakingBlockHashes)
+  const filteredNativeOps = convertNativeOps(
+    context.logger,
+    nativeAligned,
+    address,
+    stakingBlockHashes
+  )
   const tokenConverted = convertTokenOps(tokenAligned, parentMap, address)
 
   const sortedOperations = [...filteredNativeOps, ...tokenConverted].sort(
@@ -424,7 +431,7 @@ type ConvertibleOperation =
  * failures are swallowed: callers fall back to `""` for levels not in the map.
  */
 async function fetchMissingStakingBlockHashes(
-  config: TezosCoinConfig,
+  context: TezosContext,
   ops: readonly APIOperation[]
 ): Promise<Map<number, string>> {
   const missingLevels = new Set<number>()
@@ -432,10 +439,11 @@ async function fetchMissingStakingBlockHashes(
     if (isAPIStakingType(op) && !op.block) missingLevels.add(op.level)
   }
   if (missingLevels.size === 0) return new Map()
+  const config = await context.config()
   try {
     return await createTzktApi(config).getBlockHashesByLevels([...missingLevels])
   } catch (err) {
-    log('coin:tezos', 'fetchMissingStakingBlockHashes: skipped on fetch error', {
+    context.logger('coin:tezos', 'fetchMissingStakingBlockHashes: skipped on fetch error', {
       reason: String(err),
     })
     return new Map()
@@ -487,6 +495,7 @@ function resolveAmount(operation: ConvertibleOperation): bigint {
 }
 
 function resolveNormalizedType(
+  logger: Logger,
   operation: ConvertibleOperation,
   address: string,
   targetAddress: string | undefined,
@@ -499,7 +508,7 @@ function resolveNormalizedType(
   if (isAPIRevealType(operation)) return 'REVEAL'
   if (isAPIOriginationType(operation)) return operation.contractBalance > 0 ? 'OUT' : 'FEES'
   if (!isAPITransactionType(operation)) {
-    log('coin:tezos', '(logic/operations): Unknown operation type, defaulting to OUT')
+    logger('coin:tezos', '(logic/operations): Unknown operation type, defaulting to OUT')
     return 'OUT'
   }
   const isOut = operation.sender?.address === address
@@ -529,6 +538,7 @@ function getLedgerOpType(
 }
 
 function convertOperation(
+  logger: Logger,
   address: string,
   operation: ConvertibleOperation,
   stakingBlockHashes: Map<number, string>
@@ -544,7 +554,7 @@ function convertOperation(
   const targetAddress = resolveTargetAddress(operation)
   // reveal has no meaningful target; staking resolves senders/recipients separately.
   if (!targetAddress && !isAPIRevealType(operation) && !isAPIStakingType(operation)) {
-    log('coin:tezos', '(logic/operations): No target address found for operation', operation)
+    logger('coin:tezos', '(logic/operations): No target address found for operation', operation)
   }
 
   const { senders, recipients } = isAPIStakingType(operation)
@@ -561,7 +571,7 @@ function convertOperation(
     BigInt(operation.bakerFee ?? 0) +
     BigInt(operation.allocationFee ?? 0)
 
-  const normalizedType = resolveNormalizedType(operation, address, targetAddress, amount)
+  const normalizedType = resolveNormalizedType(logger, operation, address, targetAddress, amount)
 
   // Tezos uses "applied" for every success operation (something else=failed )
   const hasFailed = operation.status && operation.status !== 'applied'

package/src/logic/validateIntent.test.ts

@@ -15,9 +15,10 @@ import {
   MustDelegateBeforeStaking,
   NotEnoughBalanceToDelegate,
   TezosNotEnoughStaked,
+  TezosStakeAmountTooSmall,
   TezosStakeBlockedByPendingUnstake,
 } from '../types/errors'
-import { STAKE_USE_ALL_RESERVE_MUTEZ } from '../utils'
+import { MAPPED_TAQUITO_ERRORS, STAKE_USE_ALL_RESERVE_MUTEZ } from '../utils'
 import { validateIntent } from './validateIntent'
 
 const mockEstimateFees = jest.fn()
@@ -387,6 +388,35 @@ describe('validateIntent', () => {
       expect(result.amount).toBe(0n)
     })
 
+    it('should return TezosStakeAmountTooSmall when stake useAllAmount resolves max below the probe', async () => {
+      mockGetAccountByAddress.mockResolvedValue(
+        makeUserAccount({
+          delegate: { alias: 'baker', address: validRecipient, active: true },
+          delegationLevel: 1,
+        })
+      )
+      mockEstimateFees.mockResolvedValueOnce({
+        fees: 1000n,
+        gasLimit: 10000n,
+        storageLimit: 0n,
+        estimatedFees: 1000n,
+        amount: 999n,
+      })
+
+      const result = await validateIntent(context, {
+        intentType: 'staking',
+        asset: { type: 'native' },
+        type: 'stake',
+        sender: senderAddress,
+        recipient: '',
+        amount: 0n,
+        useAllAmount: true,
+      })
+
+      expect(result.errors.amount).toBeInstanceOf(TezosStakeAmountTooSmall)
+      expect(result.amount).toBe(999n)
+    })
+
     it('excludes already-staked funds when stake useAllAmount falls back to balance computation', async () => {
       mockGetAccountByAddress.mockResolvedValue(
         makeUserAccount({
@@ -987,6 +1017,65 @@ describe('validateIntent', () => {
       expect(result.errors.amount).toBeInstanceOf(MustDelegateBeforeStaking)
     })
 
+    it('maps stake_amount_too_small to TezosStakeAmountTooSmall', async () => {
+      mockGetAccountByAddress.mockResolvedValue(
+        makeUserAccount({
+          delegate: { alias: 'baker', address: validRecipient, active: true },
+          delegationLevel: 1,
+        })
+      )
+      mockEstimateFees.mockResolvedValue({
+        fees: 0n,
+        gasLimit: 0n,
+        storageLimit: 0n,
+        estimatedFees: 500n,
+        taquitoError: 'proto.025-PsUshuai.block_validation_plugin.stake_amount_too_small',
+      })
+
+      const result = await validateIntent(context, {
+        intentType: 'staking',
+        asset: { type: 'native' },
+        type: 'stake',
+        sender: senderAddress,
+        recipient: '',
+        amount: 1n,
+      })
+
+      expect(result.errors.amount).toBeInstanceOf(TezosStakeAmountTooSmall)
+    })
+
+    it.each(MAPPED_TAQUITO_ERRORS)(
+      'maps %s, which estimation reports instead of throwing, to a dedicated error',
+      async (id) => {
+        mockGetAccountByAddress.mockResolvedValue(
+          makeUserAccount({
+            delegate: { alias: 'baker', address: validRecipient, active: true },
+            delegationLevel: 1,
+          })
+        )
+        mockEstimateFees.mockResolvedValue({
+          fees: 0n,
+          gasLimit: 0n,
+          storageLimit: 0n,
+          estimatedFees: 500n,
+          taquitoError: `proto.025-PsUshuai.${id}`,
+        })
+
+        const result = await validateIntent(context, {
+          intentType: 'staking',
+          asset: { type: 'native' },
+          type: 'stake',
+          sender: senderAddress,
+          recipient: '',
+          amount: 1000n,
+        })
+
+        const error = result.errors.amount ?? result.errors.recipient
+        expect(error).toBeInstanceOf(Error)
+        expect(error?.name).not.toBe('Error')
+      }
+    )
+
     it('maps cannot_stake_with_unfinalizable_unstake_requests_to_another_delegate to TezosStakeBlockedByPendingUnstake for stake', async () => {
       mockGetAccountByAddress.mockResolvedValue(
         makeUserAccount({

package/src/logic/validateIntent.ts

@@ -24,6 +24,7 @@ import {
   MustDelegateBeforeStaking,
   NotEnoughBalanceToDelegate,
   TezosNotEnoughStaked,
+  TezosStakeAmountTooSmall,
   TezosStakeBlockedByPendingUnstake,
 } from '../types/errors'
 import {
@@ -31,6 +32,7 @@ import {
   parseTezosTokenAsset,
   partitionNativeBalance,
   resolveTezosOperationMode,
+  STAKE_ESTIMATION_PROBE_MUTEZ,
 } from '../utils'
 import { estimateFees } from './estimateFees'
 
@@ -156,6 +158,9 @@ function mapTaquitoErrors(taquitoError: string, intentType: string): Record<stri
     errors.amount = new TezosNotEnoughStaked()
   } else if (taquitoError.endsWith('contract.must_be_delegated_to_stake')) {
     errors.amount = new MustDelegateBeforeStaking()
+  } else if (taquitoError.endsWith('stake_amount_too_small')) {
+    // Below one staking pseudotoken the stake mints nothing; see STAKE_ESTIMATION_PROBE_MUTEZ.
+    errors.amount = new TezosStakeAmountTooSmall()
   } else if (
     taquitoError.endsWith('cannot_stake_with_unfinalizable_unstake_requests_to_another_delegate')
   ) {
@@ -166,10 +171,10 @@ function mapTaquitoErrors(taquitoError: string, intentType: string): Record<stri
     // Re-delegating (or staking) to the current baker leaves the delegate unchanged; the node
     // rejects it. Surfaces for both `delegate` and `stake` intents as "already delegated".
     errors.recipient = new InvalidAddressBecauseAlreadyDelegated()
-  } else if (taquitoError.includes('empty_implicit_contract')) {
+  } else if (taquitoError.endsWith('empty_implicit_contract')) {
     errors.amount =
       intentType === 'stake' ? new NotEnoughBalance() : new NotEnoughBalanceToDelegate()
-  } else if (taquitoError.includes('script_rejected')) {
+  } else if (taquitoError.endsWith('script_rejected')) {
     errors.amount = new NotEnoughBalance()
   } else {
     errors.amount = new Error(taquitoError)
@@ -447,6 +452,17 @@ export async function validateIntent(
       errors.amount = new NotEnoughBalance()
     }
 
+    // Estimation simulates the probe, never the resolved max; below the probe the stake may still
+    // mint no pseudotoken and be refused at injection.
+    if (
+      intent.type === 'stake' &&
+      intent.useAllAmount &&
+      amount < STAKE_ESTIMATION_PROBE_MUTEZ &&
+      !errors.amount
+    ) {
+      errors.amount = new TezosStakeAmountTooSmall()
+    }
+
     const { spendable } = partitionNativeBalance(
       BigInt(senderInfo.balance),
       BigInt(senderInfo.stakedBalance ?? 0),

package/src/network/bakers.integ.test.ts

@@ -1,16 +1,17 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
-import type { TezosCoinConfig } from '../config'
+import type { TezosCoinConfig, TezosContext } from '../config'
 import { listBakers } from '../network/bakers'
 import whitelist from '../network/bakers.whitelist-default'
 import { mockConfig } from '../test/config'
 
 describe('tezos bakers', () => {
   const config: TezosCoinConfig = mockConfig
+  const context: TezosContext = { config: async () => config, logger: jest.fn() }
 
   test('atleast 10 whitelisted bakers are online', async () => {
-    const onlineBakersCount = (await listBakers(config, whitelist))
+    const onlineBakersCount = (await listBakers(context, whitelist))
       .map((baker) => baker.address)
       .filter((address) => whitelist.includes(address)).length
     expect(onlineBakersCount).toBeGreaterThan(10)

package/src/network/bakers.test.ts

@@ -7,6 +7,7 @@ import { mockConfig } from '../test/config'
 import { asBaker, cache } from './bakers'
 
 const mockNetworkFn = jest.fn()
+const mockLogger = jest.fn()
 jest.mock('@ledgerhq/live-network/network', () => ({
   __esModule: true,
   default: (...args: unknown[]) => mockNetworkFn(...args),
@@ -183,7 +184,7 @@ describe('Tezos Baker', () => {
     })
 
     it('Ledger Baker should be first in the list', async () => {
-      const bakers = await cache(config)
+      const bakers = await cache(config, mockLogger)
       expect(bakers[0].name).toBe('Ledger by Kiln')
     })
   })

package/src/network/bakers.ts

@@ -1,11 +1,11 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
-import type { TezosCoinConfig } from '../config'
+import type { TezosCoinConfig, TezosContext } from '../config'
 import type { Baker } from '../types'
+import type { Logger } from '@ledgerhq/coin-module-framework/config'
 import { makeLRUCache } from '@ledgerhq/live-network/cache'
 import network from '@ledgerhq/live-network/network'
-import { log } from '@ledgerhq/logs'
 import { ledgerValidatorAddress } from './bakers.whitelist-default'
 
 export type TezosApiBaker = {
@@ -40,7 +40,7 @@ export type TezosBakerFeature = {
 }
 
 export const cache = makeLRUCache(
-  async (config: TezosCoinConfig): Promise<Baker[]> => {
+  async (config: TezosCoinConfig, logger: Logger): Promise<Baker[]> => {
     const bakers: Baker[] = []
     const TEZOS_API_BASE_URL = config.baker.url
     const { data } = await network<TezosApiBaker[]>({
@@ -48,7 +48,7 @@ export const cache = makeLRUCache(
     })
 
     if (data && Array.isArray(data)) {
-      log('tezos/bakers', 'found ' + data.length + ' bakers')
+      logger('tezos/bakers', 'found ' + data.length + ' bakers')
       data
         .filter((raw) => raw.status === 'active' && raw.delegation.enabled)
         .forEach((raw) => {
@@ -64,7 +64,7 @@ export const cache = makeLRUCache(
     const ledgerBakerIndex = bakers.findIndex((baker) => baker.address === ledgerValidatorAddress)
     const [ledgerBaker] = bakers.splice(ledgerBakerIndex, 1)
 
-    log('tezos/bakers', 'loaded ' + bakers.length + ' bakers')
+    logger('tezos/bakers', 'loaded ' + bakers.length + ' bakers')
     return [ledgerBaker, ...bakers]
   },
   (config: TezosCoinConfig) => config.baker.url
@@ -72,8 +72,9 @@ export const cache = makeLRUCache(
 
 let _lastBakers: Baker[] = []
 
-export const fetchAllBakers = async (config: TezosCoinConfig): Promise<Baker[]> => {
-  const allBakers = await cache.force(config)
+export const fetchAllBakers = async (context: TezosContext): Promise<Baker[]> => {
+  const config = await context.config()
+  const allBakers = await cache.force(config, context.logger)
   _lastBakers = allBakers
   return allBakers
 }
@@ -86,10 +87,11 @@ export const listBakersWithDefault = (whitelistAddresses: string[]): Baker[] =>
 }
 
 export const listBakers = async (
-  config: TezosCoinConfig,
+  context: TezosContext,
   whitelistAddresses: string[]
 ): Promise<Baker[]> => {
-  _lastBakers = await cache(config)
+  const config = await context.config()
+  _lastBakers = await cache(config, context.logger)
   return whitelist(_lastBakers, whitelistAddresses)
 }
 
@@ -97,12 +99,12 @@ export function getBakerSync(addr: string): Baker | undefined {
   return _lastBakers.find((baker) => baker.address === addr)
 }
 
-export async function loadBaker(config: TezosCoinConfig, addr: string): Promise<Baker | undefined> {
+export async function loadBaker(context: TezosContext, addr: string): Promise<Baker | undefined> {
   const cacheBaker = getBakerSync(addr)
   if (cacheBaker) return cacheBaker
-  const bakers = await cache(config)
-  const baker = bakers.find((baker) => baker.address === addr)
-  return baker
+  const config = await context.config()
+  const bakers = await cache(config, context.logger)
+  return bakers.find((baker) => baker.address === addr)
 }
 
 export const asBaker = (data: TezosApiBaker): Baker | undefined => {

package/src/network/tzkt.test.ts

@@ -1,7 +1,7 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
-import type { TezosCoinConfig } from '../config'
+import type { TezosCoinConfig, TezosContext } from '../config'
 // Import fetchBlockTransactions / fetchBlockTokenTransfers / fetchBlockDelegations through the index re-export
 // so that network/index.ts line 4 is exercised and its coverage is tracked.
 import type {
@@ -18,7 +18,6 @@ import type { DeepPartial, DeepPartialReturn } from '@ledgerhq/coin-module-frame
 // tzkt.ts imports `network` from "@ledgerhq/live-network" (the package root, not the /network
 // subpath). We mock the same path so Jest's module registry intercepts those calls.
 import network from '@ledgerhq/live-network'
-import { log } from '@ledgerhq/logs'
 import { mockConfig } from '../test/config'
 import { createTzktApi } from './tzkt'
 import {
@@ -30,9 +29,10 @@ import {
 } from '.'
 
 jest.mock('@ledgerhq/live-network')
-jest.mock('@ledgerhq/logs')
 const mockedNetwork = jest.mocked<DeepPartialReturn<typeof network>>(network)
-const mockedLog = jest.mocked(log)
+const mockLogger = jest.fn()
+const config: TezosCoinConfig = mockConfig
+const context: TezosContext = { config: async () => config, logger: mockLogger }
 
 // The tzkt module is NOT mocked here. `createTzktApi(config)` returns a plain object whose methods
 // close over `config.explorer.url` and delegate to the mocked `network`. Each method also calls its
@@ -43,7 +43,6 @@ const mockedLog = jest.mocked(log)
 // The paginated helpers (`fetchAllTransactions`, `fetchBlock*`) are top-level functions that build
 // their OWN fresh `createTzktApi(config)` instance internally, so spying on `api` cannot intercept
 // them. Those tests therefore drive the mocked `network` directly.
-const config: TezosCoinConfig = mockConfig
 const api = createTzktApi(config)
 
 // ---------------------------------------------------------------------------
@@ -808,7 +807,7 @@ describe('tzkt network API', () => {
     it('returns empty when the first page is empty', async () => {
       mockedNetwork.mockReturnValue(networkResponse([]))
 
-      const result = await fetchAllTransactions(config, 'tz1acc')
+      const result = await fetchAllTransactions(context, 'tz1acc')
 
       expect(result).toEqual([])
     })
@@ -819,7 +818,7 @@ describe('tzkt network API', () => {
         .mockReturnValueOnce(networkResponse([makeOperation(2)]))
         .mockReturnValueOnce(networkResponse([]))
 
-      const result = await fetchAllTransactions(config, 'tz1acc')
+      const result = await fetchAllTransactions(context, 'tz1acc')
 
       expect(result).toEqual([makeOperation(1), makeOperation(2)])
       expect(mockedNetwork).toHaveBeenCalledTimes(3)
@@ -840,10 +839,10 @@ describe('tzkt network API', () => {
       const opMissingId: DeepPartial<APIOperation> = { ...makeOperation(1), id: 0 }
       mockedNetwork.mockReturnValue(networkResponse([opMissingId]))
 
-      const result = await fetchAllTransactions(config, 'tz1acc')
+      const result = await fetchAllTransactions(context, 'tz1acc')
 
       expect(result).toEqual([opMissingId])
-      expect(mockedLog).toHaveBeenCalledWith('tezos', 'id missing!')
+      expect(mockLogger).toHaveBeenCalledWith('tezos', 'id missing!')
     })
 
     it('stops after maxTxQuery iterations', async () => {
@@ -852,7 +851,7 @@ describe('tzkt network API', () => {
       let nextId = 0
       mockedNetwork.mockImplementation(() => networkResponse([makeOperation(++nextId)]))
 
-      const result = await fetchAllTransactions(config, 'tz1acc')
+      const result = await fetchAllTransactions(context, 'tz1acc')
 
       expect(mockedNetwork).toHaveBeenCalledTimes(mockConfig.explorer.maxTxQuery)
       expect(result).toHaveLength(mockConfig.explorer.maxTxQuery)
@@ -992,7 +991,7 @@ describe('tzkt network API', () => {
     it('returns an empty array when the first page is empty', async () => {
       mockedNetwork.mockReturnValue(networkResponse([]))
 
-      const result = await fetchBlockTransactions(config, 100)
+      const result = await fetchBlockTransactions(context, 100)
 
       expect(result).toEqual([])
     })
@@ -1001,7 +1000,7 @@ describe('tzkt network API', () => {
       const txs = [makeTxItem(1), makeTxItem(2)]
       mockedNetwork.mockReturnValue(networkResponse(txs))
 
-      const result = await fetchBlockTransactions(config, 100)
+      const result = await fetchBlockTransactions(context, 100)
 
       expect(result).toEqual(txs)
     })
@@ -1014,7 +1013,7 @@ describe('tzkt network API', () => {
         .mockReturnValueOnce(networkResponse(fullPage))
         .mockReturnValueOnce(networkResponse(lastPage))
 
-      const result = await fetchBlockTransactions(config, 100)
+      const result = await fetchBlockTransactions(context, 100)
 
       expect(result).toHaveLength(1002)
       expect(mockedNetwork).toHaveBeenCalledTimes(2)
@@ -1031,11 +1030,11 @@ describe('tzkt network API', () => {
       const fullPage = Array.from({ length: 1000 }, (_, i) => makeTxItem(i + 1))
       mockedNetwork.mockReturnValue(networkResponse(fullPage))
 
-      const result = await fetchBlockTransactions(config, 100)
+      const result = await fetchBlockTransactions(context, 100)
 
       expect(mockedNetwork).toHaveBeenCalledTimes(100)
       expect(result).toHaveLength(100_000)
-      expect(mockedLog).toHaveBeenCalledWith(
+      expect(mockLogger).toHaveBeenCalledWith(
         'tezos',
         expect.stringContaining('fetchBlockTransactions: maxTxQuery limit reached at level 100')
       )
@@ -1051,7 +1050,7 @@ describe('tzkt network API', () => {
     it('returns an empty array when the first page is empty', async () => {
       mockedNetwork.mockReturnValue(networkResponse([]))
 
-      const result = await fetchBlockTokenTransfers(config, 100)
+      const result = await fetchBlockTokenTransfers(context, 100)
 
       expect(result).toEqual([])
     })
@@ -1060,7 +1059,7 @@ describe('tzkt network API', () => {
       const items = [makeTokenItem(1), makeTokenItem(2)]
       mockedNetwork.mockReturnValue(networkResponse(items))
 
-      const result = await fetchBlockTokenTransfers(config, 100)
+      const result = await fetchBlockTokenTransfers(context, 100)
 
       expect(result).toEqual(items)
     })
@@ -1073,7 +1072,7 @@ describe('tzkt network API', () => {
         .mockReturnValueOnce(networkResponse(fullPage))
         .mockReturnValueOnce(networkResponse(lastPage))
 
-      const result = await fetchBlockTokenTransfers(config, 100)
+      const result = await fetchBlockTokenTransfers(context, 100)
 
       expect(result).toHaveLength(1001)
       expect(mockedNetwork).toHaveBeenCalledTimes(2)
@@ -1088,11 +1087,11 @@ describe('tzkt network API', () => {
       const fullPage = Array.from({ length: 1000 }, (_, i) => makeTokenItem(i + 1))
       mockedNetwork.mockReturnValue(networkResponse(fullPage))
 
-      const result = await fetchBlockTokenTransfers(config, 100)
+      const result = await fetchBlockTokenTransfers(context, 100)
 
       expect(mockedNetwork).toHaveBeenCalledTimes(100)
       expect(result).toHaveLength(100_000)
-      expect(mockedLog).toHaveBeenCalledWith(
+      expect(mockLogger).toHaveBeenCalledWith(
         'tezos',
         expect.stringContaining('fetchBlockTokenTransfers: maxTxQuery limit reached at level 100')
       )
@@ -1147,7 +1146,7 @@ describe('tzkt network API', () => {
     it('returns an empty array when the first page is empty', async () => {
       mockedNetwork.mockReturnValue(networkResponse([]))
 
-      const result = await fetchBlockDelegations(config, 100)
+      const result = await fetchBlockDelegations(context, 100)
 
       expect(result).toEqual([])
     })
@@ -1156,7 +1155,7 @@ describe('tzkt network API', () => {
       const items = [makeDelegationItem(1), makeDelegationItem(2)]
       mockedNetwork.mockReturnValue(networkResponse(items))
 
-      const result = await fetchBlockDelegations(config, 100)
+      const result = await fetchBlockDelegations(context, 100)
 
       expect(result).toEqual(items)
     })
@@ -1169,7 +1168,7 @@ describe('tzkt network API', () => {
         .mockReturnValueOnce(networkResponse(fullPage))
         .mockReturnValueOnce(networkResponse(lastPage))
 
-      const result = await fetchBlockDelegations(config, 100)
+      const result = await fetchBlockDelegations(context, 100)
 
       expect(result).toHaveLength(1001)
       expect(mockedNetwork).toHaveBeenCalledTimes(2)
@@ -1184,11 +1183,11 @@ describe('tzkt network API', () => {
       const fullPage = Array.from({ length: 1000 }, (_, i) => makeDelegationItem(i + 1))
       mockedNetwork.mockReturnValue(networkResponse(fullPage))
 
-      const result = await fetchBlockDelegations(config, 100)
+      const result = await fetchBlockDelegations(context, 100)
 
       expect(mockedNetwork).toHaveBeenCalledTimes(100)
       expect(result).toHaveLength(100_000)
-      expect(mockedLog).toHaveBeenCalledWith(
+      expect(mockLogger).toHaveBeenCalledWith(
         'tezos',
         expect.stringContaining('fetchBlockDelegations: maxTxQuery limit reached at level 100')
       )
@@ -1243,7 +1242,7 @@ describe('tzkt network API', () => {
     it('returns an empty array when the first page is empty', async () => {
       mockedNetwork.mockReturnValue(networkResponse([]))
 
-      const result = await fetchBlockStaking(config, 100)
+      const result = await fetchBlockStaking(context, 100)
 
       expect(result).toEqual([])
     })
@@ -1252,7 +1251,7 @@ describe('tzkt network API', () => {
       const items = [makeStakingItem(1), makeStakingItem(2)]
       mockedNetwork.mockReturnValue(networkResponse(items))
 
-      const result = await fetchBlockStaking(config, 100)
+      const result = await fetchBlockStaking(context, 100)
 
       expect(result).toEqual(items)
     })
@@ -1265,7 +1264,7 @@ describe('tzkt network API', () => {
         .mockReturnValueOnce(networkResponse(fullPage))
         .mockReturnValueOnce(networkResponse(lastPage))
 
-      const result = await fetchBlockStaking(config, 100)
+      const result = await fetchBlockStaking(context, 100)
 
       expect(result).toHaveLength(1001)
       expect(mockedNetwork).toHaveBeenCalledTimes(2)
@@ -1280,11 +1279,11 @@ describe('tzkt network API', () => {
       const fullPage = Array.from({ length: 1000 }, (_, i) => makeStakingItem(i + 1))
       mockedNetwork.mockReturnValue(networkResponse(fullPage))
 
-      const result = await fetchBlockStaking(config, 100)
+      const result = await fetchBlockStaking(context, 100)
 
       expect(mockedNetwork).toHaveBeenCalledTimes(100)
       expect(result).toHaveLength(100_000)
-      expect(mockedLog).toHaveBeenCalledWith(
+      expect(mockLogger).toHaveBeenCalledWith(
         'tezos',
         expect.stringContaining('fetchBlockStaking: maxTxQuery limit reached at level 100')
       )