@ledgerhq/coin-tezos 7.12.0 → 9.0.0

package/src/logic/getBalance.ts

@@ -1,15 +1,18 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
+import type { TezosContext } from '../config'
 import type { Balance } from '@ledgerhq/coin-module-framework/api/index'
 import { log } from '@ledgerhq/logs'
 import { hasManagerKey } from '../network/types'
-import api from '../network/tzkt'
+import { createTzktApi } from '../network/tzkt'
 import { partitionNativeBalance } from '../utils'
 import { buildStakesForAccount, fetchUnstakeRequests } from './getStakes'
 
 /** Returns `[native, ...stakes, ...tokens]` per the Paris upgrade. */
-export async function getBalance(address: string): Promise<Balance[]> {
+export async function getBalance(context: TezosContext, address: string): Promise<Balance[]> {
+  const config = await context.config()
+  const api = createTzktApi(config)
   const [apiAccountResult, tokensBalancesResult] = await Promise.allSettled([
     api.getAccountByAddress(address),
     api.getTokensBalances(address),
@@ -56,7 +59,7 @@ export async function getBalance(address: string): Promise<Balance[]> {
 
   const normalized = BigInt(apiAccount.balance)
 
-  const unstakeRequests = await fetchUnstakeRequests(address, apiAccount).catch((error) => {
+  const unstakeRequests = await fetchUnstakeRequests(config, address, apiAccount).catch((error) => {
     log('coin:tezos', 'getBalance: fetchUnstakeRequests failed; degrading stakes to []', {
       error,
       address,

package/src/logic/getBlock.integ.test.ts

@@ -1,9 +1,8 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
-import type { TezosCoinConfig } from '../config'
+import type { TezosCoinConfig, TezosContext } from '../config'
 import type { OtherBlockOperation } from '@ledgerhq/coin-module-framework/api/types'
-import coinConfig from '../config'
 import { getBlock } from './getBlock'
 
 /**
@@ -15,8 +14,6 @@ import { getBlock } from './getBlock'
  * (1 mutez from a tz2 to a tz1), making the assertions deterministic.
  */
 
-let originalGetCoinConfig: () => TezosCoinConfig
-
 const shadownetConfig = (): TezosCoinConfig =>
   ({
     status: { type: 'active' },
@@ -32,20 +29,14 @@ const shadownetConfig = (): TezosCoinConfig =>
     },
   }) as TezosCoinConfig
 
-describe('getBlock', () => {
-  beforeAll(() => {
-    originalGetCoinConfig = coinConfig.getCoinConfig
-    coinConfig.setCoinConfig(shadownetConfig)
-  })
-
-  afterAll(() => {
-    if (originalGetCoinConfig) {
-      coinConfig.setCoinConfig(originalGetCoinConfig)
-    }
-  })
+const context: TezosContext = {
+  config: async () => shadownetConfig(),
+  logger: () => undefined,
+}
 
+describe('getBlock', () => {
   it('should fetch and parse block 3113219 correctly', async () => {
-    const block = await getBlock(3113219)
+    const block = await getBlock(context, 3113219)
 
     expect(block.info).toMatchObject({
       hash: 'BLAf8kVQvpsgXNA914eLo3gJKauywozjuravm7cmdvZ9VEM4QyK',
@@ -109,26 +100,13 @@ const STAKE_BLOCK = 3106279
 const UNSTAKE_BLOCK = 3106307
 
 describe('getBlock — Shadownet Paris staking ops', () => {
-  beforeAll(() => {
-    if (originalGetCoinConfig === undefined) {
-      originalGetCoinConfig = coinConfig.getCoinConfig
-    }
-    coinConfig.setCoinConfig(shadownetConfig)
-  })
-
-  afterAll(() => {
-    if (originalGetCoinConfig) {
-      coinConfig.setCoinConfig(originalGetCoinConfig)
-    }
-  })
-
   it.each([
     [STAKE_BLOCK, 'STAKE', 500_000_000],
     [UNSTAKE_BLOCK, 'UNSTAKE', 250_000_000],
   ])(
     'block %s contains a staking op with ledgerOpType=%s and stakedAmount=%s',
     async (height, expectedOpType, expectedAmount) => {
-      const block = await getBlock(height)
+      const block = await getBlock(context, height)
       const stakingTx = block.transactions.find((tx) =>
         tx.operations.some((op) => (op as Record<string, unknown>).ledgerOpType === expectedOpType)
       )