fix rewardsDistributor tests

forta-network · Feb 12, 2024 · 674f27a · 674f27a
1 parent d0bb925
commit 674f27a
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/subgraph/src/tests/rewardsDistributor.test.ts b/subgraph/src/tests/rewardsDistributor.test.ts
@@ -101,9 +101,7 @@ describe("Rewards distributor", () => {
   test("should handle a reward event on a pool with one delegator and return the correct APY", () => {
     //Given
     delegatorOne = new Staker(mockDelegatorStakerIdOne);
-    (delegatorOne as any).account = mockDelegatorStakerIdOne;
-    (delegatorOne as any).aggregateActiveStake = "";
-    (delegatorOne as any).aggregateTotalStake = "";
+    delegatorOne.account = mockDelegatorStakerIdOne;
     delegatorOne.nodePools = [mockPoolId];
     delegatorOne.save();
 
@@ -164,15 +162,11 @@ describe("Rewards distributor", () => {
     //Given
     delegatorOne = new Staker(mockDelegatorStakerIdOne);
     delegatorOne.account = mockDelegatorStakerIdOne;
-    (delegatorOne as any).aggregateActiveStake = "";
-    (delegatorOne as any).aggregateTotalStake = "";
     delegatorOne.nodePools = [mockPoolId];
     delegatorOne.save();
 
     delegatorTwo = new Staker(mockDelegatorStakerIdTwo);
     delegatorTwo.account = mockDelegatorStakerIdTwo;
-    (delegatorTwo as any).aggregateActiveStake = "";
-    (delegatorTwo as any).aggregateTotalStake = "";
     delegatorTwo.nodePools = [mockPoolId];
     delegatorTwo.save();