@humanode-hyperlane/rebalancer 0.0.0-stage → 27.2.2

package/dist/e2e/harness/TestHelpers.js

@@ -0,0 +1,103 @@
+import { assert } from '@humanode-hyperlane/utils';
+import { ERC20Test__factory } from '@humanode-hyperlane/core';
+import { expect } from 'chai';
+import { MonitorEventType } from '../../interfaces/IMonitor.js';
+import { tryRelayMessage } from './TransferHelper.js';
+import { DOMAIN_IDS, TEST_CHAINS, } from '../fixtures/routes.js';
+export async function getFirstMonitorEvent(monitor) {
+    return new Promise((resolve, reject) => {
+        let settled = false;
+        async function finalize(cb, // eslint-disable-line @typescript-eslint/no-explicit-any -- accepts both resolve and reject
+        value, timer) {
+            if (settled)
+                return;
+            settled = true;
+            clearTimeout(timer);
+            cb(value);
+            try {
+                await monitor.stop();
+            }
+            catch {
+                // stop errors are non-fatal — the event/error has already been delivered
+            }
+        }
+        const timeout = setTimeout(() => {
+            void finalize(reject, new Error('Monitor event timeout'), timeout);
+        }, 60_000);
+        monitor.on(MonitorEventType.TokenInfo, (event) => {
+            void finalize(resolve, event, timeout);
+        });
+        monitor.on(MonitorEventType.Error, (error) => {
+            void finalize(reject, error, timeout);
+        });
+        void monitor.start();
+    });
+}
+export function chainFromDomain(domain) {
+    const found = Object.entries(DOMAIN_IDS).find(([, d]) => d === domain);
+    if (!found) {
+        throw new Error(`Unknown domain: ${domain}`);
+    }
+    return found[0];
+}
+export async function getRouterBalances(localProviders, addresses) {
+    const balances = {};
+    for (const chain of TEST_CHAINS) {
+        const provider = localProviders.get(chain);
+        assert(provider, `Missing provider for chain ${chain}`);
+        balances[chain] = await provider.getBalance(addresses.monitoredRoute[chain]);
+    }
+    return balances;
+}
+export async function getErc20RouterBalances(localProviders, addresses) {
+    const balances = {};
+    for (const chain of TEST_CHAINS) {
+        const provider = localProviders.get(chain);
+        assert(provider, `Missing provider for chain ${chain}`);
+        const token = ERC20Test__factory.connect(addresses.tokens[chain], provider);
+        balances[chain] = await token.balanceOf(addresses.monitoredRoute[chain]);
+    }
+    return balances;
+}
+/**
+ * Classify chains into deficit, surplus, and neutral based on a deposit action.
+ *
+ * For inventory deposits the execution direction is swapped: the action's
+ * destination is the surplus chain (where the router pays out on delivery).
+ */
+export function classifyChains(deficitChain, depositAction) {
+    const surplusChain = chainFromDomain(depositAction.destination);
+    const neutralChain = TEST_CHAINS.find((c) => c !== deficitChain && c !== surplusChain);
+    return { deficitChain, surplusChain, neutralChain };
+}
+export async function relayInProgressInventoryDeposits(context, localProviders, multiProvider, hyperlaneCore) {
+    const inProgressActions = await context.tracker.getInProgressActions();
+    const depositActions = inProgressActions.filter((a) => a.type === 'inventory_deposit' && a.txHash && a.messageId);
+    for (const action of depositActions) {
+        const origin = chainFromDomain(action.origin);
+        const destination = chainFromDomain(action.destination);
+        const provider = localProviders.get(origin);
+        assert(provider, `Missing provider for chain ${origin}`);
+        assert(action.txHash, `Missing txHash for action ${action.origin}->${action.destination}`);
+        const dispatchTx = await provider.getTransactionReceipt(action.txHash);
+        assert(action.messageId, `Missing messageId for action ${action.origin}->${action.destination}`);
+        const relayResult = await tryRelayMessage(multiProvider, hyperlaneCore, {
+            dispatchTx,
+            messageId: action.messageId,
+            origin,
+            destination,
+        });
+        expect(relayResult.success, `Inventory deposit relay should succeed: ${relayResult.error}`).to.be.true;
+    }
+    // Use provider.send to bypass ethers v5 _maxInternalBlockNumber cache
+    // which refuses to return lower block numbers after evm_revert.
+    const tags = {};
+    for (const chain of TEST_CHAINS) {
+        const p = localProviders.get(chain);
+        assert(p, `Missing provider for chain ${chain}`);
+        const hex = await p.send('eth_blockNumber', []);
+        tags[chain] = parseInt(hex, 16);
+    }
+    await context.tracker.syncRebalanceActions(tags);
+}
+//# sourceMappingURL=TestHelpers.js.map

package/dist/e2e/harness/TestHelpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestHelpers.js","sourceRoot":"","sources":["../../../src/e2e/harness/TestHelpers.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,MAAM,EAAE,MAAM,2BAA2B,CAAC;AACnD,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAI9B,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAGhE,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEtD,OAAO,EACL,UAAU,EAGV,WAAW,GACZ,MAAM,uBAAuB,CAAC;AAE/B,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,OAAgB;IAEhB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,IAAI,OAAO,GAAG,KAAK,CAAC;QAEpB,KAAK,UAAU,QAAQ,CACrB,EAAoB,EAAE,4FAA4F;QAClH,KAAc,EACd,KAAoC;YAEpC,IAAI,OAAO;gBAAE,OAAO;YACpB,OAAO,GAAG,IAAI,CAAC;YACf,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,EAAE,CAAC,KAAK,CAAC,CAAC;YACV,IAAI,CAAC;gBACH,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;YACvB,CAAC;YAAC,MAAM,CAAC;gBACP,yEAAyE;YAC3E,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9B,KAAK,QAAQ,CAAC,MAAM,EAAE,IAAI,KAAK,CAAC,uBAAuB,CAAC,EAAE,OAAO,CAAC,CAAC;QACrE,CAAC,EAAE,MAAM,CAAC,CAAC;QAEX,OAAO,CAAC,EAAE,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC,KAAmB,EAAE,EAAE;YAC7D,KAAK,QAAQ,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,EAAE,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC,KAAY,EAAE,EAAE;YAClD,KAAK,QAAQ,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,KAAK,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,MAAc;IAC5C,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,MAAM,CAAC,CAAC;IACvE,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,mBAAmB,MAAM,EAAE,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,cAAsD,EACtD,SAAkC;IAElC,MAAM,QAAQ,GAA8B,EAAE,CAAC;IAC/C,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QAChC,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,EAAE,8BAA8B,KAAK,EAAE,CAAC,CAAC;QACxD,QAAQ,CAAC,KAAK,CAAC,GAAG,MAAM,QAAQ,CAAC,UAAU,CACzC,SAAS,CAAC,cAAc,CAAC,KAAK,CAAC,CAChC,CAAC;IACJ,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,cAAsD,EACtD,SAA0C;IAE1C,MAAM,QAAQ,GAA8B,EAAE,CAAC;IAC/C,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QAChC,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,EAAE,8BAA8B,KAAK,EAAE,CAAC,CAAC;QACxD,MAAM,KAAK,GAAG,kBAAkB,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,QAAQ,CAAC,CAAC;QAC5E,QAAQ,CAAC,KAAK,CAAC,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3E,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAQD;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAC5B,YAAoB,EACpB,aAA8B;IAE9B,MAAM,YAAY,GAAG,eAAe,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;IAChE,MAAM,YAAY,GAAG,WAAW,CAAC,IAAI,CACnC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,YAAY,IAAI,CAAC,KAAK,YAAY,CAChD,CAAC;IACF,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC;AACtD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gCAAgC,CACpD,OAA8B,EAC9B,cAAsD,EACtD,aAA4B,EAC5B,aAA4B;IAE5B,MAAM,iBAAiB,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;IACvE,MAAM,cAAc,GAAG,iBAAiB,CAAC,MAAM,CAC7C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,mBAAmB,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,SAAS,CACjE,CAAC;IAEF,KAAK,MAAM,MAAM,IAAI,cAAc,EAAE,CAAC;QACpC,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC9C,MAAM,WAAW,GAAG,eAAe,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,CAAC,QAAQ,EAAE,8BAA8B,MAAM,EAAE,CAAC,CAAC;QACzD,MAAM,CACJ,MAAM,CAAC,MAAM,EACb,6BAA6B,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,WAAW,EAAE,CACpE,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAC,qBAAqB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEvE,MAAM,CACJ,MAAM,CAAC,SAAS,EAChB,gCAAgC,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,WAAW,EAAE,CACvE,CAAC;QACF,MAAM,WAAW,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,aAAa,EAAE;YACtE,UAAU;YACV,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,MAAM;YACN,WAAW;SACZ,CAAC,CAAC;QAEH,MAAM,CACJ,WAAW,CAAC,OAAO,EACnB,2CAA2C,WAAW,CAAC,KAAK,EAAE,CAC/D,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACf,CAAC;IAED,sEAAsE;IACtE,gEAAgE;IAChE,MAAM,IAAI,GAA2B,EAAE,CAAC;IACxC,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QAChC,MAAM,CAAC,GAAG,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,CAAC,EAAE,8BAA8B,KAAK,EAAE,CAAC,CAAC;QACjD,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;QAChD,IAAI,CAAC,KAAK,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IAClC,CAAC;IACD,MAAM,OAAO,CAAC,OAAO,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;AACnD,CAAC"}

package/dist/e2e/harness/TestRebalancer.d.ts

@@ -0,0 +1,85 @@
+import { BigNumber } from 'ethers';
+import { type MultiProvider } from '@humanode-hyperlane/sdk';
+import { RebalancerConfig } from '../../config/RebalancerConfig.js';
+import { type StrategyConfig } from '../../config/types.js';
+import { RebalancerOrchestrator } from '../../core/RebalancerOrchestrator.js';
+import { RebalancerContextFactory } from '../../factories/RebalancerContextFactory.js';
+import type { ConfirmedBlockTags } from '../../interfaces/IMonitor.js';
+import type { IStrategy } from '../../interfaces/IStrategy.js';
+import type { InventoryMonitorConfig, Monitor } from '../../monitor/Monitor.js';
+import type { IActionTracker } from '../../tracking/IActionTracker.js';
+import type { ExplorerMessage } from '../../utils/ExplorerClient.js';
+import { BALANCE_PRESETS, type DeployedAddresses, type Erc20InventoryDeployedAddresses, INVENTORY_SIGNER_PRESETS, type NativeDeployedAddresses, type TestChain } from '../fixtures/routes.js';
+import { BaseLocalDeploymentManager } from './BaseLocalDeploymentManager.js';
+import { ForkIndexer } from './ForkIndexer.js';
+import { MockExplorerClient } from './MockExplorerClient.js';
+import { MockExternalBridge } from './MockExternalBridge.js';
+export interface PendingTransferParams {
+    from: TestChain;
+    to: TestChain;
+    amount: BigNumber;
+    warpRecipient?: string;
+}
+export interface TestRebalancerContext {
+    orchestrator: RebalancerOrchestrator;
+    strategy: IStrategy;
+    tracker: IActionTracker;
+    mockExplorer: MockExplorerClient;
+    forkIndexer: ForkIndexer;
+    multiProvider: MultiProvider;
+    rebalancerConfig: RebalancerConfig;
+    contextFactory: RebalancerContextFactory;
+    inventoryConfig?: InventoryMonitorConfig;
+    createMonitor(checkFrequency: number): Monitor;
+    getConfirmedBlockTags(): Promise<ConfirmedBlockTags>;
+}
+type BalancePreset = keyof typeof BALANCE_PRESETS;
+type ExecutionMode = 'propose' | 'execute';
+type InventorySignerPreset = keyof typeof INVENTORY_SIGNER_PRESETS;
+type InventorySignerBalanceConfig = InventorySignerPreset | Partial<Record<string, BigNumber>>;
+type TestErc20InventoryConfig = {
+    inventorySignerKey: string;
+    erc20DeployedAddresses: Erc20InventoryDeployedAddresses;
+};
+export declare class TestRebalancerBuilder {
+    private readonly deploymentManager;
+    private readonly multiProvider;
+    private strategyConfig;
+    private balanceConfig;
+    private pendingTransfers;
+    private mockTransfers;
+    private executionMode;
+    private inventoryConfig;
+    private erc20InventoryConfig;
+    private mockExternalBridge;
+    private readonly logger;
+    private inventorySignerBalanceConfig;
+    constructor(deploymentManager: BaseLocalDeploymentManager<DeployedAddresses | NativeDeployedAddresses>, multiProvider: MultiProvider);
+    withStrategy(config: StrategyConfig[]): this;
+    withBalances(preset: BalancePreset | Record<string, BigNumber>): this;
+    withPendingTransfer(params: PendingTransferParams): this;
+    withMockTransfer(message: ExplorerMessage): this;
+    withExecutionMode(mode: ExecutionMode): this;
+    withInventoryConfig(config: {
+        inventorySignerKey: string;
+        nativeDeployedAddresses: NativeDeployedAddresses;
+    }): this;
+    withErc20InventoryConfig(config: TestErc20InventoryConfig): this;
+    withMockExternalBridge(bridge: MockExternalBridge): this;
+    withInventorySignerBalances(config: InventorySignerBalanceConfig): this;
+    private computeConfirmedBlockTags;
+    build(): Promise<TestRebalancerContext>;
+    private getBalanceChains;
+    private getBalances;
+    private setupBalances;
+    private getInventorySignerAddress;
+    private setupInventorySignerBalances;
+    private buildMockExplorerConfig;
+    private getMonitoredRouteAddresses;
+    private getWorkingMultiProvider;
+}
+export declare class TestRebalancer {
+    static builder(deploymentManager: BaseLocalDeploymentManager<DeployedAddresses | NativeDeployedAddresses>, multiProvider: MultiProvider): TestRebalancerBuilder;
+}
+export {};
+//# sourceMappingURL=TestRebalancer.d.ts.map

package/dist/e2e/harness/TestRebalancer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestRebalancer.d.ts","sourceRoot":"","sources":["../../../src/e2e/harness/TestRebalancer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAU,MAAM,QAAQ,CAAC;AAI3C,OAAO,EAGL,KAAK,aAAa,EAEnB,MAAM,yBAAyB,CAAC;AAOjC,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAGL,KAAK,cAAc,EAEpB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,sBAAsB,EAEvB,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EAAE,wBAAwB,EAAE,MAAM,6CAA6C,CAAC;AAEvF,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AACvE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,KAAK,EAAE,sBAAsB,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AAChF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AACvE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AACrE,OAAO,EAEL,eAAe,EAEf,KAAK,iBAAiB,EAEtB,KAAK,+BAA+B,EACpC,wBAAwB,EAGxB,KAAK,uBAAuB,EAE5B,KAAK,SAAS,EAIf,MAAM,uBAAuB,CAAC;AAG/B,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAC7E,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EACL,kBAAkB,EAEnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAW7D,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,SAAS,CAAC;IAChB,EAAE,EAAE,SAAS,CAAC;IACd,MAAM,EAAE,SAAS,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,qBAAqB;IACpC,YAAY,EAAE,sBAAsB,CAAC;IACrC,QAAQ,EAAE,SAAS,CAAC;IACpB,OAAO,EAAE,cAAc,CAAC;IACxB,YAAY,EAAE,kBAAkB,CAAC;IACjC,WAAW,EAAE,WAAW,CAAC;IACzB,aAAa,EAAE,aAAa,CAAC;IAC7B,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,cAAc,EAAE,wBAAwB,CAAC;IACzC,eAAe,CAAC,EAAE,sBAAsB,CAAC;IACzC,aAAa,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC;IAC/C,qBAAqB,IAAI,OAAO,CAAC,kBAAkB,CAAC,CAAC;CACtD;AAED,KAAK,aAAa,GAAG,MAAM,OAAO,eAAe,CAAC;AAElD,KAAK,aAAa,GAAG,SAAS,GAAG,SAAS,CAAC;AAC3C,KAAK,qBAAqB,GAAG,MAAM,OAAO,wBAAwB,CAAC;AACnE,KAAK,4BAA4B,GAC7B,qBAAqB,GACrB,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC;AAOvC,KAAK,wBAAwB,GAAG;IAC9B,kBAAkB,EAAE,MAAM,CAAC;IAC3B,sBAAsB,EAAE,+BAA+B,CAAC;CACzD,CAAC;AAEF,qBAAa,qBAAqB;IAe9B,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAGlC,OAAO,CAAC,QAAQ,CAAC,aAAa;IAjBhC,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,aAAa,CAA6B;IAClD,OAAO,CAAC,gBAAgB,CAA+B;IACvD,OAAO,CAAC,aAAa,CAAyB;IAC9C,OAAO,CAAC,aAAa,CAA4B;IACjD,OAAO,CAAC,eAAe,CAAkC;IACzD,OAAO,CAAC,oBAAoB,CAAuC;IACnE,OAAO,CAAC,kBAAkB,CAAiC;IAC3D,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,4BAA4B,CAEtB;gBAGK,iBAAiB,EAAE,0BAA0B,CAC5D,iBAAiB,GAAG,uBAAuB,CAC5C,EACgB,aAAa,EAAE,aAAa;IAK/C,YAAY,CAAC,MAAM,EAAE,cAAc,EAAE,GAAG,IAAI;IAK5C,YAAY,CAAC,MAAM,EAAE,aAAa,GAAG,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,IAAI;IAKrE,mBAAmB,CAAC,MAAM,EAAE,qBAAqB,GAAG,IAAI;IAKxD,gBAAgB,CAAC,OAAO,EAAE,eAAe,GAAG,IAAI;IAKhD,iBAAiB,CAAC,IAAI,EAAE,aAAa,GAAG,IAAI;IAK5C,mBAAmB,CAAC,MAAM,EAAE;QAC1B,kBAAkB,EAAE,MAAM,CAAC;QAC3B,uBAAuB,EAAE,uBAAuB,CAAC;KAClD,GAAG,IAAI;IAKR,wBAAwB,CAAC,MAAM,EAAE,wBAAwB,GAAG,IAAI;IAKhE,sBAAsB,CAAC,MAAM,EAAE,kBAAkB,GAAG,IAAI;IAKxD,2BAA2B,CAAC,MAAM,EAAE,4BAA4B,GAAG,IAAI;YAKzD,yBAAyB;IA2BjC,KAAK,IAAI,OAAO,CAAC,qBAAqB,CAAC;IA+M7C,OAAO,CAAC,gBAAgB;IAOxB,OAAO,CAAC,WAAW;YAaL,aAAa;IAmH3B,OAAO,CAAC,yBAAyB;YAMnB,4BAA4B;IA4G1C,OAAO,CAAC,uBAAuB;IA4C/B,OAAO,CAAC,0BAA0B;YAiBpB,uBAAuB;CAsBtC;AAED,qBAAa,cAAc;IACzB,MAAM,CAAC,OAAO,CACZ,iBAAiB,EAAE,0BAA0B,CAC3C,iBAAiB,GAAG,uBAAuB,CAC5C,EACD,aAAa,EAAE,aAAa,GAC3B,qBAAqB;CAGzB"}

package/dist/e2e/harness/TestRebalancer.js

@@ -0,0 +1,427 @@
+import { BigNumber, ethers } from 'ethers';
+import { pino } from 'pino';
+import { ERC20Test__factory } from '@humanode-hyperlane/core';
+import { HyperlaneCore, MultiProtocolProvider, } from '@humanode-hyperlane/sdk';
+import { ProtocolType, addressToBytes32, assert, } from '@humanode-hyperlane/utils';
+import { RebalancerConfig } from '../../config/RebalancerConfig.js';
+import { DEFAULT_INTENT_TTL_MS, ExternalBridgeType, getStrategyChainNames, } from '../../config/types.js';
+import { RebalancerOrchestrator, } from '../../core/RebalancerOrchestrator.js';
+import { RebalancerContextFactory } from '../../factories/RebalancerContextFactory.js';
+import { ANVIL_TEST_PRIVATE_KEY, BALANCE_PRESETS, DOMAIN_IDS, ERC20_INVENTORY_MONITORED_ROUTE_ID, INVENTORY_SIGNER_PRESETS, MONITORED_ROUTE_ID, NATIVE_MONITORED_ROUTE_ID, TEST_CHAINS, buildErc20InventoryWarpRouteConfig, buildNativeWarpRouteConfig, buildWarpRouteConfig, } from '../fixtures/routes.js';
+import { setupCollateralBalances } from './BridgeSetup.js';
+import { ForkIndexer } from './ForkIndexer.js';
+import { MockExplorerClient, } from './MockExplorerClient.js';
+function encodeWarpRouteMessageBody(recipient, amount) {
+    const recipientBytes32 = addressToBytes32(recipient);
+    const amountHex = ethers.utils.hexZeroPad(amount.toHexString(), 32);
+    return recipientBytes32 + amountHex.slice(2);
+}
+export class TestRebalancerBuilder {
+    deploymentManager;
+    multiProvider;
+    strategyConfig;
+    balanceConfig = 'BALANCED';
+    pendingTransfers = [];
+    mockTransfers = [];
+    executionMode = 'propose';
+    inventoryConfig;
+    erc20InventoryConfig;
+    mockExternalBridge;
+    logger;
+    inventorySignerBalanceConfig;
+    constructor(deploymentManager, multiProvider) {
+        this.deploymentManager = deploymentManager;
+        this.multiProvider = multiProvider;
+        this.logger = pino({ level: 'debug' }).child({ module: 'TestRebalancer' });
+    }
+    withStrategy(config) {
+        this.strategyConfig = config;
+        return this;
+    }
+    withBalances(preset) {
+        this.balanceConfig = preset;
+        return this;
+    }
+    withPendingTransfer(params) {
+        this.pendingTransfers.push(params);
+        return this;
+    }
+    withMockTransfer(message) {
+        this.mockTransfers.push(message);
+        return this;
+    }
+    withExecutionMode(mode) {
+        this.executionMode = mode;
+        return this;
+    }
+    withInventoryConfig(config) {
+        this.inventoryConfig = config;
+        return this;
+    }
+    withErc20InventoryConfig(config) {
+        this.erc20InventoryConfig = config;
+        return this;
+    }
+    withMockExternalBridge(bridge) {
+        this.mockExternalBridge = bridge;
+        return this;
+    }
+    withInventorySignerBalances(config) {
+        this.inventorySignerBalanceConfig = config;
+        return this;
+    }
+    async computeConfirmedBlockTags() {
+        const blockTags = {};
+        const localProviders = this.deploymentManager.getContext().providers;
+        for (const [chain, provider] of localProviders) {
+            try {
+                const blockNumber = await provider.send('eth_blockNumber', []);
+                blockTags[chain] = parseInt(blockNumber, 16);
+                this.logger.debug({ chain, blockNumber: blockTags[chain] }, 'Computed confirmed block tag');
+            }
+            catch (error) {
+                this.logger.warn({
+                    chain,
+                    error: error instanceof Error ? error.message : String(error),
+                }, 'Failed to get block number, using undefined');
+                blockTags[chain] = undefined;
+            }
+        }
+        return blockTags;
+    }
+    async build() {
+        if (!this.strategyConfig || this.strategyConfig.length === 0) {
+            throw new Error('Strategy config is required. Call withStrategy() before build().');
+        }
+        const strategyChains = getStrategyChainNames(this.strategyConfig);
+        const balanceChains = this.getBalanceChains();
+        const missingChains = strategyChains.filter((chain) => !balanceChains.includes(chain));
+        if (missingChains.length > 0) {
+            throw new Error(`Balance config missing chains required by strategy: ${missingChains.join(', ')}. ` +
+                `Strategy chains: ${strategyChains.join(', ')}, Balance chains: ${balanceChains.join(', ')}`);
+        }
+        assert(!(this.inventoryConfig && this.erc20InventoryConfig), 'Cannot set both inventoryConfig and erc20InventoryConfig — use one or the other');
+        if (this.inventoryConfig && !this.mockExternalBridge) {
+            throw new Error('Inventory mode requires .withMockExternalBridge() to prevent hitting real external bridges in tests');
+        }
+        await this.setupBalances();
+        const inventoryModeConfig = this.inventoryConfig;
+        const erc20InventoryModeConfig = this.erc20InventoryConfig;
+        const isInventoryMode = inventoryModeConfig !== undefined;
+        const isErc20InventoryMode = erc20InventoryModeConfig !== undefined;
+        const ctx = this.deploymentManager.getContext();
+        const { providers: localProviders } = ctx;
+        const deployedAddresses = ctx.deployedAddresses;
+        if (!isInventoryMode && !('tokens' in deployedAddresses)) {
+            throw new Error('Expected ERC20 deployed addresses with tokens field');
+        }
+        const coreAddresses = {};
+        for (const chain of TEST_CHAINS) {
+            const chainAddresses = isErc20InventoryMode
+                ? erc20InventoryModeConfig.erc20DeployedAddresses.chains[chain]
+                : isInventoryMode
+                    ? inventoryModeConfig.nativeDeployedAddresses.chains[chain]
+                    : deployedAddresses.chains[chain];
+            if (!chainAddresses) {
+                throw new Error(`Missing chain addresses for ${chain}`);
+            }
+            coreAddresses[chain] = {
+                mailbox: chainAddresses.mailbox,
+                interchainSecurityModule: chainAddresses.ism,
+            };
+        }
+        const hyperlaneCore = HyperlaneCore.fromAddressesMap(coreAddresses, this.multiProvider);
+        const deployerWallet = new ethers.Wallet(ANVIL_TEST_PRIVATE_KEY);
+        const rebalancerAddresses = [deployerWallet.address];
+        const workingMultiProvider = await this.getWorkingMultiProvider();
+        const forkIndexer = new ForkIndexer(localProviders, hyperlaneCore, rebalancerAddresses, this.logger);
+        const confirmedBlockTags = await this.computeConfirmedBlockTags();
+        await forkIndexer.initialize(confirmedBlockTags);
+        const mockExplorer = new MockExplorerClient(this.buildMockExplorerConfig(), forkIndexer, () => this.computeConfirmedBlockTags());
+        const registry = this.deploymentManager.getRegistry();
+        const mpp = MultiProtocolProvider.fromMultiProvider(workingMultiProvider);
+        let rebalancerConfig;
+        let warpCoreConfig;
+        if (isErc20InventoryMode) {
+            const inventorySignerAddress = new ethers.Wallet(erc20InventoryModeConfig.inventorySignerKey).address;
+            rebalancerAddresses.push(inventorySignerAddress);
+            rebalancerConfig = new RebalancerConfig(ERC20_INVENTORY_MONITORED_ROUTE_ID, this.strategyConfig, DEFAULT_INTENT_TTL_MS, {
+                [ProtocolType.Ethereum]: {
+                    address: inventorySignerAddress,
+                    key: erc20InventoryModeConfig.inventorySignerKey,
+                },
+            }, { lifi: { integrator: 'test' } });
+            warpCoreConfig = buildErc20InventoryWarpRouteConfig(erc20InventoryModeConfig.erc20DeployedAddresses);
+        }
+        else if (isInventoryMode) {
+            const inventorySignerAddress = new ethers.Wallet(inventoryModeConfig.inventorySignerKey).address;
+            rebalancerAddresses.push(inventorySignerAddress);
+            rebalancerConfig = new RebalancerConfig(NATIVE_MONITORED_ROUTE_ID, this.strategyConfig, DEFAULT_INTENT_TTL_MS, {
+                [ProtocolType.Ethereum]: {
+                    address: inventorySignerAddress,
+                    key: inventoryModeConfig.inventorySignerKey,
+                },
+            }, { lifi: { integrator: 'test' } });
+            warpCoreConfig = buildNativeWarpRouteConfig(inventoryModeConfig.nativeDeployedAddresses);
+        }
+        else {
+            if (!('tokens' in deployedAddresses)) {
+                throw new Error('Expected ERC20 deployed addresses with tokens field');
+            }
+            rebalancerConfig = new RebalancerConfig(MONITORED_ROUTE_ID, this.strategyConfig, DEFAULT_INTENT_TTL_MS);
+            warpCoreConfig = buildWarpRouteConfig(deployedAddresses);
+        }
+        const contextFactory = await RebalancerContextFactory.create(rebalancerConfig, workingMultiProvider, mpp, registry, this.logger, undefined, warpCoreConfig);
+        const strategy = await contextFactory.createStrategy();
+        const { tracker, adapter } = await contextFactory.createActionTracker(mockExplorer);
+        await tracker.initialize();
+        this.logger.info('ActionTracker initialized with mock explorer');
+        const isAnyInventoryMode = isInventoryMode || isErc20InventoryMode;
+        const externalBridgeRegistryOverride = isAnyInventoryMode && this.mockExternalBridge
+            ? {
+                [ExternalBridgeType.LiFi]: this.mockExternalBridge,
+            }
+            : undefined;
+        const rebalancerComponents = this.executionMode === 'execute' || isAnyInventoryMode
+            ? await contextFactory.createRebalancers(tracker, undefined, externalBridgeRegistryOverride)
+            : undefined;
+        const rebalancers = this.executionMode === 'execute'
+            ? (rebalancerComponents?.rebalancers ?? [])
+            : [];
+        const inventoryConfig = rebalancerComponents?.inventoryConfig;
+        const externalBridgeRegistry = rebalancerComponents?.externalBridgeRegistry;
+        const orchestratorDeps = {
+            strategy,
+            rebalancers,
+            actionTracker: tracker,
+            inflightContextAdapter: adapter,
+            rebalancerConfig,
+            externalBridgeRegistry,
+            logger: this.logger,
+        };
+        const orchestrator = new RebalancerOrchestrator(orchestratorDeps);
+        return {
+            orchestrator,
+            strategy,
+            tracker,
+            mockExplorer,
+            forkIndexer,
+            multiProvider: workingMultiProvider,
+            rebalancerConfig,
+            contextFactory,
+            inventoryConfig,
+            createMonitor: (checkFrequency) => contextFactory.createMonitor(checkFrequency, inventoryConfig),
+            getConfirmedBlockTags: () => this.computeConfirmedBlockTags(),
+        };
+    }
+    getBalanceChains() {
+        if (typeof this.balanceConfig === 'string') {
+            return Object.keys(BALANCE_PRESETS[this.balanceConfig]);
+        }
+        return Object.keys(this.balanceConfig);
+    }
+    getBalances() {
+        if (typeof this.balanceConfig === 'string') {
+            const preset = BALANCE_PRESETS[this.balanceConfig];
+            return Object.fromEntries(Object.entries(preset).map(([chain, value]) => [
+                chain,
+                BigNumber.from(value),
+            ]));
+        }
+        return this.balanceConfig;
+    }
+    async setupBalances() {
+        const balances = this.getBalances();
+        const ctx = this.deploymentManager.getContext();
+        const { providers: localProviders } = ctx;
+        if (this.inventoryConfig) {
+            for (const [chain, balance] of Object.entries(balances)) {
+                const provider = localProviders.get(chain);
+                const monitoredRouteAddress = this.inventoryConfig.nativeDeployedAddresses.monitoredRoute[chain];
+                if (!provider) {
+                    throw new Error(`Missing local provider for inventory chain ${chain}`);
+                }
+                if (!monitoredRouteAddress) {
+                    throw new Error(`Missing monitored route address for inventory chain ${chain}`);
+                }
+                await provider.send('anvil_setBalance', [
+                    monitoredRouteAddress,
+                    ethers.utils.hexValue(balance),
+                ]);
+            }
+            this.logger.info({
+                balances: Object.fromEntries(Object.entries(balances).map(([chain, balance]) => [
+                    chain,
+                    balance.toString(),
+                ])),
+            }, 'Inventory balances configured on monitored routes');
+            await this.setupInventorySignerBalances(localProviders);
+            return;
+        }
+        if (this.erc20InventoryConfig) {
+            for (const [chain, balance] of Object.entries(balances)) {
+                const provider = localProviders.get(chain);
+                const tokenAddress = this.erc20InventoryConfig.erc20DeployedAddresses.tokens[chain];
+                const monitoredRouteAddress = this.erc20InventoryConfig.erc20DeployedAddresses.monitoredRoute[chain];
+                assert(provider, `setupBalances: missing provider for chain ${chain}`);
+                assert(tokenAddress, `setupBalances: missing token address for chain ${chain}`);
+                assert(monitoredRouteAddress, `setupBalances: missing monitored route address for chain ${chain}`);
+                const deployerSigner = new ethers.Wallet(ANVIL_TEST_PRIVATE_KEY, provider);
+                const token = ERC20Test__factory.connect(tokenAddress, deployerSigner);
+                const tx = await token.transfer(monitoredRouteAddress, balance);
+                await tx.wait();
+            }
+            this.logger.info({
+                balances: Object.fromEntries(Object.entries(balances).map(([chain, balance]) => [
+                    chain,
+                    balance.toString(),
+                ])),
+            }, 'ERC20 inventory balances configured on monitored routes');
+            await this.setupInventorySignerBalances(localProviders);
+            return;
+        }
+        const deployedAddresses = ctx.deployedAddresses;
+        if (!('tokens' in deployedAddresses)) {
+            throw new Error('Expected ERC20 deployed addresses with tokens field');
+        }
+        await setupCollateralBalances(localProviders, balances, deployedAddresses.monitoredRoute, deployedAddresses.tokens, ANVIL_TEST_PRIVATE_KEY);
+        this.logger.info({
+            balances: Object.fromEntries(Object.entries(balances).map(([chain, balance]) => [
+                chain,
+                balance.toString(),
+            ])),
+        }, 'Collateral balances configured');
+    }
+    getInventorySignerAddress() {
+        const config = this.inventoryConfig ?? this.erc20InventoryConfig;
+        assert(config, 'Expected inventoryConfig or erc20InventoryConfig');
+        return new ethers.Wallet(config.inventorySignerKey).address;
+    }
+    async setupInventorySignerBalances(localProviders) {
+        if (!this.inventorySignerBalanceConfig ||
+            (!this.inventoryConfig && !this.erc20InventoryConfig)) {
+            return;
+        }
+        const signerAddress = this.getInventorySignerAddress();
+        let balances;
+        if (typeof this.inventorySignerBalanceConfig === 'string') {
+            balances = INVENTORY_SIGNER_PRESETS[this.inventorySignerBalanceConfig];
+        }
+        else {
+            balances = Object.fromEntries(Object.entries(this.inventorySignerBalanceConfig)
+                .filter((entry) => entry[1] !== undefined)
+                .map(([chain, val]) => [chain, val.toString()]));
+        }
+        if (this.erc20InventoryConfig) {
+            const signerWallet = new ethers.Wallet(this.erc20InventoryConfig.inventorySignerKey);
+            const deployerKey = ANVIL_TEST_PRIVATE_KEY;
+            const tokens = this.erc20InventoryConfig.erc20DeployedAddresses.tokens;
+            for (const [chain, balance] of Object.entries(balances)) {
+                const provider = localProviders.get(chain);
+                assert(balance !== undefined, `setupInventorySignerBalances: missing balance for chain ${chain}`);
+                assert(provider, `setupInventorySignerBalances: missing provider for chain ${chain}`);
+                const tokenAddress = tokens[chain];
+                assert(tokenAddress, `setupInventorySignerBalances: missing token address for chain ${chain}`);
+                const connectedSigner = signerWallet.connect(provider);
+                const deployerSigner = new ethers.Wallet(deployerKey, provider);
+                const tokenAsSigner = ERC20Test__factory.connect(tokenAddress, connectedSigner);
+                const tokenAsDeployer = ERC20Test__factory.connect(tokenAddress, deployerSigner);
+                const current = await tokenAsSigner.balanceOf(signerAddress);
+                if (current.gt(0)) {
+                    const tx1 = await tokenAsSigner.transfer(deployerSigner.address, current);
+                    await tx1.wait();
+                }
+                if (BigNumber.from(balance).gt(0)) {
+                    const tx2 = await tokenAsDeployer.transfer(signerAddress, BigNumber.from(balance));
+                    await tx2.wait();
+                }
+            }
+            this.logger.info({ balances, signer: signerWallet.address }, 'ERC20 inventory signer balances configured');
+            return;
+        }
+        for (const [chain, balance] of Object.entries(balances)) {
+            const provider = localProviders.get(chain);
+            if (balance === undefined) {
+                continue;
+            }
+            if (!provider) {
+                throw new Error(`Missing local provider for inventory signer chain ${chain}`);
+            }
+            await provider.send('anvil_setBalance', [
+                signerAddress,
+                ethers.utils.hexValue(BigNumber.from(balance)),
+            ]);
+        }
+        this.logger.info({ balances, signer: signerAddress }, 'Inventory signer balances configured');
+    }
+    buildMockExplorerConfig() {
+        const monitoredRoute = this.getMonitoredRouteAddresses();
+        const userTransfers = [...this.mockTransfers];
+        for (let i = 0; i < this.pendingTransfers.length; i++) {
+            const params = this.pendingTransfers[i];
+            const warpRecipient = params.warpRecipient ?? monitoredRoute[params.to];
+            const mockTransfer = {
+                msg_id: `0x${(1000 + i).toString(16).padStart(64, '0')}`,
+                origin_domain_id: DOMAIN_IDS[params.from],
+                destination_domain_id: DOMAIN_IDS[params.to],
+                sender: monitoredRoute[params.from],
+                recipient: monitoredRoute[params.to],
+                origin_tx_hash: `0x${(2000 + i).toString(16).padStart(64, '0')}`,
+                origin_tx_sender: `0x${(3000 + i).toString(16).padStart(40, '0')}`,
+                origin_tx_recipient: monitoredRoute[params.from],
+                is_delivered: false,
+                message_body: encodeWarpRouteMessageBody(warpRecipient, params.amount),
+                send_occurred_at: null,
+            };
+            userTransfers.push(mockTransfer);
+            this.logger.debug({
+                transfer: {
+                    from: params.from,
+                    to: params.to,
+                    origin: mockTransfer.origin_domain_id,
+                    destination: mockTransfer.destination_domain_id,
+                    amount: params.amount.toString(),
+                },
+            }, 'Created mock pending transfer');
+        }
+        return {
+            userTransfers,
+            rebalanceActions: [],
+        };
+    }
+    getMonitoredRouteAddresses() {
+        if (this.erc20InventoryConfig) {
+            return this.erc20InventoryConfig.erc20DeployedAddresses.monitoredRoute;
+        }
+        if (this.inventoryConfig) {
+            return this.inventoryConfig.nativeDeployedAddresses.monitoredRoute;
+        }
+        const deployedAddresses = this.deploymentManager.getContext().deployedAddresses;
+        if (!('tokens' in deployedAddresses)) {
+            throw new Error('Expected ERC20 deployed addresses with tokens field');
+        }
+        return deployedAddresses.monitoredRoute;
+    }
+    async getWorkingMultiProvider() {
+        if (this.executionMode === 'propose') {
+            return this.multiProvider;
+        }
+        const ctx = this.deploymentManager.getContext();
+        const rebalancerMultiProvider = this.multiProvider.extendChainMetadata({});
+        const wallet = new ethers.Wallet(ANVIL_TEST_PRIVATE_KEY);
+        for (const chain of TEST_CHAINS) {
+            const provider = ctx.providers.get(chain);
+            if (provider) {
+                rebalancerMultiProvider.setSigner(chain, wallet.connect(provider));
+            }
+        }
+        this.logger.info('Created MultiProvider with deployer as rebalancer for execute mode');
+        return rebalancerMultiProvider;
+    }
+}
+export class TestRebalancer {
+    static builder(deploymentManager, multiProvider) {
+        return new TestRebalancerBuilder(deploymentManager, multiProvider);
+    }
+}
+//# sourceMappingURL=TestRebalancer.js.map