11// SPDX-License-Identifier: UNLICENSED
22pragma solidity ^ 0.8.0 ;
3+ import { Ownable } from "@openzeppelin/contracts/access/Ownable.sol " ;
34import { ISuperToken } from "@superfluid-finance/ethereum-contracts/contracts/superfluid/SuperToken.sol " ;
45import { SuperTokenV1Library } from "@superfluid-finance/ethereum-contracts/contracts/apps/SuperTokenV1Library.sol " ;
56import { FoundrySuperfluidTester } from "@superfluid-finance/ethereum-contracts/test/foundry/FoundrySuperfluidTester.t.sol " ;
@@ -108,7 +109,7 @@ contract ManagerTests is FoundrySuperfluidTester {
108109 vm.prank (admin);
109110 manager.setLimits (newMinLower, newMinUpper);
110111 // non owner can't set new limits
111- vm.expectRevert (bytes ( " Ownable: caller is not the owner " ));
112+ vm.expectRevert (abi.encodeWithSelector ( Ownable.OwnableUnauthorizedAccount. selector , address ( this ) ));
112113 manager.setLimits (newMinLower, newMinUpper);
113114 }
114115
@@ -124,7 +125,7 @@ contract ManagerTests is FoundrySuperfluidTester {
124125 manager.addApprovedStrategy (address (wrapStrategy));
125126 vm.stopPrank ();
126127 // non owner can't add new strategy
127- vm.expectRevert (bytes ( " Ownable: caller is not the owner " ));
128+ vm.expectRevert (abi.encodeWithSelector ( Ownable.OwnableUnauthorizedAccount. selector , address ( this ) ));
128129 manager.addApprovedStrategy (address (wrapStrategy));
129130 bool isStrategyApproved = manager.approvedStrategies (address (wrapStrategy));
130131 assertTrue (isStrategyApproved, "strategy should be register " );
@@ -135,7 +136,7 @@ contract ManagerTests is FoundrySuperfluidTester {
135136 //add strategy to be removed
136137 manager.addApprovedStrategy (address (wrapStrategy));
137138 // non owner can't add new strategy
138- vm.expectRevert (bytes ( " Ownable: caller is not the owner " ));
139+ vm.expectRevert (abi.encodeWithSelector ( Ownable.OwnableUnauthorizedAccount. selector , address ( this ) ));
139140 manager.removeApprovedStrategy (address (wrapStrategy));
140141 vm.startPrank (admin);
141142 vm.expectEmit (true , true , true , true );
0 commit comments