diff --git a/test/PrizeStrategy.test.js b/test/PrizeStrategy.test.js index 07f57e64..1cb97151 100644 --- 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')) }) })