@ledgerhq/coin-tezos 7.12.0 → 8.0.0

package/src/logic/getBlock.test.ts

@@ -1,6 +1,7 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
+import type { TezosCoinConfig, TezosContext } from '../config'
 import type {
   APIDelegationType,
   APIOriginationType,
@@ -12,6 +13,11 @@ import type {
 import type { OtherBlockOperation } from '@ledgerhq/coin-module-framework/api/types'
 import { getBlock } from './getBlock'
 
+const context: TezosContext = {
+  config: async () => ({}) as unknown as TezosCoinConfig,
+  logger: () => undefined,
+}
+
 // ---------------------------------------------------------------------------
 // Network mock
 // Using the same lazy-reference pattern as the XRP reference tests so that the
@@ -28,10 +34,10 @@ const mockFetchBlockOriginations = jest.fn()
 const mockFetchBlockReveals = jest.fn()
 
 jest.mock('../network', () => ({
-  tzkt: {
+  createTzktApi: () => ({
     getBlockByLevel: (...args: unknown[]) => mockGetBlockByLevel(...args),
     getOperationsOrigination: (...args: unknown[]) => mockGetOperationsOrigination(...args),
-  },
+  }),
   fetchBlockTransactions: (...args: unknown[]) => mockFetchBlockTransactions(...args),
   fetchBlockTokenTransfers: (...args: unknown[]) => mockFetchBlockTokenTransfers(...args),
   fetchBlockDelegations: (...args: unknown[]) => mockFetchBlockDelegations(...args),
@@ -204,7 +210,7 @@ describe('block info and guard', () => {
     )
 
     // When
-    const result = await getBlock(level)
+    const result = await getBlock(context, level)
 
     // Then
     expect(mockGetBlockByLevel).toHaveBeenCalledWith(level)
@@ -219,7 +225,9 @@ describe('block info and guard', () => {
 
   it('throws for height = 0 without any network calls', async () => {
     // When / Then
-    await expect(getBlock(0)).rejects.toThrow('getBlock: height must be a positive integer, got 0')
+    await expect(getBlock(context, 0)).rejects.toThrow(
+      'getBlock: height must be a positive integer, got 0'
+    )
     expect(mockGetBlockByLevel).not.toHaveBeenCalled()
     expect(mockFetchBlockTransactions).not.toHaveBeenCalled()
     expect(mockFetchBlockTokenTransfers).not.toHaveBeenCalled()
@@ -227,7 +235,7 @@ describe('block info and guard', () => {
 
   it('throws for negative height without any network calls', async () => {
     // When / Then
-    await expect(getBlock(-3)).rejects.toThrow(
+    await expect(getBlock(context, -3)).rejects.toThrow(
       'getBlock: height must be a positive integer, got -3'
     )
     expect(mockGetBlockByLevel).not.toHaveBeenCalled()
@@ -236,7 +244,9 @@ describe('block info and guard', () => {
   it.each([NaN, 1.5, Infinity, -Infinity])(
     'throws for non-safe-integer height (%s) without any network calls',
     async (height) => {
-      await expect(getBlock(height)).rejects.toThrow('getBlock: height must be a positive integer')
+      await expect(getBlock(context, height)).rejects.toThrow(
+        'getBlock: height must be a positive integer'
+      )
       expect(mockGetBlockByLevel).not.toHaveBeenCalled()
     }
   )
@@ -246,7 +256,7 @@ describe('block info and guard', () => {
     mockGetBlockByLevel.mockResolvedValue(makeBlock())
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions).toEqual([])
@@ -261,7 +271,7 @@ describe('block info and guard', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then – only the tx with a valid hash is included
     expect(result.transactions).toHaveLength(1)
@@ -276,7 +286,7 @@ describe('block info and guard', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then – the tx still appears (it has a hash), but no txIdToHash entry was created
     expect(result.transactions).toHaveLength(1)
@@ -297,7 +307,7 @@ describe('native XTZ operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
     const tx = result.transactions[0]
 
     // Then
@@ -324,7 +334,7 @@ describe('native XTZ operations', () => {
     mockFetchBlockTransactions.mockResolvedValue([makeTx({ amount: 0, bakerFee: 0 })])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then — no balance impact, no operations
     expect(result.transactions[0].operations).toEqual([])
@@ -336,7 +346,7 @@ describe('native XTZ operations', () => {
     mockFetchBlockTransactions.mockResolvedValue([makeTx({ amount: 0, bakerFee: 1500 })])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then — fees are a balance change, so operations must be emitted for fee attribution
     expect(result.transactions[0].fees).toBe(1500n)
@@ -366,7 +376,7 @@ describe('native XTZ operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then — peer must be absent (not set to undefined) when the counterparty is unknown
     expect(result.transactions[0].operations).toHaveLength(1)
@@ -387,7 +397,7 @@ describe('native XTZ operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then — peer must be absent (not set to undefined) when the counterparty is unknown
     expect(result.transactions[0].operations).toHaveLength(1)
@@ -409,7 +419,7 @@ describe('native XTZ operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then — only the tx with a hash produces a BlockTransaction
     expect(result.transactions).toHaveLength(1)
@@ -430,7 +440,7 @@ describe('fee computation and fee payer', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].fees).toBe(550n)
@@ -444,7 +454,7 @@ describe('fee computation and fee payer', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].feesPayer).toBe('tz1FeePayer')
@@ -462,7 +472,7 @@ describe('fee computation and fee payer', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].feesPayer).toBe('tz1RealUser')
@@ -474,7 +484,7 @@ describe('fee computation and fee payer', () => {
     mockFetchBlockTransactions.mockResolvedValue([makeTx({ sender: null, initiator: null })])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].feesPayer).toBeUndefined()
@@ -494,7 +504,7 @@ describe('fee computation and fee payer', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then – all three ?? 0 branches are exercised; total fees must be 0n
     expect(result.transactions[0].fees).toBe(0n)
@@ -512,7 +522,7 @@ describe('failed transactions', () => {
     mockFetchBlockTransactions.mockResolvedValue([makeTx({ status: 'failed', amount: 1_000_000 })])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].failed).toBe(true)
@@ -527,7 +537,7 @@ describe('failed transactions', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].failed).toBe(true)
@@ -543,7 +553,7 @@ describe('failed transactions', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions).toHaveLength(1)
@@ -557,7 +567,7 @@ describe('failed transactions', () => {
     mockFetchBlockTransactions.mockResolvedValue([makeTx({ status: undefined, amount: 100_000 })])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].failed).toBe(false)
@@ -587,7 +597,7 @@ describe('batch and internal operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then – one BlockTransaction with both ops' ops and combined fees
     expect(result.transactions).toHaveLength(1)
@@ -606,7 +616,7 @@ describe('batch and internal operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].feesPayer).toBe('tz1ActualPayer')
@@ -621,7 +631,7 @@ describe('batch and internal operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions).toHaveLength(2)
@@ -647,7 +657,7 @@ describe('FA token transfers', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then – one BlockTransaction with 2 native ops + 2 token ops
     expect(result.transactions).toHaveLength(1)
@@ -670,7 +680,7 @@ describe('FA token transfers', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then – failed tx has no operations at all
     expect(result.transactions).toHaveLength(1)
@@ -686,7 +696,7 @@ describe('FA token transfers', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then – one standalone BlockTransaction with 2 token ops
     expect(result.transactions).toHaveLength(1)
@@ -712,7 +722,7 @@ describe('FA token transfers', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then – merged into one standalone BlockTransaction with 4 ops (2 per transfer)
     expect(result.transactions).toHaveLength(1)
@@ -728,7 +738,7 @@ describe('FA token transfers', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then – each gets its own BlockTransaction
     expect(result.transactions).toHaveLength(2)
@@ -747,7 +757,7 @@ describe('FA token transfers', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then — peer must be absent when mint has no sender
     expect(result.transactions[0].operations).toHaveLength(1)
@@ -772,7 +782,7 @@ describe('FA token transfers', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then — peer must be absent when burn has no receiver
     expect(result.transactions[0].operations).toHaveLength(1)
@@ -792,7 +802,7 @@ describe('FA token transfers', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then – zero-amount transfer is silently discarded
     expect(result.transactions).toHaveLength(0)
@@ -816,7 +826,7 @@ describe('FA token transfers', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then — tokenId "7" must be part of the reference so different FA2 tokens under
     // the same contract are not conflated.
@@ -846,7 +856,7 @@ describe('FA token transfers', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     const op = result.transactions[0].operations[0] as any
@@ -874,7 +884,7 @@ describe('delegation operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions).toHaveLength(1)
@@ -909,7 +919,7 @@ describe('delegation operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions).toHaveLength(1)
@@ -934,7 +944,7 @@ describe('delegation operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].fees).toBe(650n)
@@ -946,7 +956,7 @@ describe('delegation operations', () => {
     mockFetchBlockDelegations.mockResolvedValue([makeDelegation({ status: 'failed' })])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then — failed but operations still present (listOperations path always includes them)
     expect(result.transactions[0].failed).toBe(true)
@@ -962,7 +972,7 @@ describe('delegation operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions).toHaveLength(1)
@@ -975,7 +985,7 @@ describe('delegation operations', () => {
     mockFetchBlockDelegations.mockResolvedValue([makeDelegation({ sender: null })])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions).toHaveLength(1)
@@ -993,7 +1003,7 @@ describe('delegation operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then - single BlockTransaction with both transfer and delegation operations, combined fees
     expect(result.transactions).toHaveLength(1)
@@ -1014,7 +1024,7 @@ describe('delegation operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then - the whole batch is marked as failed, operations cleared, fees accumulated
     expect(result.transactions).toMatchObject([{ failed: true, operations: [], fees: 800n }])
@@ -1031,7 +1041,7 @@ describe('delegation operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then - no operations (tx was failed), but fees still accumulated
     expect(result.transactions).toMatchObject([{ failed: true, operations: [], fees: 1000n }])
@@ -1044,7 +1054,7 @@ describe('delegation operations', () => {
     mockFetchBlockDelegations.mockResolvedValue([makeDelegation({ hash: 'opDelegation1' })])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions).toHaveLength(2)
@@ -1059,7 +1069,7 @@ describe('delegation operations', () => {
     mockFetchBlockDelegations.mockResolvedValue([makeDelegation({ status: undefined })])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].failed).toBe(false)
@@ -1072,7 +1082,7 @@ describe('delegation operations', () => {
     mockFetchBlockDelegations.mockResolvedValue([makeDelegation({ sender: null })])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     expect(result.transactions[0].feesPayer).toBeUndefined()
@@ -1086,7 +1096,7 @@ describe('delegation operations', () => {
     ])
 
     // When
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // Then
     const op = result.transactions[0].operations[0] as Record<string, unknown>
@@ -1112,7 +1122,7 @@ describe('staking operations', () => {
         makeStaking({ action, amount: Number(expectedStakedAmount) }),
       ])
 
-      const result = await getBlock(5_000_000)
+      const result = await getBlock(context, 5_000_000)
 
       expect(result.transactions).toHaveLength(1)
       const tx = result.transactions[0]
@@ -1140,7 +1150,7 @@ describe('staking operations', () => {
       makeStaking({ status: 'failed', amount: undefined, requestedAmount: 750_000_000 }),
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions[0].failed).toBe(true)
     expect(result.transactions[0].operations).toHaveLength(1)
@@ -1152,7 +1162,7 @@ describe('staking operations', () => {
     mockGetBlockByLevel.mockResolvedValue(makeBlock())
     mockFetchBlockStaking.mockResolvedValue([makeStaking({ sender: null })])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     expect(result.transactions[0].operations).toEqual([])
@@ -1165,7 +1175,7 @@ describe('staking operations', () => {
       makeStaking({ id: 301, hash: 'opStaking2' }),
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     expect(result.transactions[0].hash).toBe('opStaking2')
@@ -1180,7 +1190,7 @@ describe('staking operations', () => {
       makeStaking({ hash: 'opShared', action: 'stake', amount: 100, bakerFee: 300 }),
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     const tx = result.transactions[0]
@@ -1193,7 +1203,7 @@ describe('staking operations', () => {
     mockGetBlockByLevel.mockResolvedValue(makeBlock())
     mockFetchBlockStaking.mockResolvedValue([makeStaking({ baker: null })])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     const op = result.transactions[0].operations[0] as Record<string, unknown>
     expect(op).not.toHaveProperty('delegate')
@@ -1212,7 +1222,7 @@ describe('origination operations', () => {
       makeOrigination({ contractBalance: 0, counter: 42, gasLimit: 3494, storageLimit: 5852 }),
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     const tx = result.transactions[0]
@@ -1236,7 +1246,7 @@ describe('origination operations', () => {
     mockGetBlockByLevel.mockResolvedValue(makeBlock())
     mockFetchBlockOriginations.mockResolvedValue([makeOrigination({ contractBalance: 500_000 })])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     const tx = result.transactions[0]
@@ -1252,7 +1262,7 @@ describe('origination operations', () => {
     mockGetBlockByLevel.mockResolvedValue(makeBlock())
     mockFetchBlockOriginations.mockResolvedValue([makeOrigination({ contractBalance: -500 })])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions[0].operations[0]).not.toHaveProperty('amount')
   })
@@ -1263,7 +1273,7 @@ describe('origination operations', () => {
       makeOrigination({ status: 'failed', contractBalance: 500_000 }),
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     const tx = result.transactions[0]
@@ -1276,7 +1286,7 @@ describe('origination operations', () => {
     mockGetBlockByLevel.mockResolvedValue(makeBlock())
     mockFetchBlockOriginations.mockResolvedValue([makeOrigination({ hash: undefined })])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
     expect(result.transactions).toHaveLength(0)
   })
 
@@ -1286,7 +1296,7 @@ describe('origination operations', () => {
       makeOrigination({ sender: null, contractBalance: 100_000 }),
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     const tx = result.transactions[0]
@@ -1303,7 +1313,7 @@ describe('origination operations', () => {
       makeTokenTransfer({ transactionId: undefined, originationId: 400, amount: '1000' }),
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // origination tx should contain both the native transfer and the token ops
     const origTx = result.transactions.find((t) => t.hash === 'opOrig1')!
@@ -1324,7 +1334,7 @@ describe('origination operations', () => {
       { id: 999, hash: 'opCrossBlockOrig', block: 'BLsomeOtherBlock' },
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     // The token transfer should use the resolved origination hash, not "token-50"
     expect(mockGetOperationsOrigination).toHaveBeenCalledWith(0, undefined, { 'id.in': '999' })
@@ -1344,7 +1354,7 @@ describe('reveal operations', () => {
       makeReveal({ counter: 10, gasLimit: 10600, storageLimit: 0, bakerFee: 1420 }),
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     const tx = result.transactions[0]
@@ -1372,7 +1382,7 @@ describe('reveal operations', () => {
       makeReveal({ hash: 'opBatchedHash', bakerFee: 1420, sender: { address: 'tz1User' } }),
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     const tx = result.transactions[0]
@@ -1386,7 +1396,7 @@ describe('reveal operations', () => {
     mockGetBlockByLevel.mockResolvedValue(makeBlock())
     mockFetchBlockReveals.mockResolvedValue([makeReveal({ status: 'failed' })])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions[0].failed).toBe(true)
     expect(result.transactions[0].operations).toHaveLength(1)
@@ -1399,7 +1409,7 @@ describe('reveal operations', () => {
       makeReveal({ id: 501, hash: 'opReveal2' }),
     ])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     expect(result.transactions[0].hash).toBe('opReveal2')
@@ -1412,7 +1422,7 @@ describe('reveal operations', () => {
     ])
     mockFetchBlockReveals.mockResolvedValue([makeReveal({ hash: 'opBatchedHash', bakerFee: 1420 })])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     const tx = result.transactions[0]
@@ -1425,7 +1435,7 @@ describe('reveal operations', () => {
     mockGetBlockByLevel.mockResolvedValue(makeBlock())
     mockFetchBlockReveals.mockResolvedValue([makeReveal({ sender: null })])
 
-    const result = await getBlock(5_000_000)
+    const result = await getBlock(context, 5_000_000)
 
     expect(result.transactions).toHaveLength(1)
     expect(result.transactions[0].operations).toEqual([])
@@ -1443,24 +1453,24 @@ describe('network calls', () => {
     mockGetBlockByLevel.mockResolvedValue(makeBlock(height))
 
     // When
-    await getBlock(height)
+    await getBlock(context, height)
 
     // Then – getBlockByLevel is called once for the block and once for the parent
     expect(mockGetBlockByLevel).toHaveBeenCalledTimes(2)
     expect(mockGetBlockByLevel).toHaveBeenCalledWith(height)
     expect(mockGetBlockByLevel).toHaveBeenCalledWith(height - 1)
     expect(mockFetchBlockTransactions).toHaveBeenCalledTimes(1)
-    expect(mockFetchBlockTransactions).toHaveBeenCalledWith(height)
+    expect(mockFetchBlockTransactions).toHaveBeenCalledWith(expect.anything(), height)
     expect(mockFetchBlockTokenTransfers).toHaveBeenCalledTimes(1)
-    expect(mockFetchBlockTokenTransfers).toHaveBeenCalledWith(height)
+    expect(mockFetchBlockTokenTransfers).toHaveBeenCalledWith(expect.anything(), height)
     expect(mockFetchBlockDelegations).toHaveBeenCalledTimes(1)
-    expect(mockFetchBlockDelegations).toHaveBeenCalledWith(height)
+    expect(mockFetchBlockDelegations).toHaveBeenCalledWith(expect.anything(), height)
     expect(mockFetchBlockStaking).toHaveBeenCalledTimes(1)
-    expect(mockFetchBlockStaking).toHaveBeenCalledWith(height)
+    expect(mockFetchBlockStaking).toHaveBeenCalledWith(expect.anything(), height)
     expect(mockFetchBlockOriginations).toHaveBeenCalledTimes(1)
-    expect(mockFetchBlockOriginations).toHaveBeenCalledWith(height)
+    expect(mockFetchBlockOriginations).toHaveBeenCalledWith(expect.anything(), height)
     expect(mockFetchBlockReveals).toHaveBeenCalledTimes(1)
-    expect(mockFetchBlockReveals).toHaveBeenCalledWith(height)
+    expect(mockFetchBlockReveals).toHaveBeenCalledWith(expect.anything(), height)
   })
 
   it('propagates errors from getBlockByLevel', async () => {
@@ -1468,7 +1478,7 @@ describe('network calls', () => {
     mockGetBlockByLevel.mockRejectedValue(new Error('block fetch failed'))
 
     // When / Then
-    await expect(getBlock(1_000_000)).rejects.toThrow('block fetch failed')
+    await expect(getBlock(context, 1_000_000)).rejects.toThrow('block fetch failed')
   })
 
   it('propagates errors from fetchBlockTransactions', async () => {
@@ -1477,7 +1487,7 @@ describe('network calls', () => {
     mockFetchBlockTransactions.mockRejectedValue(new Error('tx fetch failed'))
 
     // When / Then
-    await expect(getBlock(1_000_000)).rejects.toThrow('tx fetch failed')
+    await expect(getBlock(context, 1_000_000)).rejects.toThrow('tx fetch failed')
   })
 
   it('propagates errors from fetchBlockTokenTransfers', async () => {
@@ -1486,7 +1496,7 @@ describe('network calls', () => {
     mockFetchBlockTokenTransfers.mockRejectedValue(new Error('token fetch failed'))
 
     // When / Then
-    await expect(getBlock(1_000_000)).rejects.toThrow('token fetch failed')
+    await expect(getBlock(context, 1_000_000)).rejects.toThrow('token fetch failed')
   })
 
   it('propagates errors from fetchBlockDelegations', async () => {
@@ -1495,6 +1505,6 @@ describe('network calls', () => {
     mockFetchBlockDelegations.mockRejectedValue(new Error('delegation fetch failed'))
 
     // When / Then
-    await expect(getBlock(1_000_000)).rejects.toThrow('delegation fetch failed')
+    await expect(getBlock(context, 1_000_000)).rejects.toThrow('delegation fetch failed')
   })
 })