Adding getDeposit function to MezoPortalStub

thesis · Apr 8, 2024 · ba702cb · ba702cb
1 parent 98c6f00
commit ba702cb
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/core/contracts/test/MezoPortalStub.sol b/core/contracts/test/MezoPortalStub.sol
@@ -21,4 +21,15 @@ contract MezoPortalStub {
         depositCount++;
         IERC20(token).safeTransferFrom(msg.sender, address(this), amount);
     }
+
+    function getDeposit(
+        address depositor,
+        address token,
+        uint256 depositId
+    ) external view returns (uint96 balance, uint256 unlockAt) {
+        return (
+            uint96(IERC20(token).balanceOf(address(this))),
+            block.timestamp
+        );
+    }
 }