test_RevertIf_NotDelegateCall

llamaxyz · Apr 8, 2024 · cf87865 · cf87865
1 parent a4d9265
commit cf87865
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/test/rewards-claimer/LlamaRewardsClaimGuard.t.sol b/test/rewards-claimer/LlamaRewardsClaimGuard.t.sol
@@ -98,4 +98,23 @@ contract ValidateActionCreation is LlamaRewardsClaimGuardTest {
     );
     CORE.createAction(CORE_TEAM_ROLE, STRATEGY, address(ACCOUNT), 0, data, "");
   }
+
+  function test_RevertIf_NotDelegateCall() public {
+    LlamaRewardsClaimAccountExtension.TargetData[] memory targetData = _setupClaimRewardsData();
+
+    bytes memory accountExtensionData = abi.encodeCall(LlamaRewardsClaimAccountExtension.claimRewards, (targetData));
+    bytes memory data =
+      abi.encodeCall(ILlamaAccount.execute, (address(rewardsClaimAccountExtension), false, 0, accountExtensionData));
+
+    vm.prank(coreTeam1);
+    vm.expectRevert(
+      abi.encodeWithSelector(
+        LlamaRewardsClaimGuard.UnauthorizedCall.selector,
+        address(rewardsClaimAccountExtension),
+        LlamaRewardsClaimAccountExtension.claimRewards.selector,
+        false
+      )
+    );
+    CORE.createAction(CORE_TEAM_ROLE, STRATEGY, address(ACCOUNT), 0, data, "");
+  }
 }