fix: W16: Event inconsistencies

lidofinance · Sep 11, 2024 · 1eaef58 · 1eaef58
1 parent 3b268f9
commit 1eaef58
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/CSModule.sol b/src/CSModule.sol
@@ -138,6 +138,10 @@ contract CSModule is
         uint256 indexed nodeOperatorId,
         uint256 amount
     );
+    event ELRewardsStealingPenaltyCompensated(
+        uint256 indexed nodeOperatorId,
+        uint256 amount
+    );
     event ELRewardsStealingPenaltySettled(uint256 indexed nodeOperatorId);
 
     error SenderIsNotEligible();
@@ -1089,6 +1093,7 @@ contract CSModule is
             nodeOperatorId: nodeOperatorId,
             incrementNonceIfUpdated: true
         });
+        emit ELRewardsStealingPenaltyCompensated(nodeOperatorId, msg.value);
     }
 
     /// @notice Report Node Operator's key as withdrawn and settle withdrawn amount

diff --git a/test/CSModule.t.sol b/test/CSModule.t.sol
@@ -5925,6 +5925,9 @@ contract CSMCompensateELRewardsStealingPenalty is CSMCommon {
 
         uint256 nonce = csm.getNonce();
 
+        vm.expectEmit(true, true, true, true, address(csm));
+        emit CSModule.ELRewardsStealingPenaltyCompensated(noId, amount + fine);
+
         vm.expectCall(
             address(accounting),
             abi.encodeWithSelector(
@@ -5956,6 +5959,9 @@ contract CSMCompensateELRewardsStealingPenalty is CSMCommon {
 
         uint256 nonce = csm.getNonce();
 
+        vm.expectEmit(true, true, true, true, address(csm));
+        emit CSModule.ELRewardsStealingPenaltyCompensated(noId, amount);
+
         vm.expectCall(
             address(accounting),
             abi.encodeWithSelector(