Chore: claim tests (#305)

* ✅ Add claim tests * 🐛 fix total claimable bug * 🧪 update tests on claimables * 🔀 fix conflict
Carbonable · Dec 6, 2023 · 06cd1c8 · 06cd1c8
1 parent 2a2d95e
commit 06cd1c8
Show file tree

Hide file tree

Showing 4 changed files with 479 additions and 2 deletions.
diff --git a/src/components/offset/module.cairo b/src/components/offset/module.cairo
@@ -90,7 +90,8 @@ mod Offset {
         fn get_total_claimable(self: @ContractState) -> u256 {
             let total_absorption = self.get_total_absorption();
             let claimed = self._offset_total_claimed.read();
-            total_absorption + claimed
+            assert(total_absorption >= claimed, 'Total absorption is too low');
+            total_absorption - claimed
         }
 
         fn get_total_claimed(self: @ContractState) -> u256 {

diff --git a/src/components/yield/module.cairo b/src/components/yield/module.cairo
@@ -151,7 +151,8 @@ mod Yield {
         fn get_total_claimable(self: @ContractState) -> u256 {
             let total_sale = self.get_total_sale();
             let claimed = self._yield_total_claimed.read();
-            total_sale + claimed
+            assert(total_sale >= claimed, 'Total absorption is too low');
+            total_sale - claimed
         }
 
         fn get_total_claimed(self: @ContractState) -> u256 {

diff --git a/src/lib.cairo b/src/lib.cairo
@@ -50,6 +50,7 @@ mod tests {
     mod test_yielder;
     mod test_apr;
     mod scenarios {
+        mod test_iso_yielder_claims;
         mod test_iso_yielder_setting;
         mod test_iso_offseter_setting;
         mod test_iso_las_delicias;