@@ -21,6 +21,8 @@ contract ZTokenTest is Test {
2121
2222 uint256 private constant MINIMUM_AMOUNT = 1e18 ;
2323
24+ uint256 public constant SECONDS_PER_YEAR = 365 days ;
25+
2426 function setUp () public {
2527 IZToken.MarketConfig memory config = IZToken.MarketConfig ({
2628 underlyingToken: underlyingToken,
@@ -39,9 +41,9 @@ contract ZTokenTest is Test {
3941
4042 assertEq (zToken.UTILIZATION_THRESHOLD (), UTILIZATION_THRESHOLD);
4143
42- assertEq (zToken.SLOPE_BEFORE_KINK (), SLOPE_BEFORE_KINK);
44+ assertEq (zToken.SLOPE_BEFORE_KINK (), SLOPE_BEFORE_KINK / SECONDS_PER_YEAR );
4345
44- assertEq (zToken.SLOPE_AFTER_KINK (), SLOPE_AFTER_KINK);
46+ assertEq (zToken.SLOPE_AFTER_KINK (), SLOPE_AFTER_KINK / SECONDS_PER_YEAR );
4547
4648 assertEq (zToken.BASE_BORROW_RATE (), BASE_BORROW_RATE);
4749
@@ -55,13 +57,11 @@ contract ZTokenTest is Test {
5557 }
5658
5759 function test_GetExchangeRate_WhenATokenHasBeenBorrowed (uint256 _cash , uint256 _borrow ) external {
58- zToken.setTotalBorrow (_borrow);
59-
60- _cash = bound (_cash, MINIMUM_AMOUNT, type (uint128 ).max);
60+ _cash = bound (_cash, MINIMUM_AMOUNT, type (uint96 ).max);
6161 _borrow = bound (_borrow, _cash / 10 , _cash);
6262
6363 zToken.setTotalCash (_cash);
64- zToken.setTotalSupply (_cash.mulDiv (zToken.INITIAL_EXCHANGE_RATE (), 1e18 ));
64+ zToken.setTotalSupply (_cash.mulWad (zToken.INITIAL_EXCHANGE_RATE ()));
6565
6666 zToken.setTotalBorrow (_borrow);
6767 zToken.setTotalReserves (0 );
0 commit comments