Skip to content

Commit 22ad0e7

Browse files
committed
improved mock tests
1 parent c94e921 commit 22ad0e7

File tree

2 files changed

+208
-97
lines changed

2 files changed

+208
-97
lines changed

test/mocks/MockFailedTransferFrom.sol

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ import {Ownable} from "@openzeppelin/contracts/access/Ownable.sol";
77
contract MockFailedTransferFrom is ERC20Burnable, Ownable {
88
error MockFailedTransferFrom__AmountMustBeMoreThanZero();
99
error MockFailedTransferFrom__BurnAmountExceedsBalance();
10-
error MockFailedTransferFrom__NotZeroAddress();
10+
error MockFailedTransferFrom__CanNotBeZeroAddress();
1111

1212
/*
1313
In future versions of OpenZeppelin contracts package, Ownable must be declared with an address of the contract owner
@@ -30,8 +30,11 @@ contract MockFailedTransferFrom is ERC20Burnable, Ownable {
3030
super.burn(_amount);
3131
}
3232

33-
function mint(address account, uint256 amount) public {
34-
_mint(account, amount);
33+
function mint(address _to, uint256 _amount) external onlyOwner returns (bool) {
34+
if (_to == address(0)) revert MockFailedTransferFrom__CanNotBeZeroAddress();
35+
if (_amount <= 0) revert MockFailedTransferFrom__AmountMustBeMoreThanZero();
36+
_mint(_to, _amount);
37+
return true;
3538
}
3639

3740
function transferFrom(address, /*sender*/ address, /*recipient*/ uint256 /*amount*/ )

0 commit comments

Comments
 (0)