Add owner to StakeDeposited

uniswapfoundation · Feb 7, 2024 · 6b2d20e · 6b2d20e
1 parent f071a89
commit 6b2d20e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/src/UniStaker.sol b/src/UniStaker.sol
@@ -31,7 +31,9 @@ contract UniStaker is INotifiableRewardReceiver, ReentrancyGuard, Multicall {
 
   /// @notice Emitted when stake is deposited by a depositor, either to a new deposit or one that
   /// already exists.
-  event StakeDeposited(DepositIdentifier indexed depositId, uint256 amount, uint256 depositBalance);
+  event StakeDeposited(
+    address owner, DepositIdentifier indexed depositId, uint256 amount, uint256 depositBalance
+  );
 
   /// @notice Emitted when a depositor withdraws some portion of stake from a given deposit.
   event StakeWithdrawn(DepositIdentifier indexed depositId, uint256 amount, uint256 depositBalance);
@@ -260,7 +262,7 @@ contract UniStaker is INotifiableRewardReceiver, ReentrancyGuard, Multicall {
     totalDeposits[msg.sender] += _amount;
     earningPower[deposit.beneficiary] += _amount;
     deposit.balance += _amount;
-    emit StakeDeposited(_depositId, _amount, deposit.balance);
+    emit StakeDeposited(msg.sender, _depositId, _amount, deposit.balance);
   }
 
   /// @notice For an existing deposit, change the address to which governance voting power is
@@ -426,7 +428,7 @@ contract UniStaker is INotifiableRewardReceiver, ReentrancyGuard, Multicall {
       delegatee: _delegatee,
       beneficiary: _beneficiary
     });
-    emit StakeDeposited(_depositId, _amount, _amount);
+    emit StakeDeposited(msg.sender, _depositId, _amount, _amount);
     emit BeneficiaryAltered(_depositId, address(0), _beneficiary);
     emit DelegateeAltered(_depositId, address(0), _delegatee);
   }

diff --git a/test/UniStaker.t.sol b/test/UniStaker.t.sol
@@ -182,6 +182,7 @@ contract Stake is UniStakerTest {
     govToken.approve(address(uniStaker), _amount);
     vm.expectEmit();
     emit UniStaker.StakeDeposited(
+      _depositor,
       UniStaker.DepositIdentifier.wrap(UniStaker.DepositIdentifier.unwrap(depositId) + 1),
       _amount,
       _amount
@@ -255,6 +256,7 @@ contract Stake is UniStakerTest {
     govToken.approve(address(uniStaker), _amount);
     vm.expectEmit();
     emit UniStaker.StakeDeposited(
+      _depositor,
       UniStaker.DepositIdentifier.wrap(UniStaker.DepositIdentifier.unwrap(depositId) + 1),
       _amount,
       _amount
@@ -811,7 +813,9 @@ contract StakeMore is UniStakerTest {
     uniStaker.stakeMore(_depositId, _addAmount);
 
     vm.expectEmit();
-    emit UniStaker.StakeDeposited(_depositId, _addAmount, _depositAmount + _totalAdditionalStake);
+    emit UniStaker.StakeDeposited(
+      _depositor, _depositId, _addAmount, _depositAmount + _totalAdditionalStake
+    );
 
     uniStaker.stakeMore(_depositId, _addAmount);
     vm.stopPrank();