Fix failing test in Uniswap V3 handler

Author: 0xJem

subgraphs/ethereum/src/liquidity/LiquidityUniswapV3.ts

@@ -278,6 +278,7 @@ export function getUniswapV3PairTotalValue(pairAddress: string, excludeOhmValue:
 
     // If the token is OHM, add the balance to the OHM balance
     if (currentToken.toLowerCase() == ERC20_OHM_V2.toLowerCase()) {
+      log.debug("getUniswapV3PairTotalValue: Adding OHM balance for pair {}: {}", [getContractName(pairAddress), ohmBalance.toString()])
       ohmBalance = ohmBalance.plus(currentBalance);
     }
 

subgraphs/ethereum/src/utils/ContractHelper.ts

@@ -344,11 +344,13 @@ export function getERC20DecimalBalance(tokenAddress: string, walletAddress: stri
   const tokenSnapshot = getOrCreateERC20TokenSnapshot(tokenAddress, blockNumber);
   const tokenContract = getERC20(tokenAddress, blockNumber);
   if (tokenContract === null || tokenSnapshot.totalSupply === null) {
+    log.info("getERC20DecimalBalance: totalSupply is zero. Token: {}", [getContractName(tokenAddress)]);
     return BigDecimal.zero();
   }
 
   const balanceResult = tokenContract.try_balanceOf(Address.fromString(walletAddress));
   if (balanceResult.reverted) {
+    log.info("getERC20DecimalBalance: balanceOf() reverted. Token: {}, wallet: {}, block: {}", [getContractName(tokenAddress), getContractName(walletAddress), blockNumber.toString()]);
     return BigDecimal.zero();
   }
 
@@ -529,12 +531,12 @@ export function getERC20TokenRecordFromWallet(
 /**
  * Generates TokenRecord objects for the given ERC20 token, representing deposits into a
  * Vendor Finance market.
- * 
- * @param timestamp 
- * @param contractAddress 
- * @param rate 
- * @param blockNumber 
- * @returns 
+ *
+ * @param timestamp
+ * @param contractAddress
+ * @param rate
+ * @param blockNumber
+ * @returns
  */
 export function getVendorFinanceRecords(
   timestamp: BigInt,
@@ -584,12 +586,12 @@ export function getVendorFinanceRecords(
 /**
  * Generates TokenRecord objects for the given ERC20 token, representing deposits into a
  * Myso Finance market.
- * 
- * @param timestamp 
- * @param contractAddress 
- * @param rate 
- * @param blockNumber 
- * @returns 
+ *
+ * @param timestamp
+ * @param contractAddress
+ * @param rate
+ * @param blockNumber
+ * @returns
  */
 export function getMysoFinanceRecords(
   timestamp: BigInt,

subgraphs/ethereum/tests/price.test.ts

@@ -29,6 +29,7 @@ import {
   ERC20_FXS,
   ERC20_OHM_V1,
   ERC20_OHM_V2,
+  ERC20_OHM_V2_BLOCK,
   ERC20_SYN,
   ERC20_TRIBE,
   ERC20_USDC,
@@ -54,7 +55,7 @@ import { getUSDRate, getUSDRateBalancer } from "../src/utils/Price";
 import { getWalletAddressesForContract } from "../src/utils/ProtocolAddresses";
 import { mockClearinghouseRegistryAddressNull, mockTreasuryAddressNull } from "./bophadesHelper";
 import { mockStablecoinsPriceFeeds } from "./chainlink";
-import { ERC20_STANDARD_DECIMALS, mockERC20Balance } from "./erc20Helper";
+import { ERC20_STANDARD_DECIMALS, mockERC20Balance, mockERC20TotalSupply } from "./erc20Helper";
 import {
   getERC20UsdRate,
   getEthUsdRate,
@@ -215,13 +216,16 @@ describe("OHM-USD rate", () => {
     mockERC20Balance(ERC20_OHM_V2, PAIR_UNISWAP_V3_WETH_OHM, toBigInt(ohmBalance, OHM_V2_DECIMALS));
     mockERC20Balance(ERC20_WETH, PAIR_UNISWAP_V3_WETH_OHM, toBigInt(ethBalance, ERC20_STANDARD_DECIMALS));
 
+    // Mock total supply (required by Uniswap V3 handler)
+    mockERC20TotalSupply(ERC20_OHM_V2, OHM_V2_DECIMALS, toBigInt(ohmBalance, OHM_V2_DECIMALS));
+
     // 919.574080927401380445 * 1898.01397374 / 130454.081369749 = 13.3791479512
     // TODO figure out why this is not being returned
     const calculatedRate = BigDecimal.fromString("13.3835");
 
     assert.stringEquals(
       calculatedRate.truncate(4).toString(),
-      getUSDRate(ERC20_OHM_V2, BigInt.zero()).truncate(4).toString(),
+      getUSDRate(ERC20_OHM_V2, BigInt.fromString(ERC20_OHM_V2_BLOCK)).truncate(4).toString(),
     );
   });