forked from crytic/properties
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathERC20PausableProperties.sol
89 lines (75 loc) · 2.62 KB
/
ERC20PausableProperties.sol
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
// SPDX-License-Identifier: AGPL-3.0-or-later
pragma solidity ^0.8.13;
import "../util/ERC20TestBase.sol";
import "@openzeppelin/contracts/token/ERC20/extensions/ERC20Pausable.sol";
abstract contract CryticERC20PausableProperties is
CryticERC20Base,
ERC20Pausable
{
constructor() {}
////////////////////////////////////////
// Helper functions - May need tweaking for non-OZ tokens
function _overridePause(bool paused) internal {
if (paused) {
_pause();
} else {
_unpause();
}
}
// Override for pausable tokens
function _beforeTokenTransfer(
address from,
address to,
uint256 amount
) internal virtual override(ERC20, ERC20Pausable) {
ERC20Pausable._beforeTokenTransfer(from, to, amount);
}
////////////////////////////////////////
// Properties
// Transfers should not be possible during paused state
function test_ERC20_pausedTransfer(address target, uint256 amount) public {
uint256 balance_sender = balanceOf(address(this));
uint256 balance_receiver = balanceOf(target);
require(balance_sender > 0);
uint256 transfer_amount = amount % (balance_sender + 1);
_pause();
bool r = this.transfer(target, transfer_amount);
assertWithMsg(r == false, "Tokens transferred while paused");
assertEq(
balanceOf(address(this)),
balance_sender,
"Transfer while paused altered source balance"
);
assertEq(
balanceOf(target),
balance_receiver,
"Transfer while paused altered target balance"
);
_unpause();
}
// Transfers should not be possible during paused state
function test_ERC20_pausedTransferFrom(
address target,
uint256 amount
) public {
uint256 balance_sender = balanceOf(msg.sender);
uint256 balance_receiver = balanceOf(target);
uint256 allowance = allowance(msg.sender, address(this));
require(balance_sender > 0 && allowance > balance_sender);
uint256 transfer_amount = amount % (balance_sender + 1);
_pause();
bool r = this.transferFrom(msg.sender, target, transfer_amount);
assertWithMsg(r == false, "Tokens transferred while paused");
assertEq(
balanceOf(msg.sender),
balance_sender,
"Transfer while paused altered source balance"
);
assertEq(
balanceOf(target),
balance_receiver,
"Transfer while paused altered target balance"
);
_unpause();
}
}