fix steth bridging integraion tests. use shares to check ballances

lidofinance · Sep 25, 2024 · b5e3656 · b5e3656
1 parent a05c70a
commit b5e3656
Show file tree

Hide file tree

Showing 3 changed files with 154 additions and 160 deletions.
diff --git a/contracts/stubs/StETHStub.sol b/contracts/stubs/StETHStub.sol
@@ -23,4 +23,8 @@ contract StETHStub is IStETH, ERC20 {
     function getTotalPooledEther() external pure returns (uint256) {
         return 0;
     }
+
+    function sharesOf(address _account) external view returns (uint256) {
+        return 0;
+    }
 }
diff --git a/contracts/token/interfaces/IStETH.sol b/contracts/token/interfaces/IStETH.sol
@@ -10,4 +10,5 @@ import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
 interface IStETH is IERC20 {
     function getTotalShares() external view returns (uint256);
     function getTotalPooledEther() external view returns (uint256);
+    function sharesOf(address _account) external view returns (uint256);
 }