@ledgerhq/coin-tezos 9.0.0 → 10.1.1

package/src/api/index-mainnet.integ.test.ts

@@ -2,7 +2,6 @@
 // SPDX-License-Identifier: Apache-2.0
 
 import type { TezosCoinConfig, TezosConfig, TezosContext } from '../config'
-import type { TezosApi } from './types'
 import { localForger } from '@taquito/local-forging'
 import { OpKind } from '@taquito/rpc'
 import { createTzktApi } from '../network/tzkt'
@@ -55,7 +54,7 @@ type DecodedOperation = {
  * Use this test suite in last resort
  */
 describe('Tezos Api - Mainnet', () => {
-  const module: TezosApi = createApi()
+  const module = createApi()
   const context: TezosContext = {
     config: async () => defaultConfig as TezosCoinConfig,
     logger: () => undefined,

package/src/api/index.integ.test.ts

@@ -2,7 +2,6 @@
 // SPDX-License-Identifier: Apache-2.0
 
 import type { TezosCoinConfig, TezosConfig, TezosContext } from '../config'
-import type { TezosApi } from './types'
 import type { SendTransactionIntent } from '@ledgerhq/coin-module-framework/api/types'
 import { localForger } from '@taquito/local-forging'
 import { createApi } from '.'
@@ -26,7 +25,7 @@ const defaultConfig: TezosConfig = {
  * https://api.shadownet.tzkt.io
  */
 describe('Tezos Api', () => {
-  const module: TezosApi = createApi()
+  const module = createApi()
   const context: TezosContext = {
     config: async () => defaultConfig as TezosCoinConfig,
     logger: () => undefined,

package/src/api/index.test.ts

@@ -995,6 +995,8 @@ describe('getBalance', () => {
 })
 
 describe('getAccountInfo', () => {
+  // `getAccountInfo` is optional on the generic CoinModuleApi (ADR-045 / LIVE-33431); coin-tezos
+  // always implements it, so the non-null assertions below are safe.
   afterEach(() => {
     mockGetAccountByAddress.mockClear()
   })
@@ -1013,7 +1015,7 @@ describe('getAccountInfo', () => {
       firstActivityTime: '2021-01-01T00:00:00Z',
     } as APIAccount)
 
-    await expect(api.getAccountInfo(context, 'tz1test')).resolves.toEqual({
+    await expect(api.getAccountInfo!(context, 'tz1test')).resolves.toEqual({
       type: 'tezos',
       revealed: true,
     })
@@ -1034,7 +1036,7 @@ describe('getAccountInfo', () => {
       firstActivityTime: '2021-01-01T00:00:00Z',
     } as APIAccount)
 
-    await expect(api.getAccountInfo(context, 'tz2unrevealed')).resolves.toEqual({
+    await expect(api.getAccountInfo!(context, 'tz2unrevealed')).resolves.toEqual({
       type: 'tezos',
       revealed: false,
     })
@@ -1054,7 +1056,7 @@ describe('getAccountInfo', () => {
     } as APIAccount)
 
     await expect(
-      api.getAccountInfo(context, 'tz3RDC3Jdn4j15J7bBHZd29EUee9gVB1CxD9')
+      api.getAccountInfo!(context, 'tz3RDC3Jdn4j15J7bBHZd29EUee9gVB1CxD9')
     ).resolves.toEqual({
       type: 'tezos',
       revealed: true,
@@ -1069,7 +1071,7 @@ describe('getAccountInfo', () => {
       counter: 0,
     } as APIAccount)
 
-    await expect(api.getAccountInfo(context, 'tz1empty')).resolves.toEqual({
+    await expect(api.getAccountInfo!(context, 'tz1empty')).resolves.toEqual({
       type: 'tezos',
       revealed: false,
     })
@@ -1079,6 +1081,6 @@ describe('getAccountInfo', () => {
   it('propagates errors from getAccountByAddress', async () => {
     mockGetAccountByAddress.mockRejectedValueOnce(new Error('boom'))
 
-    await expect(api.getAccountInfo(context, 'tz1boom')).rejects.toThrow('boom')
+    await expect(api.getAccountInfo!(context, 'tz1boom')).rejects.toThrow('boom')
   })
 })

package/src/api/index.ts

@@ -1,10 +1,10 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
-import type { TezosContext } from '../config'
+import type { TezosCoinConfig, TezosContext } from '../config'
 import type { CoreAccountInfo, CoreTransactionInfo, EstimatedFees } from '../logic/estimateFees'
 import type { TezosOperationMode } from '../types/model'
-import type { TezosAccountInfo, TezosApi, TezosFeeEstimation } from './types'
+import type { TezosAccountInfo, TezosFeeEstimation } from './types'
 import type {
   CraftedTransaction,
   Cursor,
@@ -15,11 +15,16 @@ import type {
 } from '@ledgerhq/coin-module-framework/api/index'
 import type {
   BalanceOptions,
+  CoinModuleApi,
   FeeEstimation,
   TransactionIntent,
 } from '@ledgerhq/coin-module-framework/api/types'
 import { rejectBalanceOptions } from '@ledgerhq/coin-module-framework/api/getBalance/rejectBalanceOptions'
-import { IncorrectTypeError, type Operation } from '@ledgerhq/coin-module-framework/api/index'
+import {
+  IncorrectTypeError,
+  notSupported,
+  type Operation,
+} from '@ledgerhq/coin-module-framework/api/index'
 import { craftTransactionData } from '@ledgerhq/coin-module-framework/logic/craftTransactionData'
 import { log } from '@ledgerhq/logs'
 import { getRevealFee } from '@taquito/taquito'
@@ -52,7 +57,7 @@ import {
 } from '../utils'
 
 // The caller builds the {@link TezosContext} (config + logger) and passes it to each method (ADR-019).
-export function createApi(): TezosApi {
+export function createApi(): CoinModuleApi<TezosCoinConfig> {
   return {
     broadcast: (context, tx) => broadcast(context, tx),
     async call() {
@@ -96,6 +101,7 @@ export function createApi(): TezosApi {
     getRewards(_context, _address: string, _options?: { cursor?: Cursor }): Promise<Page<Reward>> {
       throw new Error('getRewards is not supported')
     },
+    register: notSupported('register'),
     getValidators(_context, _options?: { cursor?: Cursor }): Promise<Page<Validator>> {
       throw new Error('getValidators is not supported')
     },

package/src/api/types.ts

@@ -1,8 +1,7 @@
 // SPDX-FileCopyrightText: © 2026 LEDGER SAS
 // SPDX-License-Identifier: Apache-2.0
 
-import type { TezosCoinConfig, TezosContext } from '../config'
-import type { CoinModuleApi, FeeEstimation } from '@ledgerhq/coin-module-framework/api/types'
+import type { FeeEstimation } from '@ledgerhq/coin-module-framework/api/types'
 
 export type TezosFeeParameters = {
   gasLimit: bigint
@@ -19,20 +18,9 @@ export type TezosSender = { address: string; xpub?: string }
 /**
  * Network-specific account metadata for Tezos (ADR-045 shape).
  * `revealed` is `true` once the account has published its public key on-chain.
- */
-export type TezosAccountInfo = { type: 'tezos'; revealed: boolean }
-
-/**
- * coin-tezos public API surface: the generic {@link CoinModuleApi} plus `getAccountInfo`,
- * which exposes the account's on-chain reveal state.
  *
- * Kept local (rather than relying on the future framework-side reveal-state type from ADR-045)
- * so coin-tezos does not depend on that framework change landing first. The `{ type: "tezos" }`
- * shape is intentionally identical to the ADR-045 variant. Once the framework's `CoinModuleApi`
- * ships its own `getAccountInfo`, migrating means dropping this local override entirely (the two
- * signatures would otherwise intersect into an incompatible-return-type collision) and letting the
- * implementation satisfy the generic method directly.
+ * Concrete return type of the module's `getAccountInfo`. The framework's generic
+ * `CoinModuleApi.getAccountInfo` (added by ADR-045 / LIVE-33431) widens to the open
+ * `AccountInfo` shape; this alias keeps the reveal state strongly typed at the implementation.
  */
-export type TezosApi = CoinModuleApi<TezosCoinConfig> & {
-  getAccountInfo: (context: TezosContext, address: string) => Promise<TezosAccountInfo>
-}
+export type TezosAccountInfo = { type: 'tezos'; revealed: boolean }

package/src/logic/getBalance.test.ts

@@ -3,10 +3,27 @@
 
 import type { TezosCoinConfig, TezosContext } from '../config'
 import type { APITokenBalance } from '../network/types'
-import { HttpResponse, http } from 'msw'
-import { setupServer } from 'msw/node'
 import { getBalance } from './getBalance'
 
+const mockedGetAccountByAddress = jest.fn()
+const mockedGetTokensBalances = jest.fn()
+const mockedGetUnstakeRequests = jest.fn()
+
+jest.mock('../network/tzkt', () => {
+  const originalModule = jest.requireActual('../network/tzkt')
+  return {
+    ...originalModule,
+    createTzktApi: () => {
+      return {
+        ...originalModule.api,
+        getAccountByAddress: mockedGetAccountByAddress,
+        getTokensBalances: mockedGetTokensBalances,
+        getUnstakeRequests: mockedGetUnstakeRequests,
+      }
+    },
+  }
+})
+
 const mockTokensBalancesWithAllInfo: APITokenBalance[] = [
   {
     id: 2398642347442177,
@@ -51,14 +68,8 @@ const mockTokensBalancesWithMissingInfo: APITokenBalance[] = [
 ]
 
 describe('getBalance', () => {
-  const mockServer = setupServer()
-
-  beforeAll(() => {
-    mockServer.listen({ onUnhandledRequest: 'error' })
-  })
-
-  afterAll(() => {
-    mockServer.close()
+  beforeEach(() => {
+    jest.resetAllMocks()
   })
 
   const context: TezosContext = {
@@ -70,35 +81,31 @@ describe('getBalance', () => {
     logger: () => undefined,
   }
   it('gets the balance of a Tezos account', async () => {
-    mockServer.use(
-      http.get('http://tezos.explorer.com/v1/accounts/tz1WvvbEGpBXGeTVbLiR6DYBe1izmgiYuZbq', () =>
-        HttpResponse.json({ type: 'empty' })
-      ),
-      http.get('http://tezos.explorer.com/v1/accounts/tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oSM8', () =>
-        HttpResponse.json({ type: 'user', balance: 25 })
-      ),
-      http.get('http://tezos.explorer.com/v1/accounts/tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oMMM', () =>
-        HttpResponse.json({
+    mockedGetAccountByAddress.mockImplementation((address: string) => {
+      if (address === 'tz1WvvbEGpBXGeTVbLiR6DYBe1izmgiYuZbq') {
+        return { type: 'empty' }
+      } else if (address === 'tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oSM8') {
+        return { type: 'user', balance: 25 }
+      } else if (address === 'tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oMMM') {
+        return {
           type: 'user',
           balance: 15,
           delegate: { address: 'tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oMMM' },
-        })
-      ),
-      http.get('http://tezos.explorer.com/v1/tokens/balances', ({ request }) => {
-        const url = new URL(request.url)
-        const account = url.searchParams.get('account')
-
-        // Return a mocked JSON response when the "account"
-        // search parameter equals a specific address.
-        if (account === 'tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oSM8') {
-          return HttpResponse.json(mockTokensBalancesWithAllInfo)
-        } else if (account === 'tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oMMM') {
-          return HttpResponse.json(mockTokensBalancesWithMissingInfo)
-        } else {
-          return HttpResponse.json([])
         }
-      })
-    )
+      } else {
+        throw new Error('address not handled by the test: ' + address)
+      }
+    })
+
+    mockedGetTokensBalances.mockImplementation((address: string) => {
+      if (address === 'tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oSM8') {
+        return mockTokensBalancesWithAllInfo
+      } else if (address === 'tz1TzrmTBSuiVHV2VfMnGRMYvTEPCP42oMMM') {
+        return mockTokensBalancesWithMissingInfo
+      } else {
+        return []
+      }
+    })
 
     expect(await getBalance(context, 'tz1WvvbEGpBXGeTVbLiR6DYBe1izmgiYuZbq')).toEqual([
       {
@@ -164,15 +171,9 @@ describe('getBalance', () => {
     const delegateAddress = 'tz1BakerAddr'
 
     function mockAccount(account: Record<string, unknown>) {
-      mockServer.use(
-        http.get(`http://tezos.explorer.com/v1/accounts/${address}`, () =>
-          HttpResponse.json({ type: 'user', ...account })
-        ),
-        http.get('http://tezos.explorer.com/v1/tokens/balances', () => HttpResponse.json([])),
-        http.get('http://tezos.explorer.com/v1/staking/unstake_requests', () =>
-          HttpResponse.json([])
-        )
-      )
+      mockedGetAccountByAddress.mockResolvedValue({ type: 'user', ...account })
+      mockedGetTokensBalances.mockResolvedValue([])
+      mockedGetUnstakeRequests.mockResolvedValue([])
     }
 
     it('attaches a delegation Stake when only delegate is set', async () => {
@@ -271,31 +272,25 @@ describe('getBalance', () => {
     })
 
     it("attaches an unstaking Stake with state 'deactivating' when a pending request exists", async () => {
-      mockServer.use(
-        http.get(`http://tezos.explorer.com/v1/accounts/${address}`, () =>
-          HttpResponse.json({
-            type: 'user',
-            balance: 100,
-            stakedBalance: 30,
-            unstakedBalance: 10,
-            delegate: { address: delegateAddress },
-          })
-        ),
-        http.get('http://tezos.explorer.com/v1/staking/unstake_requests', () =>
-          HttpResponse.json([
-            {
-              id: 77,
-              cycle: 100,
-              baker: { address: delegateAddress },
-              staker: { address },
-              firstTime: '2026-05-01T00:00:00Z',
-              status: 'pending',
-              actualAmount: 10,
-            },
-          ])
-        ),
-        http.get('http://tezos.explorer.com/v1/tokens/balances', () => HttpResponse.json([]))
-      )
+      mockedGetAccountByAddress.mockResolvedValue({
+        type: 'user',
+        balance: 100,
+        stakedBalance: 30,
+        unstakedBalance: 10,
+        delegate: { address: delegateAddress },
+      })
+      mockedGetTokensBalances.mockResolvedValue([])
+      mockedGetUnstakeRequests.mockResolvedValue([
+        {
+          id: 77,
+          cycle: 100,
+          baker: { address: delegateAddress },
+          staker: { address },
+          firstTime: '2026-05-01T00:00:00Z',
+          status: 'pending',
+          actualAmount: 10,
+        },
+      ])
 
       const result = await getBalance(context, address)
 
@@ -318,39 +313,33 @@ describe('getBalance', () => {
     })
 
     it('emits per-request unstaking and finalizable Stakes when both statuses are present', async () => {
-      mockServer.use(
-        http.get(`http://tezos.explorer.com/v1/accounts/${address}`, () =>
-          HttpResponse.json({
-            type: 'user',
-            balance: 100,
-            unstakedBalance: 50,
-            delegate: { address: delegateAddress },
-          })
-        ),
-        http.get('http://tezos.explorer.com/v1/staking/unstake_requests', () =>
-          HttpResponse.json([
-            {
-              id: 1,
-              cycle: 100,
-              baker: { address: delegateAddress },
-              staker: { address },
-              firstTime: '2026-05-01T00:00:00Z',
-              status: 'pending',
-              actualAmount: 20,
-            },
-            {
-              id: 2,
-              cycle: 99,
-              baker: { address: delegateAddress },
-              staker: { address },
-              firstTime: '2026-04-25T00:00:00Z',
-              status: 'finalizable',
-              actualAmount: 30,
-            },
-          ])
-        ),
-        http.get('http://tezos.explorer.com/v1/tokens/balances', () => HttpResponse.json([]))
-      )
+      mockedGetAccountByAddress.mockResolvedValue({
+        type: 'user',
+        balance: 100,
+        unstakedBalance: 50,
+        delegate: { address: delegateAddress },
+      })
+      mockedGetUnstakeRequests.mockResolvedValue([
+        {
+          id: 1,
+          cycle: 100,
+          baker: { address: delegateAddress },
+          staker: { address },
+          firstTime: '2026-05-01T00:00:00Z',
+          status: 'pending',
+          actualAmount: 20,
+        },
+        {
+          id: 2,
+          cycle: 99,
+          baker: { address: delegateAddress },
+          staker: { address },
+          firstTime: '2026-04-25T00:00:00Z',
+          status: 'finalizable',
+          actualAmount: 30,
+        },
+      ])
+      mockedGetTokensBalances.mockResolvedValue([])
 
       const result = await getBalance(context, address)
 
@@ -396,30 +385,25 @@ describe('getBalance', () => {
       // A delegate has no `delegate` field (it is its own baker), so there is no delegation position —
       // only its self-stake and any unstake requests. The unstakedBalance > 0 exercises the
       // delegate → fetchUnstakeRequests path.
-      mockServer.use(
-        http.get(`http://tezos.explorer.com/v1/accounts/${address}`, () =>
-          HttpResponse.json({
-            type: 'delegate',
-            balance: 100,
-            stakedBalance: 30,
-            unstakedBalance: 10,
-          })
-        ),
-        http.get('http://tezos.explorer.com/v1/tokens/balances', () => HttpResponse.json([])),
-        http.get('http://tezos.explorer.com/v1/staking/unstake_requests', () =>
-          HttpResponse.json([
-            {
-              id: 77,
-              cycle: 100,
-              baker: { address },
-              staker: { address },
-              firstTime: '2026-05-01T00:00:00Z',
-              status: 'pending',
-              actualAmount: 10,
-            },
-          ])
-        )
-      )
+
+      mockedGetAccountByAddress.mockResolvedValue({
+        type: 'delegate',
+        balance: 100,
+        stakedBalance: 30,
+        unstakedBalance: 10,
+      })
+      mockedGetTokensBalances.mockResolvedValue([])
+      mockedGetUnstakeRequests.mockResolvedValue([
+        {
+          id: 77,
+          cycle: 100,
+          baker: { address },
+          staker: { address },
+          firstTime: '2026-05-01T00:00:00Z',
+          status: 'pending',
+          actualAmount: 10,
+        },
+      ])
 
       expect(await getBalance(context, address)).toEqual([
         { value: 100n, asset: { type: 'native' }, locked: 40n },
@@ -461,20 +445,14 @@ describe('getBalance', () => {
     })
 
     it('still excludes unstaked-frozen funds from spendable when the unstake_requests endpoint fails', async () => {
-      mockServer.use(
-        http.get(`http://tezos.explorer.com/v1/accounts/${address}`, () =>
-          HttpResponse.json({
-            type: 'user',
-            balance: 100,
-            unstakedBalance: 10,
-            delegate: { address: delegateAddress },
-          })
-        ),
-        http.get('http://tezos.explorer.com/v1/staking/unstake_requests', () =>
-          HttpResponse.json({ error: 'internal' }, { status: 500 })
-        ),
-        http.get('http://tezos.explorer.com/v1/tokens/balances', () => HttpResponse.json([]))
-      )
+      mockedGetAccountByAddress.mockResolvedValue({
+        type: 'user',
+        balance: 100,
+        unstakedBalance: 10,
+        delegate: { address: delegateAddress },
+      })
+      mockedGetUnstakeRequests.mockRejectedValue(new Error('endpoint failure'))
+      mockedGetTokensBalances.mockResolvedValue([])
 
       const result = await getBalance(context, address)