fix: morphoblue token test for logoUri

dinngo · Jun 25, 2024 · b9334c4 · b9334c4
1 parent 0841e6d
commit b9334c4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/test/logics/morphoblue/supply-collateral.test.ts b/test/logics/morphoblue/supply-collateral.test.ts
@@ -5,6 +5,7 @@ import * as core from '@protocolink/core';
 import { expect } from 'chai';
 import hre from 'hardhat';
 import * as morphoblue from 'src/logics/morphoblue';
+import omit from 'lodash/omit';
 import * as utils from 'test/utils';
 
 describe('mainnet-pb: Test Morphoblue SupplyCollateral Logic', function () {
@@ -68,7 +69,10 @@ describe('mainnet-pb: Test Morphoblue SupplyCollateral Logic', function () {
       await expect(user.sendTransaction(transactionRequest)).to.not.be.reverted;
       await expect(user.address).to.changeBalance(input.token, -input.amount);
       const collateralBalance = await service.getCollateralBalance(marketId, user.address);
-      expect(collateralBalance).to.be.deep.eq(new common.TokenAmount(input.token.wrapped, input.amount));
+      expect(collateralBalance.amount).eq(input.amount);
+      expect(JSON.stringify(omit(collateralBalance.token.toObject(), 'logoUri'))).eq(
+        JSON.stringify(omit(input.token.toObject(), 'logoUri'))
+      );
     });
   });
 });
diff --git a/test/logics/morphoblue/supply.test.ts b/test/logics/morphoblue/supply.test.ts
@@ -5,6 +5,7 @@ import * as core from '@protocolink/core';
 import { expect } from 'chai';
 import hre from 'hardhat';
 import * as morphoblue from 'src/logics/morphoblue';
+import omit from 'lodash/omit';
 import * as utils from 'test/utils';
 
 describe('mainnet-pb: Test Morphoblue Supply Logic', function () {
@@ -76,6 +77,10 @@ describe('mainnet-pb: Test Morphoblue Supply Logic', function () {
       await expect(user.address).to.changeBalance(input.token, -input.amount);
       const supplyBalance = await service.getSupplyBalance(marketId, user.address);
       expect(supplyBalance).to.be.deep.eq(new common.TokenAmount(input.token.wrapped, input.amount));
+      expect(supplyBalance.amount).eq(input.amount);
+      expect(JSON.stringify(omit(supplyBalance.token.toObject(), 'logoUri'))).eq(
+        JSON.stringify(omit(input.token.toObject(), 'logoUri'))
+      );
     });
   });
 });