added creationTimestamp field to VoterInPoolPair entity

dexe-network · Sep 8, 2023 · bd42f6d · bd42f6d
1 parent 50fd0c7
commit bd42f6d
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 2 deletions.
diff --git a/DaoPools/schema.graphql b/DaoPools/schema.graphql
@@ -325,6 +325,8 @@ type TreasuryDelegationHistory @entity(immutable: true) {
 # @param delegateNfts the current delegated nfts
 type VoterInPoolPair @entity {
   id: Bytes! # pool address + from address + to address
+  creationTimestamp: BigInt!
+
   delegator: VoterInPool!
   delegatee: VoterInPool!
 

diff --git a/DaoPools/src/entities/Voters/VoterInPoolPair.ts b/DaoPools/src/entities/Voters/VoterInPoolPair.ts
@@ -1,13 +1,15 @@
 import { BigInt } from "@graphprotocol/graph-ts";
 import { VoterInPool, VoterInPoolPair } from "../../../generated/schema";
 
-export function getVoterInPoolPair(from: VoterInPool, to: VoterInPool): VoterInPoolPair {
+export function getVoterInPoolPair(from: VoterInPool, to: VoterInPool, timestamp: BigInt): VoterInPoolPair {
   let id = from.pool.concat(from.voter).concat(to.voter);
   let pair = VoterInPoolPair.load(id);
 
   if (pair == null) {
     pair = new VoterInPoolPair(id);
 
+    pair.creationTimestamp = timestamp;
+
     pair.delegator = from.id;
     pair.delegatee = to.id;
 

diff --git a/DaoPools/src/mappings/DaoPool.ts b/DaoPools/src/mappings/DaoPool.ts
@@ -80,7 +80,7 @@ export function onDelegated(event: Delegated): void {
   let toVoterInPool = getVoterInPool(pool, to, event.block.timestamp);
   let fromVoterInPool = getVoterInPool(pool, from, event.block.timestamp);
 
-  let pair = getVoterInPoolPair(fromVoterInPool, toVoterInPool);
+  let pair = getVoterInPoolPair(fromVoterInPool, toVoterInPool, event.block.timestamp);
 
   const usdAmount = getUSDValue(pool.erc20Token, event.params.amount);
 

diff --git a/DaoPools/tests/DaoPool.test.ts b/DaoPools/tests/DaoPool.test.ts
@@ -542,6 +542,12 @@ describe("DaoPool", () => {
       DelegationType.DELEGATE.toString()
     );
 
+    assert.fieldEquals(
+      "VoterInPoolPair",
+      contractSender.concat(from).concat(to).toHexString(),
+      "creationTimestamp",
+      block.timestamp.toString()
+    );
     assert.fieldEquals(
       "VoterInPoolPair",
       contractSender.concat(from).concat(to).toHexString(),
@@ -632,6 +638,12 @@ describe("DaoPool", () => {
       DelegationType.UNDELEGATE.toString()
     );
 
+    assert.fieldEquals(
+      "VoterInPoolPair",
+      contractSender.concat(from).concat(to).toHexString(),
+      "creationTimestamp",
+      block.timestamp.toString()
+    );
     assert.fieldEquals(
       "VoterInPoolPair",
       contractSender.concat(from).concat(to).toHexString(),