feat: update name

yearn · Sep 6, 2024 · 7ad27d8 · 7ad27d8
1 parent 59ece7f
commit 7ad27d8
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 0 deletions.
diff --git a/src/TokenizedStrategy.sol b/src/TokenizedStrategy.sol
@@ -1594,6 +1594,14 @@ contract TokenizedStrategy {
         emit UpdateProfitMaxUnlockTime(_profitMaxUnlockTime);
     }
 
+    /**
+     * @notice Updates the name for the strategy.
+     * @param _name The new name for the strategy.
+     */
+    function setName(string calldata _name) external onlyManagement {
+        _strategyStorage().name = _name;
+    }
+
     /*//////////////////////////////////////////////////////////////
                         ERC20 METHODS
     //////////////////////////////////////////////////////////////*/

diff --git a/src/interfaces/ITokenizedStrategy.sol b/src/interfaces/ITokenizedStrategy.sol
@@ -150,6 +150,8 @@ interface ITokenizedStrategy is IERC4626, IERC20Permit {
 
     function setProfitMaxUnlockTime(uint256 _profitMaxUnlockTime) external;
 
+    function setName(string calldata _newName) external;
+
     function shutdownStrategy() external;
 
     function emergencyWithdraw(uint256 _amount) external;

diff --git a/src/test/AccessControl.t.sol b/src/test/AccessControl.t.sol
@@ -369,4 +369,19 @@ contract AccessControlTest is Setup {
         vm.prank(keeper);
         strategy.tend();
     }
+
+    function test_setName(address _address) public {
+        vm.assume(_address != address(strategy) && _address != management);
+
+        string memory newName = "New Strategy Name";
+
+        vm.prank(_address);
+        vm.expectRevert("!management");
+        strategy.setName(newName);
+
+        vm.prank(management);
+        strategy.setName(newName);
+
+        assertEq(strategy.name(), newName);
+    }
 }