Skip to content

Commit a5aadf3

Browse files
committed
correct slope scaling and stabilize exchange rate test using mulWad and cleaned borrow setup
1 parent 117986e commit a5aadf3

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

test/unit/ZToken.t.sol

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)