Primitive test fix

m0-foundation · Jul 3, 2024 · 1d4b119 · 1d4b119
1 parent 7c6feb4
commit 1d4b119
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/test/utils/Mocks.sol b/test/utils/Mocks.sol
@@ -8,6 +8,7 @@ contract MockM {
     uint128 public currentIndex;
 
     mapping(address account => uint256 balance) public balanceOf;
+    mapping(address account => bool isEarning) public isEarning;
 
     function transfer(address, uint256) external returns (bool success_) {
         return true;
@@ -28,6 +29,14 @@ contract MockM {
     function setTtgRegistrar(address ttgRegistrar_) external {
         ttgRegistrar = ttgRegistrar_;
     }
+
+    function startEarning() external {
+        isEarning[msg.sender] = true;
+    }
+
+    function stopEarning() external {
+        isEarning[msg.sender] = false;
+    }
 }
 
 contract MockRegistrar {