cApe Migration Optimization (#411)

chore: add to parameter
parallel-finance · Aug 31, 2023 · f345040 · f345040
1 parent 804435c
commit f345040
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/contracts/misc/HelperContract.sol b/contracts/misc/HelperContract.sol
@@ -61,9 +61,9 @@ contract HelperContract is Initializable, OwnableUpgradeable {
         IERC20(apeCoin).safeTransfer(msg.sender, amount);
     }
 
-    function cApeMigration(uint256 amount) external {
+    function cApeMigration(uint256 amount, address to) external {
         IERC20(cApeV1).safeTransferFrom(msg.sender, address(this), amount);
         IAutoCompoundApe(cApeV1).withdraw(amount);
-        IAutoCompoundApe(cApe).deposit(msg.sender, amount);
+        IAutoCompoundApe(cApe).deposit(to, amount);
     }
 }
diff --git a/test/helper_contract.spec.ts b/test/helper_contract.spec.ts
@@ -119,7 +119,7 @@ describe("Helper contract Test", () => {
     await waitForTx(
         await helperContract
             .connect(user1.signer)
-            .cApeMigration(parseEther("10000"))
+            .cApeMigration(parseEther("10000"), user1.address)
     );
     expect(await cApe.balanceOf(user1.address)).to.be.eq(parseEther("10000"))
   });