update EigenPod unit and integration tests

Layr-Labs · Dec 4, 2023 · 950eee3 · 950eee3
1 parent c017fa2
commit 950eee3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/test/unit/EigenPodUnit.t.sol b/src/test/unit/EigenPodUnit.t.sol
@@ -6,13 +6,20 @@ import "@openzeppelin/contracts/utils/Create2.sol";
 
 import "src/contracts/pods/EigenPod.sol";
 
+<<<<<<< HEAD
 import "src/test/mocks/ETHDepositMock.sol";
 import "src/test/mocks/DelayedWithdrawalRouterMock.sol";
 import "src/test/mocks/ERC20Mock.sol";
 import "src/test/harnesses/EigenPodHarness.sol";
 import "src/test/utils/ProofParsing.sol";
 import "src/test/utils/EigenLayerUnitTestSetup.sol";
 import "src/test/events/IEigenPodEvents.sol";
+=======
+contract EigenPodUnitTests is Test, ProofParsing {
+
+    using BytesLib for bytes;    
+    using BeaconChainProofs for *;
+>>>>>>> 880a0dca (update EigenPod unit and integration tests)
 
 contract EigenPodUnitTests is EigenLayerUnitTestSetup {
     // Contract Under Test: EigenPod

diff --git a/src/test/utils/ProofParsing.sol b/src/test/utils/ProofParsing.sol
@@ -159,6 +159,11 @@ contract ProofParsing is Test {
         return getWithdrawalCredentialProof();
     }
 
+    function getBalanceUpdateProof() public returns(bytes32[] memory) {
+        // Balance update proofs are the same as withdrawal credential proofs
+        return getWithdrawalCredentialProof();
+    }
+
     function getWithdrawalCredentialProof() public returns(bytes32[] memory) {
         bytes32[] memory withdrawalCredentialProof = new bytes32[](46);
         for (uint i = 0; i < 46; i++) {