Update tests to ensure sum-tree is updated

pooltogether · Aug 12, 2020 · 0c15f89 · 0c15f89
1 parent 2cada74
commit 0c15f89
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/PrizeStrategy.test.js b/test/PrizeStrategy.test.js
@@ -406,6 +406,9 @@ describe('PrizeStrategy', function() {
       // have the mock update the number of prize tickets
       await prizePool.call(prizeStrategy, 'afterDepositTo', wallet._address, toWei('10'), ticket.address, []);
 
+      // confirm odds of winning
+      expect(await prizeStrategy.chanceOf(wallet._address)).to.equal(toWei('10'))
+
       // ensure prize period is over
       await prizeStrategy.setCurrentTime(await prizeStrategy.prizePeriodEndAt());
 
@@ -445,6 +448,8 @@ describe('PrizeStrategy', function() {
 
       expect(await call(prizeStrategy, 'balanceOfCredit', wallet._address)).to.equal(toWei('1.1'))
 
+      // confirm increased odds of winning
+      expect(await prizeStrategy.chanceOf(wallet._address)).to.equal(toWei('11'))
     })
   })