55 lines
1.5 KiB
Solidity
55 lines
1.5 KiB
Solidity
pragma solidity 0.5.17;
|
|
|
|
import "../../Staking.sol";
|
|
|
|
import "../../lib/os/SafeMath.sol";
|
|
import "./TimeHelpersMock.sol";
|
|
|
|
|
|
contract StakingMock is Staking, TimeHelpersMock {
|
|
using SafeMath for uint256;
|
|
|
|
event LogGas(uint256 gas);
|
|
|
|
string private constant ERROR_TOKEN_NOT_CONTRACT = "STAKING_TOKEN_NOT_CONTRACT";
|
|
|
|
uint64 private constant MAX_UINT64 = uint64(-1);
|
|
|
|
modifier measureGas {
|
|
uint256 initialGas = gasleft();
|
|
_;
|
|
emit LogGas(initialGas - gasleft());
|
|
}
|
|
|
|
constructor(ERC20 _stakingToken) public {
|
|
require(isContract(address(_stakingToken)), ERROR_TOKEN_NOT_CONTRACT);
|
|
initialized();
|
|
stakingToken = _stakingToken;
|
|
}
|
|
|
|
function unlockedBalanceOfGas() external returns (uint256) {
|
|
uint256 initialGas = gasleft();
|
|
_unlockedBalanceOf(msg.sender);
|
|
uint256 gasConsumed = initialGas - gasleft();
|
|
emit LogGas(gasConsumed);
|
|
return gasConsumed;
|
|
}
|
|
|
|
function transferGas(address _to, address, uint256 _amount) external measureGas {
|
|
// have enough unlocked funds
|
|
require(_amount <= _unlockedBalanceOf(msg.sender));
|
|
|
|
_transfer(msg.sender, _to, _amount);
|
|
}
|
|
|
|
function setBlockNumber(uint64 _mockedBlockNumber) public {
|
|
mockedBlockNumber = _mockedBlockNumber;
|
|
}
|
|
|
|
// Override petrify functions to allow mocking the initialization process
|
|
function petrify() internal onlyInit {
|
|
// solium-disable-previous-line no-empty-blocks
|
|
// initializedAt(PETRIFIED_BLOCK);
|
|
}
|
|
}
|