@shapeshiftoss/unchained-client 10.14.11 → 10.16.0

package/dist/cjs/tron/__tests__/api.test.js

@@ -0,0 +1,425 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const api_1 = require("../api");
+const successResponse = {
+    result: { result: true },
+    energy_used: 64285,
+    energy_penalty: 49635,
+    constant_result: ['0000000000000000000000000000000000000000000000000000000000000000'],
+    transaction: { ret: [{}] },
+};
+// A reverted simulation still reports result.result: true - the failure is only visible on
+// transaction.ret and the message
+const revertResponse = {
+    result: { result: true, message: 'REVERT opcode executed' },
+    energy_used: 7896,
+    energy_penalty: 6079,
+    constant_result: [''],
+    transaction: { ret: [{ ret: 'FAILED' }] },
+};
+(0, vitest_1.describe)('TronApi', () => {
+    const api = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+    (0, vitest_1.beforeEach)(() => {
+        vitest_1.vi.spyOn(api, 'getChainPrices').mockResolvedValue({
+            bandwidthPrice: 1000,
+            energyPrice: 100,
+            memoFee: 1_000_000,
+        });
+        vitest_1.vi.spyOn(api, 'getContractEnergyShare').mockResolvedValue({
+            callerPercent: 100,
+            originEnergyLimit: 0,
+            originEnergyAvailable: 0,
+        });
+    });
+    (0, vitest_1.afterEach)(() => {
+        vitest_1.vi.restoreAllMocks();
+        vitest_1.vi.unstubAllGlobals();
+    });
+    (0, vitest_1.describe)('estimateContractCallFee', () => {
+        const params = {
+            contractAddress: 'TAfbit1ENsRmtZbPQfYU3srURpfYuWYS7K',
+            from: 'TT2T17KZhoDu47i2E4FWxfG79zdkEWkU9N',
+            data: '0x2213bc0b',
+        };
+        (0, vitest_1.it)('prices a successful simulation at energy_used * energyPrice', async () => {
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockResolvedValue({ ok: true, json: () => successResponse }));
+            (0, vitest_1.expect)(await api.estimateContractCallFee(params)).toBe('6428500');
+        });
+        (0, vitest_1.it)('prices the caller in full when the share lookup fails', async () => {
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockResolvedValue({ ok: true, json: () => successResponse }));
+            vitest_1.vi.mocked(api.getContractEnergyShare).mockRejectedValue(new Error('429'));
+            (0, vitest_1.expect)(await api.estimateContractCallFee(params)).toBe('6428500');
+        });
+        (0, vitest_1.it)('fails instead when the caller requires the share', async () => {
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockResolvedValue({ ok: true, json: () => successResponse }));
+            vitest_1.vi.mocked(api.getContractEnergyShare).mockRejectedValue(new Error('429'));
+            await (0, vitest_1.expect)(api.estimateContractCallFee({ ...params, requireEnergyShare: true })).rejects.toThrow('429');
+        });
+        (0, vitest_1.it)('bills the caller only their share when the deployer covers the rest', async () => {
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockResolvedValue({ ok: true, json: () => successResponse }));
+            vitest_1.vi.mocked(api.getContractEnergyShare).mockResolvedValue({
+                callerPercent: 5,
+                originEnergyLimit: 10_000_000,
+                originEnergyAvailable: 5_000_000,
+            });
+            // 64285 - floor(64285 * 0.95) = 3215
+            (0, vitest_1.expect)(await api.estimateContractCallFee(params)).toBe('321500');
+        });
+        (0, vitest_1.it)('throws on a reverted simulation rather than trusting the partial energy', async () => {
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockResolvedValue({ ok: true, json: () => revertResponse }));
+            await (0, vitest_1.expect)(api.estimateContractCallFee(params)).rejects.toThrow('REVERT opcode executed');
+        });
+        (0, vitest_1.it)('reports a failed request as such rather than as a revert', async () => {
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockResolvedValue({ ok: false, status: 429, json: () => ({}) }));
+            await (0, vitest_1.expect)(api.estimateContractCallFee(params)).rejects.toThrow('request failed: 429');
+        });
+    });
+    (0, vitest_1.describe)('getCallerEnergy', () => {
+        (0, vitest_1.it)('charges the caller in full when the contract sets no deployer share', () => {
+            const share = {
+                callerPercent: 100,
+                originEnergyLimit: 0,
+                originEnergyAvailable: 0,
+            };
+            (0, vitest_1.expect)((0, api_1.getCallerEnergy)(901_682, share)).toBe(901_682);
+        });
+        (0, vitest_1.it)('charges the caller their percent when the deployer has the energy staked', () => {
+            const share = {
+                callerPercent: 5,
+                originEnergyLimit: 10_000_000,
+                originEnergyAvailable: 5_000_000,
+            };
+            (0, vitest_1.expect)((0, api_1.getCallerEnergy)(901_682, share)).toBe(45_085);
+        });
+        (0, vitest_1.it)('charges the caller in full when the deployer has nothing staked', () => {
+            const share = {
+                callerPercent: 30,
+                originEnergyLimit: 10_000_000,
+                originEnergyAvailable: 0,
+            };
+            (0, vitest_1.expect)((0, api_1.getCallerEnergy)(130_285, share)).toBe(130_285);
+        });
+        (0, vitest_1.it)('caps the deployer share at their per-call limit', () => {
+            const share = {
+                callerPercent: 0,
+                originEnergyLimit: 100_000,
+                originEnergyAvailable: 5_000_000,
+            };
+            (0, vitest_1.expect)((0, api_1.getCallerEnergy)(300_000, share)).toBe(200_000);
+        });
+        (0, vitest_1.it)('caps the deployer share at what they have left', () => {
+            const share = {
+                callerPercent: 0,
+                originEnergyLimit: 10_000_000,
+                originEnergyAvailable: 40_000,
+            };
+            (0, vitest_1.expect)((0, api_1.getCallerEnergy)(300_000, share)).toBe(260_000);
+        });
+    });
+    (0, vitest_1.describe)('getContractEnergyShare', () => {
+        const contractResponse = {
+            consume_user_resource_percent: 5,
+            origin_energy_limit: 10_000_000,
+            origin_address: 'TDeployer',
+        };
+        const resourceResponse = { EnergyLimit: 6_000_000, EnergyUsed: 1_000_000 };
+        const respond = (...bodies) => {
+            const fetchMock = vitest_1.vi.fn();
+            bodies.forEach(body => fetchMock.mockResolvedValueOnce({ ok: true, json: () => body }));
+            vitest_1.vi.stubGlobal('fetch', fetchMock);
+            return fetchMock;
+        };
+        // the contract record is cached per instance, so each case gets its own
+        const freshApi = () => new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+        (0, vitest_1.it)('reads the caller percent, per-call limit and the unspent deployer energy', async () => {
+            respond(contractResponse, resourceResponse);
+            (0, vitest_1.expect)(await freshApi().getContractEnergyShare('TRouter')).toEqual({
+                callerPercent: 5,
+                originEnergyLimit: 10_000_000,
+                originEnergyAvailable: 5_000_000,
+            });
+        });
+        (0, vitest_1.it)('reads an omitted percent as the deployer paying everything', async () => {
+            respond({ origin_energy_limit: 10_000_000, origin_address: 'TDeployer' }, resourceResponse);
+            (0, vitest_1.expect)(await freshApi().getContractEnergyShare('TRouter')).toMatchObject({ callerPercent: 0 });
+        });
+        (0, vitest_1.it)('reads an omitted per-call limit as the creator default', async () => {
+            respond({ consume_user_resource_percent: 5, origin_address: 'TDeployer' }, resourceResponse);
+            (0, vitest_1.expect)(await freshApi().getContractEnergyShare('TRouter')).toMatchObject({
+                originEnergyLimit: 10_000_000,
+            });
+        });
+        (0, vitest_1.it)('reads a plain address as a call the caller pays in full without a resource lookup', async () => {
+            const fetchMock = respond({});
+            (0, vitest_1.expect)(await freshApi().getContractEnergyShare('TPlain')).toEqual({
+                callerPercent: 100,
+                originEnergyLimit: 0,
+                originEnergyAvailable: 0,
+            });
+            (0, vitest_1.expect)(fetchMock).toHaveBeenCalledTimes(1);
+        });
+        (0, vitest_1.it)('re-reads an empty body rather than caching it as a plain address', async () => {
+            const fetchMock = respond({}, contractResponse, resourceResponse);
+            const api = freshApi();
+            await api.getContractEnergyShare('TRouter');
+            (0, vitest_1.expect)(await api.getContractEnergyShare('TRouter')).toMatchObject({ callerPercent: 5 });
+            (0, vitest_1.expect)(fetchMock).toHaveBeenCalledTimes(3);
+        });
+        (0, vitest_1.it)('rejects a failed lookup rather than guessing the split', async () => {
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockResolvedValue({ ok: false, status: 429 }));
+            await (0, vitest_1.expect)(freshApi().getContractEnergyShare('TRouter')).rejects.toThrow('429');
+        });
+        (0, vitest_1.it)('does not cache an error body the node returns with a 200', async () => {
+            const fetchMock = respond({ Error: 'class org.tron.core.exception' }, contractResponse, resourceResponse);
+            const api = freshApi();
+            await (0, vitest_1.expect)(api.getContractEnergyShare('TRouter')).rejects.toThrow('getcontract failed');
+            (0, vitest_1.expect)(await api.getContractEnergyShare('TRouter')).toMatchObject({ callerPercent: 5 });
+            (0, vitest_1.expect)(fetchMock).toHaveBeenCalledTimes(3);
+        });
+        (0, vitest_1.it)('reads the contract once and the deployer energy once per refresh window', async () => {
+            vitest_1.vi.useFakeTimers();
+            const fetchMock = respond(contractResponse, resourceResponse, resourceResponse);
+            const api = freshApi();
+            await api.getContractEnergyShare('TRouter');
+            await api.getContractEnergyShare('TRouter');
+            (0, vitest_1.expect)(fetchMock).toHaveBeenCalledTimes(2);
+            vitest_1.vi.advanceTimersByTime(16_000);
+            await api.getContractEnergyShare('TRouter');
+            (0, vitest_1.expect)(fetchMock).toHaveBeenCalledTimes(3);
+            const contractReads = fetchMock.mock.calls.filter(([url]) => String(url).endsWith('/wallet/getcontract'));
+            (0, vitest_1.expect)(contractReads).toHaveLength(1);
+            vitest_1.vi.useRealTimers();
+        });
+        (0, vitest_1.it)('re-reads the contract after a minute, since its split can change', async () => {
+            vitest_1.vi.useFakeTimers();
+            const fetchMock = respond(contractResponse, resourceResponse, contractResponse, resourceResponse);
+            const api = freshApi();
+            await api.getContractEnergyShare('TRouter');
+            vitest_1.vi.advanceTimersByTime(61_000);
+            await api.getContractEnergyShare('TRouter');
+            const contractReads = fetchMock.mock.calls.filter(([url]) => String(url).endsWith('/wallet/getcontract'));
+            (0, vitest_1.expect)(contractReads).toHaveLength(2);
+            vitest_1.vi.useRealTimers();
+        });
+    });
+    (0, vitest_1.describe)('getChainPrices', () => {
+        (0, vitest_1.it)('reads the energy, bandwidth and memo parameters', async () => {
+            vitest_1.vi.restoreAllMocks();
+            const tronWeb = api.getTronWeb();
+            vitest_1.vi.spyOn(tronWeb.trx, 'getChainParameters').mockResolvedValue([
+                { key: 'getTransactionFee', value: 1000 },
+                { key: 'getEnergyFee', value: 100 },
+                { key: 'getMemoFee', value: 1_000_000 },
+            ]);
+            (0, vitest_1.expect)(await api.getChainPrices()).toEqual({
+                bandwidthPrice: 1000,
+                energyPrice: 100,
+                memoFee: 1_000_000,
+            });
+        });
+        (0, vitest_1.it)('throws rather than defaulting when the node is unreachable', async () => {
+            vitest_1.vi.restoreAllMocks();
+            const tronWeb = api.getTronWeb();
+            vitest_1.vi.spyOn(tronWeb.trx, 'getChainParameters').mockRejectedValue(new Error('ECONNREFUSED'));
+            await (0, vitest_1.expect)(api.getChainPrices()).rejects.toThrow('ECONNREFUSED');
+        });
+    });
+    (0, vitest_1.describe)('getTrc20Balance', () => {
+        (0, vitest_1.it)('reads balanceOf(address) with a single constant call', async () => {
+            const tronWeb = api.getTronWeb();
+            const trigger = vitest_1.vi
+                .spyOn(tronWeb.transactionBuilder, 'triggerConstantContract')
+                .mockResolvedValue({
+                result: { result: true },
+                constant_result: ['00000000000000000000000000000000000000000000000000000000028e6fb0'],
+            });
+            const actual = await api.getTrc20Balance({
+                contractAddress: 'TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t',
+                address: 'TE6oHVdTbcp1Q9XBYx5VzjWbZEg3t3Jrnc',
+            });
+            (0, vitest_1.expect)(actual).toBe('42889136');
+            (0, vitest_1.expect)(trigger).toHaveBeenCalledWith('TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t', 'balanceOf(address)', {}, [{ type: 'address', value: 'TE6oHVdTbcp1Q9XBYx5VzjWbZEg3t3Jrnc' }], 'TE6oHVdTbcp1Q9XBYx5VzjWbZEg3t3Jrnc');
+        });
+    });
+    (0, vitest_1.describe)('getTrc20Decimals', () => {
+        const USDT = 'TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t';
+        const eighteen = '0000000000000000000000000000000000000000000000000000000000000012';
+        (0, vitest_1.it)('reads decimals() with a single constant call', async () => {
+            const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+            const tronWeb = freshApi.getTronWeb();
+            const trigger = vitest_1.vi
+                .spyOn(tronWeb.transactionBuilder, 'triggerConstantContract')
+                .mockResolvedValue({ result: { result: true }, constant_result: [eighteen] });
+            (0, vitest_1.expect)(await freshApi.getTrc20Decimals({ contractAddress: USDT })).toBe(18);
+            (0, vitest_1.expect)(trigger).toHaveBeenCalledWith(USDT, 'decimals()', {}, [], USDT);
+        });
+        (0, vitest_1.it)('reads a reverted decimals() as unknown', async () => {
+            const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+            const tronWeb = freshApi.getTronWeb();
+            const trigger = vitest_1.vi
+                .spyOn(tronWeb.transactionBuilder, 'triggerConstantContract')
+                .mockResolvedValueOnce({
+                result: { result: true, message: 'REVERT opcode executed' },
+                constant_result: ['08c379a0' + '0'.repeat(56)],
+                transaction: { ret: [{ ret: 'FAILED' }] },
+            })
+                .mockResolvedValue({ result: { result: true }, constant_result: [eighteen] });
+            (0, vitest_1.expect)(await freshApi.getTrc20Decimals({ contractAddress: USDT })).toBeUndefined();
+            (0, vitest_1.expect)(await freshApi.getTrc20Decimals({ contractAddress: USDT })).toBe(18);
+            (0, vitest_1.expect)(trigger).toHaveBeenCalledTimes(2);
+        });
+        (0, vitest_1.it)('reads a value outside a uint8 as unknown', async () => {
+            const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+            const tronWeb = freshApi.getTronWeb();
+            vitest_1.vi.spyOn(tronWeb.transactionBuilder, 'triggerConstantContract').mockResolvedValue({
+                result: { result: true },
+                constant_result: ['0'.repeat(61) + '100'],
+            });
+            (0, vitest_1.expect)(await freshApi.getTrc20Decimals({ contractAddress: USDT })).toBeUndefined();
+        });
+        (0, vitest_1.it)('gives up on a call that hangs', async () => {
+            vitest_1.vi.useFakeTimers();
+            try {
+                const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+                const tronWeb = freshApi.getTronWeb();
+                vitest_1.vi.spyOn(tronWeb.transactionBuilder, 'triggerConstantContract').mockReturnValue(new Promise(() => undefined));
+                const pending = freshApi.getTrc20Decimals({ contractAddress: USDT });
+                await vitest_1.vi.runAllTimersAsync();
+                (0, vitest_1.expect)(await pending).toBeUndefined();
+            }
+            finally {
+                vitest_1.vi.useRealTimers();
+            }
+        });
+        (0, vitest_1.it)('reads a failed call as unknown', async () => {
+            const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+            const tronWeb = freshApi.getTronWeb();
+            const trigger = vitest_1.vi
+                .spyOn(tronWeb.transactionBuilder, 'triggerConstantContract')
+                .mockRejectedValueOnce(new Error('429'))
+                .mockResolvedValue({ result: { result: true }, constant_result: [eighteen] });
+            (0, vitest_1.expect)(await freshApi.getTrc20Decimals({ contractAddress: USDT })).toBeUndefined();
+            (0, vitest_1.expect)(await freshApi.getTrc20Decimals({ contractAddress: USDT })).toBe(18);
+            (0, vitest_1.expect)(trigger).toHaveBeenCalledTimes(2);
+        });
+    });
+    (0, vitest_1.describe)('getTrc10Precision', () => {
+        const respond = (body) => {
+            const fetchMock = vitest_1.vi.fn().mockResolvedValue({ ok: true, json: () => body });
+            vitest_1.vi.stubGlobal('fetch', fetchMock);
+            return fetchMock;
+        };
+        (0, vitest_1.it)('reads the precision of an issued token', async () => {
+            const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+            const fetchMock = respond({ success: true, data: [{ id: '1002000', precision: 6 }] });
+            (0, vitest_1.expect)(await freshApi.getTrc10Precision({ id: '1002000' })).toBe(6);
+            (0, vitest_1.expect)(fetchMock).toHaveBeenCalledWith('https://tron.example/v1/assets/1002000', vitest_1.expect.objectContaining({ headers: vitest_1.expect.anything() }));
+        });
+        (0, vitest_1.it)('reads a zero precision as issued', async () => {
+            const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+            respond({ success: true, data: [{ id: '1000001', precision: 0 }] });
+            (0, vitest_1.expect)(await freshApi.getTrc10Precision({ id: '1000001' })).toBe(0);
+        });
+        (0, vitest_1.it)('reads an unknown id as unknown', async () => {
+            const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+            respond({ success: false, error: 'A valid account address or asset id is required.' });
+            (0, vitest_1.expect)(await freshApi.getTrc10Precision({ id: '9999999' })).toBeUndefined();
+        });
+        (0, vitest_1.it)('reads an empty listing as unknown', async () => {
+            const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+            respond({ success: true, data: [] });
+            (0, vitest_1.expect)(await freshApi.getTrc10Precision({ id: '1009999' })).toBeUndefined();
+        });
+        (0, vitest_1.it)('gives up on a request that hangs', async () => {
+            vitest_1.vi.useFakeTimers();
+            try {
+                const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+                vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockReturnValue(new Promise(() => undefined)));
+                const pending = freshApi.getTrc10Precision({ id: '1002000' });
+                await vitest_1.vi.runAllTimersAsync();
+                (0, vitest_1.expect)(await pending).toBeUndefined();
+            }
+            finally {
+                vitest_1.vi.useRealTimers();
+            }
+        });
+        (0, vitest_1.it)('gives up on a response whose body stalls', async () => {
+            vitest_1.vi.useFakeTimers();
+            try {
+                const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+                const fetchMock = vitest_1.vi
+                    .fn()
+                    .mockResolvedValue({ ok: true, json: () => new Promise(() => undefined) });
+                vitest_1.vi.stubGlobal('fetch', fetchMock);
+                const pending = freshApi.getTrc10Precision({ id: '1002000' });
+                await vitest_1.vi.runAllTimersAsync();
+                (0, vitest_1.expect)(await pending).toBeUndefined();
+                (0, vitest_1.expect)(fetchMock).toHaveBeenCalledWith('https://tron.example/v1/assets/1002000', vitest_1.expect.objectContaining({ signal: vitest_1.expect.any(AbortSignal) }));
+            }
+            finally {
+                vitest_1.vi.useRealTimers();
+            }
+        });
+        (0, vitest_1.it)('reads a failed request as unknown', async () => {
+            const freshApi = new api_1.TronApi({ rpcUrl: 'https://tron.example' });
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockRejectedValue(new Error('429')));
+            (0, vitest_1.expect)(await freshApi.getTrc10Precision({ id: '1002000' })).toBeUndefined();
+        });
+    });
+    (0, vitest_1.describe)('getTrc20Allowance', () => {
+        (0, vitest_1.it)('reads allowance(owner, spender) with a single constant call', async () => {
+            const tronWeb = api.getTronWeb();
+            const trigger = vitest_1.vi
+                .spyOn(tronWeb.transactionBuilder, 'triggerConstantContract')
+                .mockResolvedValue({
+                result: { result: true },
+                constant_result: ['000000000000000000000000000000000000000000000000000000000000007b'],
+            });
+            const actual = await api.getTrc20Allowance({
+                contractAddress: 'TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t',
+                owner: 'TT2T17KZhoDu47i2E4FWxfG79zdkEWkU9N',
+                spender: 'TAfbit1ENsRmtZbPQfYU3srURpfYuWYS7K',
+            });
+            (0, vitest_1.expect)(actual).toBe('123');
+            (0, vitest_1.expect)(trigger).toHaveBeenCalledWith('TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t', 'allowance(address,address)', {}, [
+                { type: 'address', value: 'TT2T17KZhoDu47i2E4FWxfG79zdkEWkU9N' },
+                { type: 'address', value: 'TAfbit1ENsRmtZbPQfYU3srURpfYuWYS7K' },
+            ], 'TT2T17KZhoDu47i2E4FWxfG79zdkEWkU9N');
+        });
+    });
+    (0, vitest_1.describe)('isAccountActivated', () => {
+        (0, vitest_1.it)('is true for an account the node knows', async () => {
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockResolvedValue({ ok: true, json: () => ({ address: '41abc', balance: 1 }) }));
+            (0, vitest_1.expect)(await api.isAccountActivated('TT2T17KZhoDu47i2E4FWxfG79zdkEWkU9N')).toBe(true);
+        });
+        (0, vitest_1.it)('is false for a fresh address', async () => {
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockResolvedValue({ ok: true, json: () => ({}) }));
+            (0, vitest_1.expect)(await api.isAccountActivated('TT2T17KZhoDu47i2E4FWxfG79zdkEWkU9N')).toBe(false);
+        });
+        (0, vitest_1.it)('throws on an error response rather than reading it as not activated', async () => {
+            vitest_1.vi.stubGlobal('fetch', vitest_1.vi.fn().mockResolvedValue({ ok: false, status: 429, json: () => ({ Error: 'throttled' }) }));
+            await (0, vitest_1.expect)(api.isAccountActivated('TT2T17KZhoDu47i2E4FWxfG79zdkEWkU9N')).rejects.toThrow('429');
+        });
+    });
+    (0, vitest_1.describe)('estimateTrc20TransferFee', () => {
+        const params = {
+            contractAddress: 'TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t',
+            from: 'TT2T17KZhoDu47i2E4FWxfG79zdkEWkU9N',
+            to: 'TT2T17KZhoDu47i2E4FWxfG79zdkEWkU9N',
+            amount: '1',
+        };
+        const stubTriggerConstantContract = (response) => {
+            const tronWeb = api.getTronWeb();
+            vitest_1.vi.spyOn(tronWeb.transactionBuilder, 'triggerConstantContract').mockResolvedValue(response);
+        };
+        (0, vitest_1.it)('prices a successful simulation at energy_used * energyPrice', async () => {
+            stubTriggerConstantContract(successResponse);
+            (0, vitest_1.expect)(await api.estimateTrc20TransferFee(params)).toBe('6428500');
+        });
+        (0, vitest_1.it)('throws on a reverted simulation rather than trusting the partial energy', async () => {
+            stubTriggerConstantContract(revertResponse);
+            await (0, vitest_1.expect)(api.estimateTrc20TransferFee(params)).rejects.toThrow('REVERT opcode executed');
+        });
+    });
+});

package/dist/cjs/tron/api.d.ts

@@ -3,10 +3,18 @@ export interface TronApiConfig {
     rpcUrl: string;
     apiKey?: string;
 }
+export type TronContractEnergyShare = {
+    callerPercent: number;
+    originEnergyLimit: number;
+    originEnergyAvailable: number;
+};
+export declare const getCallerEnergy: (energyUsed: number, share: TronContractEnergyShare) => number;
 export declare class TronApi {
     private readonly rpcUrl;
     private readonly apiKey;
     private tronWeb;
+    private readonly contracts;
+    private readonly originEnergy;
     private requestQueue;
     private readonly minRequestInterval;
     constructor(config: TronApiConfig);
@@ -24,10 +32,22 @@ export declare class TronApi {
             balance: string;
         }[];
     }>;
-    getTRC20Balance(params: {
+    getTrc20Decimals(params: {
+        contractAddress: string;
+    }): Promise<number | undefined>;
+    getTrc10Precision(params: {
+        id: string;
+    }): Promise<number | undefined>;
+    getTrc20Balance(params: {
+        contractAddress: string;
         address: string;
+    }): Promise<string>;
+    getTrc20Allowance(params: {
         contractAddress: string;
+        owner: string;
+        spender: string;
     }): Promise<string>;
+    isAccountActivated(address: string): Promise<boolean>;
     getTxHistory(_params: {
         pubkey: string;
         pageSize?: number;
@@ -47,23 +67,30 @@ export declare class TronApi {
             hex: string;
         };
     }): Promise<string>;
-    private getChainPrices;
-    estimateFees(params: {
-        estimateFeesBody: {
-            serializedTx: string;
-        };
-    }): Promise<string>;
-    estimateTRC20TransferFee(params: {
+    getChainPrices(): Promise<{
+        bandwidthPrice: number;
+        energyPrice: number;
+        memoFee: number;
+    }>;
+    getContractEnergyShare(contractAddress: string): Promise<TronContractEnergyShare>;
+    private getContract;
+    private getPricingContext;
+    private getOriginEnergyAvailable;
+    private post;
+    estimateTrc20TransferFee(params: {
         contractAddress: string;
         from: string;
         to: string;
         amount: string;
+        requireEnergyShare?: boolean;
     }): Promise<string>;
-    getPriorityFees(): Promise<{
-        baseFee: string;
-        fast: string;
-        average: string;
-        slow: string;
-        estimatedBandwidth: string;
-    }>;
+    estimateContractCallFee(params: {
+        contractAddress: string;
+        from: string;
+        data: string;
+        callValue?: string;
+        requireEnergyShare?: boolean;
+    }): Promise<string>;
+    private isReverted;
+    private getSimulatedEnergy;
 }