Fix PythBalance parser to work with commas (#280)

* Fix * Fix commas * Do it
pyth-network · Nov 20, 2023 · c8b37a1 · c8b37a1 · vercel · Nov 20, 2023
1 parent 7b16e33
commit c8b37a1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/staking/app/pythBalance.ts b/staking/app/pythBalance.ts
@@ -27,6 +27,8 @@ export class PythBalance {
   }
 
   static fromString(amount: string) {
+    amount = amount.split(",").join("");
+
     if (amount.match(INTEGER_REGEXP)) {
       return new PythBalance(new BN(amount).mul(new BN(10 ** PYTH_DECIMALS)));
     } else if (amount.match(DECIMAL_REGEXP)) {

diff --git a/staking/tests/pyth_balance.ts b/staking/tests/pyth_balance.ts
@@ -100,6 +100,11 @@ describe("pyth balance tests", async () => {
     assert(amount.eq(new PythBalance(new BN(60_969_430_243))));
     assert(!amount.isZero());
 
+    amount = PythBalance.fromString("60,969.430243");
+    assert.equal(amount.toString(), "60,969.430243");
+    assert(amount.eq(new PythBalance(new BN(60_969_430_243))));
+    assert(!amount.isZero());
+
     amount = PythBalance.fromString("060969.430243");
     assert.equal(amount.toString(), "60,969.430243");
     assert(amount.eq(new PythBalance(new BN(60_969_430_243))));