Skip to content

Commit 0a2ed8c

Browse files
toninorairtoninorair
committed
Add more checks to tests
1 parent db07202 commit 0a2ed8c

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

‎test/integration/Protocol.t.sol

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,7 @@ contract ProtocolIntegrationTests is TestBase {
100100
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply);
101101
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield);
102102
assertEq(_wrappedMToken.excess(), _excess -= 2);
103+
assertEq(_wrappedMToken.roundingError(), 1);
103104

104105
assertGe(
105106
int256(_wrapperBalanceOfM),
@@ -124,6 +125,7 @@ contract ProtocolIntegrationTests is TestBase {
124125
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply += 50_000000);
125126
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield);
126127
assertEq(_wrappedMToken.excess(), _excess);
128+
assertEq(_wrappedMToken.roundingError(), 1);
127129

128130
assertGe(
129131
int256(_wrapperBalanceOfM),
@@ -176,6 +178,7 @@ contract ProtocolIntegrationTests is TestBase {
176178
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply);
177179
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield);
178180
assertEq(_wrappedMToken.excess(), _excess -= 2);
181+
assertEq(_wrappedMToken.roundingError(), 2);
179182

180183
assertGe(
181184
int256(_wrapperBalanceOfM),
@@ -201,6 +204,7 @@ contract ProtocolIntegrationTests is TestBase {
201204
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply += 150_000000);
202205
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield);
203206
assertEq(_wrappedMToken.excess(), _excess -= 2);
207+
assertEq(_wrappedMToken.roundingError(), 3);
204208

205209
assertGe(
206210
int256(_wrapperBalanceOfM),
@@ -222,6 +226,7 @@ contract ProtocolIntegrationTests is TestBase {
222226
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply);
223227
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield -= 1_190592);
224228
assertEq(_wrappedMToken.excess(), _excess);
229+
assertEq(_wrappedMToken.roundingError(), 3);
225230

226231
assertGe(
227232
int256(_wrapperBalanceOfM),
@@ -288,6 +293,7 @@ contract ProtocolIntegrationTests is TestBase {
288293
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply += 100_000000);
289294
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield);
290295
assertEq(_wrappedMToken.excess(), _excess -= 2);
296+
assertEq(_wrappedMToken.roundingError(), 1);
291297

292298
assertGe(
293299
int256(_wrapperBalanceOfM),
@@ -338,6 +344,7 @@ contract ProtocolIntegrationTests is TestBase {
338344
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply += _daveBalance + _aliceBalance);
339345
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield += 1);
340346
assertEq(_wrappedMToken.excess(), _excess -= 1);
347+
assertEq(_wrappedMToken.roundingError(), 1);
341348

342349
// Assert Carol (Non-Earner)
343350
assertEq(_wrappedMToken.balanceOf(_carol), _carolBalance += _aliceBalance);
@@ -367,6 +374,7 @@ contract ProtocolIntegrationTests is TestBase {
367374
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply -= 50_000000);
368375
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield += 1);
369376
assertEq(_wrappedMToken.excess(), _excess -= 1);
377+
assertEq(_wrappedMToken.roundingError(), 1);
370378

371379
assertGe(
372380
int256(_wrapperBalanceOfM),
@@ -426,6 +434,7 @@ contract ProtocolIntegrationTests is TestBase {
426434
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply += 100_000000);
427435
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield);
428436
assertEq(_wrappedMToken.excess(), _excess -= 2);
437+
assertEq(_wrappedMToken.roundingError(), 1);
429438

430439
assertGe(
431440
int256(_wrapperBalanceOfM),
@@ -458,6 +467,7 @@ contract ProtocolIntegrationTests is TestBase {
458467
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply += 100_000000);
459468
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield += 1);
460469
assertEq(_wrappedMToken.excess(), _excess -= 1);
470+
assertEq(_wrappedMToken.roundingError(), 1);
461471

462472
assertGe(
463473
int256(_wrapperBalanceOfM),
@@ -490,6 +500,7 @@ contract ProtocolIntegrationTests is TestBase {
490500
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply += _aliceBalance);
491501
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield -= 3_614473 + 1);
492502
assertEq(_wrappedMToken.excess(), _excess += 1);
503+
assertEq(_wrappedMToken.roundingError(), 1);
493504

494505
assertGe(
495506
int256(_wrapperBalanceOfM),
@@ -510,6 +521,7 @@ contract ProtocolIntegrationTests is TestBase {
510521
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply -= _carolBalance);
511522
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield += 1);
512523
assertEq(_wrappedMToken.excess(), _excess -= 1);
524+
assertEq(_wrappedMToken.roundingError(), 1);
513525

514526
assertGe(
515527
int256(_wrapperBalanceOfM),
@@ -539,6 +551,7 @@ contract ProtocolIntegrationTests is TestBase {
539551
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply -= _aliceBalance);
540552
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield);
541553
assertEq(_wrappedMToken.excess(), _excess);
554+
assertEq(_wrappedMToken.roundingError(), 1);
542555

543556
// Assert Alice (Non-Earner)
544557
assertEq(_mToken.balanceOf(_alice), _aliceBalance);
@@ -558,6 +571,7 @@ contract ProtocolIntegrationTests is TestBase {
558571
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply);
559572
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield += 1);
560573
assertEq(_wrappedMToken.excess(), _excess -= 3);
574+
assertEq(_wrappedMToken.roundingError(), 2);
561575

562576
// Assert Bob (Earner)
563577
assertEq(_mToken.balanceOf(_bob), _bobBalance);
@@ -577,6 +591,7 @@ contract ProtocolIntegrationTests is TestBase {
577591
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply);
578592
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield += 1);
579593
assertEq(_wrappedMToken.excess(), _excess -= 1);
594+
assertEq(_wrappedMToken.roundingError(), 2);
580595

581596
// Assert Carol (Earner)
582597
assertEq(_mToken.balanceOf(_carol), _carolBalance);
@@ -595,6 +610,7 @@ contract ProtocolIntegrationTests is TestBase {
595610
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply -= _daveBalance);
596611
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield);
597612
assertEq(_wrappedMToken.excess(), _excess -= 2);
613+
assertEq(_wrappedMToken.roundingError(), 3);
598614

599615
// Assert Dave (Non-Earner)
600616
assertEq(_mToken.balanceOf(_dave), _daveBalance);
@@ -615,8 +631,9 @@ contract ProtocolIntegrationTests is TestBase {
615631
// Assert Globals
616632
assertEq(_wrappedMToken.totalEarningSupply(), _totalEarningSupply);
617633
assertEq(_wrappedMToken.totalNonEarningSupply(), _totalNonEarningSupply);
618-
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield, "3");
619-
assertEq(_wrappedMToken.excess(), _excess -= _excess, "4");
634+
assertEq(_wrappedMToken.totalAccruedYield(), _totalAccruedYield);
635+
assertEq(_wrappedMToken.excess(), _excess -= _excess);
636+
assertEq(_wrappedMToken.roundingError(), 3);
620637

621638
assertGe(
622639
int256(_wrapperBalanceOfM),

0 commit comments

Comments
 (0)