@circle-fin/usdckit 0.21.0 → 0.22.0

package/dist/cjs/actions/generateTransferLink.d.ts

@@ -1,5 +1,5 @@
-import type { Account, Address, NativeUnits, BaseUnits, ExtendedViemClient } from '../types.js';
-import type { Abi, Chain as ViemChain } from 'viem';
+import type { Account, Address, NativeUnits, BaseUnits } from '../types.js';
+import type { Abi, Chain as ViemChain, Client } from 'viem';
 export type GenerateTransferLinkParams = {
     /** The recipient address. */
     to: Address | Account;
@@ -32,4 +32,4 @@ export type GenerateTransferLinkParams = {
  * // => 'ethereum:0xRecipientAddress@11155111?value=100&token=0xTokenAddress'
  * ```
  */
-export declare function generateTransferLink(client: ExtendedViemClient, params: GenerateTransferLinkParams): Promise<string>;
+export declare function generateTransferLink(client: Client, params: GenerateTransferLinkParams): Promise<string>;

package/dist/cjs/actions/generateTransferLink.js

@@ -37,6 +37,9 @@ const toERC681_js_1 = require("../utils/toERC681.js");
  */
 async function generateTransferLink(client, params) {
     const { to, amount: _amount, chain: _chain = client.chain } = params;
+    if (_chain === undefined) {
+        throw new Error('chain is required');
+    }
     const toAccount = (0, parseAccount_js_1.parseAccount)(to, { chain: _chain });
     const chain = toAccount.chain;
     const token = params.token != null ? (0, parseAddress_js_1.parseAddress)(params.token) : null;

package/dist/cjs/actions/generateTransferLink.test.js

@@ -61,4 +61,21 @@ vitest_1.describe.concurrent('generateTransferLink', () => {
     ])('return a valid URL', async (params, expectedResult) => {
         await (0, vitest_1.expect)((0, generateTransferLink_js_1.generateTransferLink)(client, params)).resolves.toBe(expectedResult);
     });
+    (0, vitest_1.test)('should throw error when chain is undefined', async () => {
+        const clientWithoutChain = {
+            chain: undefined,
+            getContract: vitest_1.vi.fn(({ address }) => ({
+                address,
+                read: {
+                    decimals: vitest_1.vi.fn(() => 6),
+                },
+                abi: viem_1.erc20Abi,
+            })),
+        };
+        const params = {
+            to: '0x1c7D4B196Cb0C7B01d743Fbc6116a902379C7238',
+            amount: '1',
+        };
+        await (0, vitest_1.expect)((0, generateTransferLink_js_1.generateTransferLink)(clientWithoutChain, params)).rejects.toThrow('chain is required');
+    });
 });

package/dist/cjs/actions/getSwapExactInQuote.d.ts

@@ -1,6 +1,6 @@
 import type { Chain } from '../providers/uniswap/index.js';
-import type { Address, BaseUnits, ControlFlowOptions, ExtendedViemClient, NativeUnits } from '../types.js';
-import type { Abi } from 'viem';
+import type { Address, BaseUnits, ControlFlowOptions, NativeUnits } from '../types.js';
+import type { Abi, Client } from 'viem';
 /**
  * Parameters for the {@link getSwapExactInQuote} function.
  */
@@ -68,4 +68,4 @@ export type GetSwapExactInQuoteReturnType = {
  * await client.swap({ ...quote, account })
  * ```
  */
-export declare function getSwapExactInQuote(client: ExtendedViemClient, params: GetSwapExactInQuoteParams & ControlFlowOptions): Promise<GetSwapExactInQuoteReturnType>;
+export declare function getSwapExactInQuote(client: Client, params: GetSwapExactInQuoteParams & ControlFlowOptions): Promise<GetSwapExactInQuoteReturnType>;

package/dist/cjs/actions/getTokenBalance.d.ts

@@ -0,0 +1,30 @@
+import type { Account, Address, BaseUnits, ChainContract, NativeUnits } from '../types.js';
+import type { Chain, Client } from 'viem';
+export type GetTokenBalanceInput = {
+    account: Account;
+    chain?: Chain;
+    token?: Address | ChainContract;
+};
+export type GetTokenBalanceReturnType = {
+    /**
+     * The token address, or undefined for native token
+     */
+    token: Address | undefined;
+    /**
+     * The number of decimals used by the token
+     */
+    decimals: number | undefined;
+    /**
+     * The token symbol (e.g., "ETH", "USDC")
+     */
+    symbol: string | undefined;
+    /**
+     * The balance value in the smallest indivisible unit
+     */
+    value: BaseUnits;
+    /**
+     * The formatted balance
+     */
+    formatted: NativeUnits;
+};
+export declare function getTokenBalance(client: Client, params: GetTokenBalanceInput): Promise<GetTokenBalanceReturnType>;

package/dist/cjs/actions/getTokenBalance.js

@@ -0,0 +1,51 @@
+"use strict";
+// Copyright (c) 2025, Circle Internet Group, Inc.
+// All rights reserved.
+//
+// Circle Internet Group, Inc. CONFIDENTIAL
+//
+// This file includes unpublished proprietary source code of Circle Internet
+// Group, Inc. The copyright notice above does not
+// evidence any actual or intended publication of such source code. Disclosure
+// of this source code or any related proprietary information is strictly
+// prohibited without the express written permission of Circle Internet Group, Inc.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getTokenBalance = getTokenBalance;
+const viem_1 = require("viem");
+const actions_1 = require("viem/actions");
+const utils_1 = require("viem/utils");
+const getClient_js_1 = require("../utils/getClient.js");
+async function getTokenBalance(client, params) {
+    const { account, token } = params;
+    // Get the client for the given chain.
+    const chain = (account['chain'] ?? params.chain ?? client.chain);
+    const _client = (0, getClient_js_1.getClient)(client, chain);
+    const readContract = (0, utils_1.getAction)(_client, actions_1.readContract, 'readContract');
+    const getBalance = (0, utils_1.getAction)(_client, actions_1.getBalance, 'getBalance');
+    // If token is given, get the balance of the given token from reading the contract.
+    if (token !== undefined) {
+        const address = typeof token === 'string' ? token : token.address;
+        const abi = viem_1.erc20Abi;
+        const [balance, decimals, symbol] = await Promise.all([
+            readContract({ address, abi, functionName: 'balanceOf', args: [account.address] }),
+            readContract({ address, abi, functionName: 'decimals' }),
+            readContract({ address, abi, functionName: 'symbol' }),
+        ]);
+        return {
+            token: address,
+            decimals,
+            symbol,
+            value: balance,
+            formatted: (0, utils_1.formatUnits)(balance, decimals),
+        };
+    }
+    // Get the balance of the native token.
+    const balance = await getBalance({ address: account.address });
+    return {
+        token: undefined,
+        decimals: chain.nativeCurrency.decimals,
+        symbol: chain.nativeCurrency.symbol,
+        value: balance,
+        formatted: (0, utils_1.formatUnits)(balance, chain.nativeCurrency.decimals),
+    };
+}

package/dist/cjs/actions/getTokenBalance.test.js

@@ -0,0 +1,165 @@
+"use strict";
+// Copyright (c) 2025, Circle Internet Group, Inc.
+// All rights reserved.
+//
+// Circle Internet Group, Inc. CONFIDENTIAL
+//
+// This file includes unpublished proprietary source code of Circle Internet
+// Group, Inc. The copyright notice above does not
+// evidence any actual or intended publication of such source code. Disclosure
+// of this source code or any related proprietary information is strictly
+// prohibited without the express written permission of Circle Internet Group, Inc.
+Object.defineProperty(exports, "__esModule", { value: true });
+const viem_1 = require("viem");
+const vitest_1 = require("vitest");
+const index_js_1 = require("../chains/index.js");
+const getTokenBalance_js_1 = require("./getTokenBalance.js");
+vitest_1.vi.mock('../utils/getClient.js', () => ({
+    getClient: vitest_1.vi.fn((client, _chain) => client),
+}));
+(0, vitest_1.describe)('getTokenBalance()', () => {
+    (0, vitest_1.beforeEach)(() => {
+        vitest_1.vi.restoreAllMocks();
+    });
+    const mockClient = {
+        chain: index_js_1.ETH_SEPOLIA,
+        readContract: vitest_1.vi.fn(),
+        getBalance: vitest_1.vi.fn(),
+    };
+    (0, vitest_1.it)('should get ERC20 token balance', async () => {
+        const account = { address: '0x123' };
+        const tokenAddress = '0xabc';
+        mockClient.readContract
+            .mockResolvedValueOnce(1000000n) // balanceOf
+            .mockResolvedValueOnce(6) // decimals
+            .mockResolvedValueOnce('USDC'); // symbol
+        const result = await (0, getTokenBalance_js_1.getTokenBalance)(mockClient, {
+            account,
+            token: tokenAddress,
+        });
+        (0, vitest_1.expect)(mockClient.readContract).toHaveBeenCalledTimes(3);
+        (0, vitest_1.expect)(mockClient.readContract).toHaveBeenNthCalledWith(1, {
+            address: tokenAddress,
+            abi: viem_1.erc20Abi,
+            functionName: 'balanceOf',
+            args: [account.address],
+        });
+        (0, vitest_1.expect)(mockClient.readContract).toHaveBeenNthCalledWith(2, {
+            address: tokenAddress,
+            abi: viem_1.erc20Abi,
+            functionName: 'decimals',
+        });
+        (0, vitest_1.expect)(mockClient.readContract).toHaveBeenNthCalledWith(3, {
+            address: tokenAddress,
+            abi: viem_1.erc20Abi,
+            functionName: 'symbol',
+        });
+        (0, vitest_1.expect)(result).toEqual({
+            token: tokenAddress,
+            decimals: 6,
+            symbol: 'USDC',
+            value: 1000000n,
+            formatted: '1',
+        });
+    });
+    (0, vitest_1.it)('should get ERC20 token balance with ChainContract input', async () => {
+        const account = { address: '0x123' };
+        const token = { address: '0xabc' };
+        mockClient.readContract
+            .mockResolvedValueOnce(500000n) // balanceOf
+            .mockResolvedValueOnce(18) // decimals
+            .mockResolvedValueOnce('WETH'); // symbol
+        const result = await (0, getTokenBalance_js_1.getTokenBalance)(mockClient, {
+            account,
+            token,
+        });
+        (0, vitest_1.expect)(result).toEqual({
+            token: token.address,
+            decimals: 18,
+            symbol: 'WETH',
+            value: 500000n,
+            formatted: '0.0000000000005',
+        });
+    });
+    (0, vitest_1.it)('should get native token balance', async () => {
+        const account = { address: '0x123' };
+        mockClient.getBalance.mockResolvedValueOnce(2000000000000000000n); // 2 ETH
+        const result = await (0, getTokenBalance_js_1.getTokenBalance)(mockClient, {
+            account,
+        });
+        (0, vitest_1.expect)(mockClient.getBalance).toHaveBeenCalledWith({
+            address: account.address,
+        });
+        (0, vitest_1.expect)(result).toEqual({
+            token: undefined,
+            decimals: index_js_1.ETH_SEPOLIA.nativeCurrency.decimals,
+            symbol: index_js_1.ETH_SEPOLIA.nativeCurrency.symbol,
+            value: 2000000000000000000n,
+            formatted: '2',
+        });
+    });
+    (0, vitest_1.it)('should use account chain when provided', async () => {
+        const account = { address: '0x123', chain: index_js_1.MATIC_AMOY };
+        mockClient.getBalance.mockResolvedValueOnce(1000000000000000000n); // 1 MATIC
+        const result = await (0, getTokenBalance_js_1.getTokenBalance)(mockClient, {
+            account,
+        });
+        (0, vitest_1.expect)(result).toEqual({
+            token: undefined,
+            decimals: index_js_1.MATIC_AMOY.nativeCurrency.decimals,
+            symbol: index_js_1.MATIC_AMOY.nativeCurrency.symbol,
+            value: 1000000000000000000n,
+            formatted: '1',
+        });
+    });
+    (0, vitest_1.it)('should use explicit chain parameter', async () => {
+        const account = { address: '0x123' };
+        mockClient.getBalance.mockResolvedValueOnce(1500000000000000000n); // 1.5 MATIC
+        const result = await (0, getTokenBalance_js_1.getTokenBalance)(mockClient, {
+            account,
+            chain: index_js_1.MATIC_AMOY,
+        });
+        (0, vitest_1.expect)(result).toEqual({
+            token: undefined,
+            decimals: index_js_1.MATIC_AMOY.nativeCurrency.decimals,
+            symbol: index_js_1.MATIC_AMOY.nativeCurrency.symbol,
+            value: 1500000000000000000n,
+            formatted: '1.5',
+        });
+    });
+    (0, vitest_1.it)('should handle string token address', async () => {
+        const account = { address: '0x123' };
+        const token = '0xdef';
+        mockClient.readContract
+            .mockResolvedValueOnce(750000n) // balanceOf
+            .mockResolvedValueOnce(8) // decimals
+            .mockResolvedValueOnce('DAI'); // symbol
+        const result = await (0, getTokenBalance_js_1.getTokenBalance)(mockClient, {
+            account,
+            token,
+        });
+        (0, vitest_1.expect)(result.token).toBe(token);
+        (0, vitest_1.expect)(result.decimals).toBe(8);
+        (0, vitest_1.expect)(result.symbol).toBe('DAI');
+        (0, vitest_1.expect)(result.value).toBe(750000n);
+        (0, vitest_1.expect)(result.formatted).toBe('0.0075');
+    });
+    (0, vitest_1.it)('should handle zero balances', async () => {
+        const account = { address: '0x123' };
+        mockClient.getBalance.mockResolvedValueOnce(0n);
+        const result = await (0, getTokenBalance_js_1.getTokenBalance)(mockClient, {
+            account,
+        });
+        (0, vitest_1.expect)(result.value).toBe(0n);
+        (0, vitest_1.expect)(result.formatted).toBe('0');
+    });
+    (0, vitest_1.it)('should handle contract call failures', async () => {
+        const account = { address: '0x123' };
+        const tokenAddress = '0xabc';
+        mockClient.readContract.mockRejectedValueOnce(new Error('Contract call failed'));
+        await (0, vitest_1.expect)((0, getTokenBalance_js_1.getTokenBalance)(mockClient, {
+            account,
+            token: tokenAddress,
+        })).rejects.toThrow('Contract call failed');
+    });
+});

package/dist/cjs/actions/index.d.ts

@@ -1,11 +1,14 @@
 import { generateTransferLink } from './generateTransferLink.js';
 import { getSwapExactInQuote } from './getSwapExactInQuote.js';
+import { getTokenBalance } from './getTokenBalance.js';
 import { swapExactIn } from './swapExactIn.js';
 import { sweep } from './sweep.js';
 import { transfer } from './transfer.js';
-import type { Account, Chain } from '../types.js';
+import type { Chain } from '../chains.js';
+import type { Account } from '../types.js';
 import type { GenerateTransferLinkParams } from './generateTransferLink.js';
 import type { GetSwapExactInQuoteParams } from './getSwapExactInQuote.js';
+import type { GetTokenBalanceInput } from './getTokenBalance.js';
 import type { SwapExactInParams } from './swapExactIn.js';
 import type { SweepParams } from './sweep.js';
 import type { TransferParams } from './transfer.js';
@@ -14,6 +17,7 @@ export * from './transfer.js';
 export * from './sweep.js';
 export * from './generateTransferLink.js';
 export * from './getSwapExactInQuote.js';
+export * from './getTokenBalance.js';
 export * from './swapExactIn.js';
 export { FEE_LEVEL_WITH_PREPARE } from '../providers/circle-wallets/actions/index.js';
 export type Actions<transport extends Transport = Transport, chain extends ViemChain | undefined = Chain | undefined, account extends Account | undefined = Account | undefined> = {
@@ -29,6 +33,8 @@ export type Actions<transport extends Transport = Transport, chain extends ViemC
     getSwapQuote: (params: Parameters<typeof getSwapExactInQuote>[1]) => ReturnType<typeof getSwapExactInQuote>;
     /** See {@link getSwapExactInQuote} action */
     getSwapExactInQuote: (params: Parameters<typeof getSwapExactInQuote>[1]) => ReturnType<typeof getSwapExactInQuote>;
+    /** See {@link getTokenBalance} action */
+    getTokenBalance: (params: Parameters<typeof getTokenBalance>[1]) => ReturnType<typeof getTokenBalance>;
     /** See {@link swapExactIn} action */
     swap: (params: Parameters<typeof swapExactIn>[1]) => ReturnType<typeof swapExactIn>;
     /** See {@link swapExactIn} action */
@@ -53,6 +59,8 @@ export declare function actions(client: Client): {
     getSwapQuote: (params: GetSwapExactInQuoteParams) => Promise<import("./getSwapExactInQuote.js").GetSwapExactInQuoteReturnType>;
     /** See {@link getSwapExactInQuote} action */
     getSwapExactInQuote: (params: GetSwapExactInQuoteParams) => Promise<import("./getSwapExactInQuote.js").GetSwapExactInQuoteReturnType>;
+    /** See {@link getTokenBalance} action */
+    getTokenBalance: (params: GetTokenBalanceInput) => Promise<import("./getTokenBalance.js").GetTokenBalanceReturnType>;
     /** See {@link swapExactIn} action */
     swap: (params: SwapExactInParams) => Promise<`0x${string}`>;
     /** See {@link swapExactIn} action */

package/dist/cjs/actions/index.js

@@ -28,6 +28,7 @@ exports.FEE_LEVEL_WITH_PREPARE = void 0;
 exports.actions = actions;
 const generateTransferLink_js_1 = require("./generateTransferLink.js");
 const getSwapExactInQuote_js_1 = require("./getSwapExactInQuote.js");
+const getTokenBalance_js_1 = require("./getTokenBalance.js");
 const swapExactIn_js_1 = require("./swapExactIn.js");
 const sweep_js_1 = require("./sweep.js");
 const transfer_js_1 = require("./transfer.js");
@@ -35,6 +36,7 @@ __exportStar(require("./transfer.js"), exports);
 __exportStar(require("./sweep.js"), exports);
 __exportStar(require("./generateTransferLink.js"), exports);
 __exportStar(require("./getSwapExactInQuote.js"), exports);
+__exportStar(require("./getTokenBalance.js"), exports);
 __exportStar(require("./swapExactIn.js"), exports);
 var index_js_1 = require("../providers/circle-wallets/actions/index.js");
 Object.defineProperty(exports, "FEE_LEVEL_WITH_PREPARE", { enumerable: true, get: function () { return index_js_1.FEE_LEVEL_WITH_PREPARE; } });
@@ -45,23 +47,24 @@ Object.defineProperty(exports, "FEE_LEVEL_WITH_PREPARE", { enumerable: true, get
  * @returns An object containing the bound action functions.
  */
 function actions(client) {
-    const _client = client;
     return {
         /** See {@link sweep} action */
-        sweep: (params) => (0, sweep_js_1.sweep)(_client, params),
+        sweep: (params) => (0, sweep_js_1.sweep)(client, params),
         /** See {@link sweep} action */
-        drainFrom: (params) => (0, sweep_js_1.sweep)(_client, params),
+        drainFrom: (params) => (0, sweep_js_1.sweep)(client, params),
         /** See {@link transfer} action */
-        transfer: (params) => (0, transfer_js_1.transfer)(_client, params),
+        transfer: (params) => (0, transfer_js_1.transfer)(client, params),
         /** See {@link generateTransferLink} action */
-        generateTransferLink: (params) => (0, generateTransferLink_js_1.generateTransferLink)(_client, params),
+        generateTransferLink: (params) => (0, generateTransferLink_js_1.generateTransferLink)(client, params),
         /** See {@link getSwapExactInQuote} action */
-        getSwapQuote: (params) => (0, getSwapExactInQuote_js_1.getSwapExactInQuote)(_client, params),
+        getSwapQuote: (params) => (0, getSwapExactInQuote_js_1.getSwapExactInQuote)(client, params),
         /** See {@link getSwapExactInQuote} action */
-        getSwapExactInQuote: (params) => (0, getSwapExactInQuote_js_1.getSwapExactInQuote)(_client, params),
+        getSwapExactInQuote: (params) => (0, getSwapExactInQuote_js_1.getSwapExactInQuote)(client, params),
+        /** See {@link getTokenBalance} action */
+        getTokenBalance: (params) => (0, getTokenBalance_js_1.getTokenBalance)(client, params),
         /** See {@link swapExactIn} action */
-        swap: (params) => (0, swapExactIn_js_1.swapExactIn)(_client, params),
+        swap: (params) => (0, swapExactIn_js_1.swapExactIn)(client, params),
         /** See {@link swapExactIn} action */
-        swapExactIn: (params) => (0, swapExactIn_js_1.swapExactIn)(_client, params),
+        swapExactIn: (params) => (0, swapExactIn_js_1.swapExactIn)(client, params),
     };
 }

package/dist/cjs/actions/index.test.js

@@ -36,6 +36,8 @@ vitest_1.describe.concurrent('actions', () => {
             swap: vitest_1.expect.any(Function),
             // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
             swapExactIn: vitest_1.expect.any(Function),
+            // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+            getTokenBalance: vitest_1.expect.any(Function),
         });
     });
 });

package/dist/cjs/actions/swapExactIn.d.ts

@@ -1,6 +1,6 @@
 import type { Chain } from '../providers/uniswap/index.js';
-import type { Account, Address, BaseUnits, ControlFlowOptions, ExtendedViemClient, FeeValuesEIP1559, NativeUnits } from '../types.js';
-import type { Abi } from 'viem';
+import type { Account, Address, BaseUnits, ControlFlowOptions, FeeValuesEIP1559, NativeUnits } from '../types.js';
+import type { Abi, Client } from 'viem';
 /**
  * Parameters for the {@link swapExactIn} function.
  */
@@ -36,4 +36,4 @@ export type SwapExactInParamsReturnType = Address;
  * @param params - Parameters for the operation
  * @returns The address of the transaction that was executed.
  */
-export declare function swapExactIn(client: ExtendedViemClient, params: SwapExactInParams & ControlFlowOptions): Promise<SwapExactInParamsReturnType>;
+export declare function swapExactIn(client: Client, params: SwapExactInParams & ControlFlowOptions): Promise<SwapExactInParamsReturnType>;

package/dist/cjs/actions/sweep.d.ts

@@ -1,5 +1,6 @@
 import type { TransferReturnType, TransferParams } from './transfer.js';
-import type { ControlFlowOptions, ExtendedViemClient } from '../types.js';
+import type { ControlFlowOptions } from '../types.js';
+import type { Client } from 'viem';
 /**
  * Parameters for sweeping tokens from accounts
  */
@@ -13,4 +14,4 @@ export type SweepReturnType = TransferReturnType;
  * @param params - Parameters for the sweep operation
  * @returns An array of transfer results.
  */
-export declare function sweep(client: ExtendedViemClient, params: SweepParams & ControlFlowOptions): Promise<SweepReturnType>;
+export declare function sweep(client: Client, params: SweepParams & ControlFlowOptions): Promise<SweepReturnType>;

package/dist/cjs/actions/sweep.js

@@ -12,7 +12,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.sweep = sweep;
 const utils_1 = require("viem/utils");
-const getTokenBalances_js_1 = require("../providers/circle-wallets/actions/getTokenBalances.js");
+const getTokenBalance_js_1 = require("./getTokenBalance.js");
 const transfer_js_1 = require("./transfer.js");
 /**
  * Sweeps the specified token balance to the destination wallet.
@@ -23,8 +23,8 @@ const transfer_js_1 = require("./transfer.js");
  * @returns An array of transfer results.
  */
 async function sweep(client, params) {
-    const getTokenBalances = (0, utils_1.getAction)(client, getTokenBalances_js_1.getTokenBalances, 'getTokenBalances');
-    const { token = null } = params;
+    const getTokenBalance = (0, utils_1.getAction)(client, getTokenBalance_js_1.getTokenBalance, 'getTokenBalance');
+    const { token } = params;
     /**
      * Gets the current balance of tokens for an account to be transferred.
      *
@@ -32,8 +32,8 @@ async function sweep(client, params) {
      * @returns The current balance in base units, or 0n if no balance is found
      */
     async function zeroAmount(account) {
-        const balances = (await getTokenBalances({ account, token }));
-        return balances.at(0)?.value ?? 0n;
+        const balance = await getTokenBalance({ account, token: token ?? undefined });
+        return balance.value;
     }
     return await (0, transfer_js_1.transfer)(client, { ...params, amount: zeroAmount });
 }

package/dist/cjs/actions/sweep.test.js

@@ -12,26 +12,21 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const vitest_1 = require("vitest");
 const index_js_1 = require("../chains/index.js");
+const getTokenBalance_js_1 = require("./getTokenBalance.js");
 const sweep_js_1 = require("./sweep.js");
 const transfer_js_1 = require("./transfer.js");
 vitest_1.vi.mock('./transfer.js', () => ({
     transfer: vitest_1.vi.fn(),
 }));
+vitest_1.vi.mock('./getTokenBalance.js', () => ({
+    getTokenBalance: vitest_1.vi.fn(),
+}));
 (0, vitest_1.describe)('sweep()', () => {
     (0, vitest_1.beforeEach)(() => {
         vitest_1.vi.restoreAllMocks();
     });
     const client = {
         chain: index_js_1.ETH_SEPOLIA,
-        getTokenBalances: vitest_1.vi.fn(() => [
-            {
-                token: '0x1c7d4b196cb0c7b01d743fbc6116a902379c7238',
-                decimals: 6,
-                symbol: 'USDC',
-                value: 1237999924n,
-                formatted: '1237.999924',
-            },
-        ]),
     };
     (0, vitest_1.it)('should call transfer() with the full amount of account', async () => {
         const params = {
@@ -40,6 +35,14 @@ vitest_1.vi.mock('./transfer.js', () => ({
             token: index_js_1.ETH_SEPOLIA.contracts.USDC,
         };
         const expectedResult = [{ hash: '0xtx-hash' }];
+        // Mock getTokenBalance to return a specific balance
+        vitest_1.vi.mocked(getTokenBalance_js_1.getTokenBalance).mockResolvedValue({
+            token: '0x1c7d4b196cb0c7b01d743fbc6116a902379c7238',
+            decimals: 6,
+            symbol: 'USDC',
+            value: 1237999924n,
+            formatted: '1237.999924',
+        });
         vitest_1.vi.mocked(transfer_js_1.transfer).mockImplementation(async (_, { amount, from }) => {
             if (typeof amount === 'function') {
                 (0, vitest_1.expect)(await amount(from)).toBe(1237999924n);
@@ -53,6 +56,6 @@ vitest_1.vi.mock('./transfer.js', () => ({
             to: { address: '0x321', type: 'json-rpc' },
             token: index_js_1.ETH_SEPOLIA.contracts.USDC,
         }));
-        (0, vitest_1.expect)(client.getTokenBalances).toHaveBeenCalled();
+        (0, vitest_1.expect)(getTokenBalance_js_1.getTokenBalance).toHaveBeenCalled();
     });
 });

package/dist/cjs/actions/transfer.d.ts

@@ -1,5 +1,5 @@
 import type { Chain } from '../providers/circle-wallets/index.js';
-import type { Account, Address, Contract, ControlFlowOptions, NativeUnits, BaseUnits, FeeValuesEIP1559, ExtendedViemClient } from '../types.js';
+import type { Account, Address, Contract, ControlFlowOptions, NativeUnits, BaseUnits, FeeValuesEIP1559 } from '../types.js';
 import type { Abi, Client } from 'viem';
 /**
  * Parameters for the {@link transfer} function.
@@ -121,4 +121,4 @@ export type EstimateFeesPerGasParams = {
  * })
  * ```
  */
-export declare function transfer(client: ExtendedViemClient, params: TransferParams & ControlFlowOptions): Promise<TransferReturnType>;
+export declare function transfer(client: Client, params: TransferParams & ControlFlowOptions): Promise<TransferReturnType>;