add unit tests for intervalUpdates

Uniswap · May 29, 2024 · 83777e6 · 83777e6
1 parent 5e0fb21
commit 83777e6
Show file tree

Hide file tree

Showing 2 changed files with 490 additions and 2 deletions.
diff --git a/src/utils/intervalUpdates.ts b/src/utils/intervalUpdates.ts
@@ -18,8 +18,8 @@ import { FACTORY_ADDRESS } from './constants'
  * Tracks global aggregate data over daily windows
  * @param event
  */
-export function updateUniswapDayData(event: ethereum.Event): UniswapDayData {
-  const uniswap = Factory.load(FACTORY_ADDRESS)!
+export function updateUniswapDayData(event: ethereum.Event, factoryAddress: string = FACTORY_ADDRESS): UniswapDayData {
+  const uniswap = Factory.load(factoryAddress)!
   const timestamp = event.block.timestamp.toI32()
   const dayID = timestamp / 86400 // rounded
   const dayStartTimestamp = dayID * 86400
@@ -72,6 +72,7 @@ export function updatePoolDayData(event: ethereum.Event): PoolDayData {
   poolDayData.sqrtPrice = pool.sqrtPrice
   poolDayData.token0Price = pool.token0Price
   poolDayData.token1Price = pool.token1Price
+  poolDayData.close = pool.token0Price
   poolDayData.tick = pool.tick
   poolDayData.tvlUSD = pool.totalValueLockedUSD
   poolDayData.txCount = poolDayData.txCount.plus(ONE_BI)