Testing Send ETH
This commit is contained in:
parent
35b6124869
commit
446edaca3e
|
@ -4,11 +4,15 @@ pragma solidity ^0.8.20;
|
||||||
contract Wallet {
|
contract Wallet {
|
||||||
address payable public owner;
|
address payable public owner;
|
||||||
|
|
||||||
|
event Deposit(address account, uint256 amount);
|
||||||
|
|
||||||
constructor () payable {
|
constructor () payable {
|
||||||
owner = payable(msg.sender);
|
owner = payable(msg.sender);
|
||||||
}
|
}
|
||||||
|
|
||||||
receive() external payable {}
|
receive() external payable {
|
||||||
|
emit Deposit(msg.sender, msg.value);
|
||||||
|
}
|
||||||
|
|
||||||
function withdraw(uint256 _amount) external {
|
function withdraw(uint256 _amount) external {
|
||||||
require(msg.sender == owner, "caller is not owner");
|
require(msg.sender == owner, "caller is not owner");
|
||||||
|
|
|
@ -0,0 +1,48 @@
|
||||||
|
// SPDX-License-Identifier: UNLICENSED
|
||||||
|
pragma solidity ^0.8.20;
|
||||||
|
|
||||||
|
import "forge-std/Test.sol";
|
||||||
|
import {Wallet} from "../src/Wallet.sol";
|
||||||
|
|
||||||
|
// Examples of deal and hoax
|
||||||
|
// deal(address, uint) - Set balance of address
|
||||||
|
// hoax(address, uint) - deal + prank, Sets up a prank and set balance
|
||||||
|
|
||||||
|
contract WalletTest is Test {
|
||||||
|
Wallet public wallet;
|
||||||
|
|
||||||
|
function setUp() public {
|
||||||
|
wallet = new Wallet{value: 1e18}();
|
||||||
|
}
|
||||||
|
|
||||||
|
function _send(uint256 amount) private {
|
||||||
|
(bool ok,) = address(wallet).call{value: amount}("");
|
||||||
|
require(ok, "send ETH failed");
|
||||||
|
}
|
||||||
|
|
||||||
|
function testEthBalance() public {
|
||||||
|
console.log("ETH balance", address(this).balance / 1e18);
|
||||||
|
}
|
||||||
|
|
||||||
|
function testSendEth() public {
|
||||||
|
uint bal = address(wallet).balance;
|
||||||
|
|
||||||
|
// deal(address, uint) - Set balance of address
|
||||||
|
deal(address(1), 100);
|
||||||
|
assertEq(address(1).balance, 100);
|
||||||
|
|
||||||
|
deal(address(1), 10);
|
||||||
|
assertEq(address(1).balance, 10);
|
||||||
|
|
||||||
|
// hoax(address, uint) - deal + prank, Sets up a prank and set balance
|
||||||
|
deal(address(1), 123e18);
|
||||||
|
vm.prank(address(1));
|
||||||
|
_send(123e18);
|
||||||
|
|
||||||
|
hoax(address(1), 456e18);
|
||||||
|
_send(456e18);
|
||||||
|
|
||||||
|
assertEq(address(wallet).balance, bal + 123e18 + 456e18);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue