fill up the pool with higher stake if both are below min

ChorusOne · Jan 15, 2025 · 02034c4 · 02034c4
1 parent b5e1aea
commit 02034c4
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/packages/staking-cli/src/cmd/ton.ts b/packages/staking-cli/src/cmd/ton.ts
@@ -121,6 +121,7 @@ async function runTx (
       {
         delegator: config.delegatorAddress,
         validatorAddress: config.validatorAddress,
+        validatorAddress2: config.validatorAddress2,
         messageType: msgType,
         args: arg,
         broadcast: broadcastEnabled

diff --git a/packages/ton/src/TonPoolStaker.ts b/packages/ton/src/TonPoolStaker.ts
@@ -295,8 +295,8 @@ export class TonPoolStaker extends TonBaseStaker {
 
       // prioritize filling a pool that hasn't reached the minStake
       if (!hasReachedMinStake(balancePool1) && !hasReachedMinStake(balancePool2)) {
-          // if neither pool has reached minStake, prioritize the one with the smaller balance
-          return balancePool1 <= balancePool2 ? 0 : 1;
+          // if neither pool has reached minStake, prioritize the one with the higher balance
+          return balancePool1 >= balancePool2 ? 0 : 1;
       } else if (!hasReachedMinStake(balancePool1)) {
           return 0; // fill pool 1 to meet minStake
       } else if (!hasReachedMinStake(balancePool2)) {

diff --git a/packages/ton/test/pool-staker.spec.ts b/packages/ton/test/pool-staker.spec.ts
@@ -8,9 +8,9 @@ describe('TonPoolStaker', () => {
       expect(result).to.equal(0); // Pool 1 needs to reach minStake
   })
 
-  it('should prioritize the pool with a smaller balance if both are below minStake', () => {
+  it('should prioritize the pool with a higher balance if both are below minStake', () => {
       const result = TonPoolStaker.selectPool(200n, 1000n, [100n, 150n]);
-      expect(result).to.equal(0); // Pool 1 has a smaller balance
+      expect(result).to.equal(1); // Pool 2 has a higher balance
   })
 
   it('should balance the pools if both have reached minStake but are below maxStake', () => {