fix(Dataworker): Account for total required netSendAmount when executing PoolRebalanceLeaves

src/clients/BalanceAllocator.ts

@@ -97,6 +97,12 @@ export class BalanceAllocator {
     return this.requestBalanceAllocations([{ chainId, tokens, holder, amount }]);
   }
 
+  async getBalanceSubUsed(chainId: number, token: string, holder: string): Promise<BigNumber> {
+    const balance = await this.getBalance(chainId, token, holder);
+    const used = this.getUsed(chainId, token, holder);
+    return balance.sub(used);
+  }
+
   async getBalance(chainId: number, token: string, holder: string): Promise<BigNumber> {
     if (!this.balances?.[chainId]?.[token]?.[holder]) {
       const balance = await this._queryBalance(chainId, token, holder);
@@ -114,6 +120,12 @@ export class BalanceAllocator {
     return this.balances[chainId][token][holder];
   }
 
+  testSetBalance(chainId: number, token: string, holder: string, balance: BigNumber): void {
+    this.balances[chainId] ??= {};
+    this.balances[chainId][token] ??= {};
+    this.balances[chainId][token][holder] = balance;
+  }
+
   getUsed(chainId: number, token: string, holder: string): BigNumber {
     if (!this.used?.[chainId]?.[token]?.[holder]) {
       // Note: cannot use assign because it breaks the BigNumber object.

src/common/Constants.ts

@@ -611,3 +611,31 @@ export const DEFAULT_GAS_MULTIPLIER: { [chainId: number]: number } = {
 };
 
 export const CONSERVATIVE_BUNDLE_FREQUENCY_SECONDS = 3 * 60 * 60; // 3 hours is a safe assumption for the time
+
+export const ARBITRUM_ORBIT_L1L2_MESSAGE_FEE_DATA: {
+  [chainId: number]: {
+    // Amount of tokens required to send a single message to the L2
+    amountWei: number;
+    // Multiple of the required amount above to send to the feePayer in case
+    // we are short funds. For example, if set to 10, then everytime we need to load more funds
+    // we'll send 10x the required amount.
+    amountMultipleToFund: number;
+    // Account that pays the fees on-chain that we will load more fee tokens into.
+    feePayer?: string;
+    // Token that the feePayer will pay the fees in.
+    feeToken?: string;
+  };
+} = {
+  // Leave feePayer undefined if feePayer is HubPool.
+  // Leave feeToken undefined if feeToken is ETH.
+  [CHAIN_IDs.ARBITRUM]: {
+    amountWei: 0.02,
+    amountMultipleToFund: 1,
+  },
+  [CHAIN_IDs.ALEPH_ZERO]: {
+    amountWei: 0.49,
+    amountMultipleToFund: 20,
+    feePayer: "0x0d57392895Db5aF3280e9223323e20F3951E81B1", // DonationBox
+    feeToken: TOKEN_SYMBOLS_MAP.AZERO.addresses[CHAIN_IDs.MAINNET],
+  },
+};