From 950eee3511faa5a135f4f6f701252423a13bedfb Mon Sep 17 00:00:00 2001
From: Yash Patil <ypatil12@gmail.com>
Date: Mon, 6 Nov 2023 09:53:40 -0500
Subject: [PATCH] update EigenPod unit and integration tests

---
 src/test/unit/EigenPodUnit.t.sol | 7 +++++++
 src/test/utils/ProofParsing.sol  | 5 +++++
 2 files changed, 12 insertions(+)

diff --git a/src/test/unit/EigenPodUnit.t.sol b/src/test/unit/EigenPodUnit.t.sol
index b2cf50bee..3c37c44a4 100644
--- a/src/test/unit/EigenPodUnit.t.sol
+++ b/src/test/unit/EigenPodUnit.t.sol
@@ -6,6 +6,7 @@ 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";
@@ -13,6 +14,12 @@ 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
index 44b7de2de..1abd620f6 100644
--- 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++) {