Skip to content

Commit

Permalink
fix: cotroller timelock test fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Van0k committed May 14, 2023
1 parent 146b6d0 commit 7bedd08
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 18 deletions.
2 changes: 2 additions & 0 deletions contracts/test/mocks/GeneralMock.sol
Original file line number Diff line number Diff line change
Expand Up @@ -9,4 +9,6 @@ contract GeneralMock {
fallback() external {
data = msg.data;
}

receive() external payable {}
}
33 changes: 16 additions & 17 deletions contracts/test/unit/support/ControllerTimelock.t.sol
Original file line number Diff line number Diff line change
Expand Up @@ -684,10 +684,12 @@ contract ControllerTimelockTest is Test, IControllerTimelockEvents, IControllerT
pool, abi.encodeWithSelector(IPool4626.creditManagerBorrowed.selector, creditManager), abi.encode(0)
);

uint40 expirationDate = uint40(block.timestamp + 2 days);

Policy memory policy = Policy({
enabled: false,
flags: 1,
exactValue: block.timestamp + 5,
exactValue: expirationDate,
minValue: 0,
maxValue: 0,
referencePoint: 0,
Expand All @@ -702,15 +704,15 @@ contract ControllerTimelockTest is Test, IControllerTimelockEvents, IControllerT
vm.prank(CONFIGURATOR);
controllerTimelock.setPolicy(POLICY_CODE, policy);

uint40 expirationDate = uint40(block.timestamp + 1 days);

// VERIFY THAT THE FUNCTION IS QUEUED AND EXECUTED CORRECTLY
bytes32 txHash = keccak256(
abi.encode(creditConfigurator, "setExpirationDate(uint40)", abi.encode(block.timestamp + 5), expirationDate)
abi.encode(
creditConfigurator, "setExpirationDate(uint40)", abi.encode(expirationDate), block.timestamp + 1 days
)
);

vm.prank(admin);
controllerTimelock.setExpirationDate(creditManager, uint40(block.timestamp + 5));
controllerTimelock.setExpirationDate(creditManager, expirationDate);

vm.expectRevert(CallerNotAdminException.selector);

Expand All @@ -729,20 +731,17 @@ contract ControllerTimelockTest is Test, IControllerTimelockEvents, IControllerT

vm.warp(block.timestamp - 10 days);

// vm.mockCallRevert(
// creditConfigurator,
// abi.encodeWithSelector(
// ICreditConfigurator.setExpirationDate.selector,
// expirationDate
// ),
// abi.encode("error")
// );
vm.mockCallRevert(
creditConfigurator,
abi.encodeWithSelector(ICreditConfigurator.setExpirationDate.selector, expirationDate),
abi.encode("error")
);

// vm.expectRevert(TxExecutionRevertedException.selector);
// vm.prank(admin);
// controllerTimelock.executeTransaction(txHash);
vm.expectRevert(TxExecutionRevertedException.selector);
vm.prank(admin);
controllerTimelock.executeTransaction(txHash);

// vm.clearMockedCalls();
vm.clearMockedCalls();

vm.expectEmit(true, false, false, false);
emit ExecuteTransaction(txHash);
Expand Down

0 comments on commit 7bedd08

Please sign in to comment.