mirror of https://github.com/vacp2p/minime.git
206 lines
6.7 KiB
Solidity
206 lines
6.7 KiB
Solidity
// SPDX-License-Identifier: UNLICENSED
|
|
pragma solidity ^0.8.19;
|
|
|
|
import { Test } from "forge-std/Test.sol";
|
|
import { Deploy } from "../script/Deploy.s.sol";
|
|
import { DeploymentConfig } from "../script/DeploymentConfig.s.sol";
|
|
|
|
import { MiniMeToken } from "../contracts/MiniMeToken.sol";
|
|
import { MiniMeTokenFactory } from "../contracts/MiniMeToken.sol";
|
|
|
|
contract MiniMeTokenTest is Test {
|
|
DeploymentConfig internal deploymentConfig;
|
|
MiniMeTokenFactory internal minimeTokenFactory;
|
|
MiniMeToken internal minimeToken;
|
|
|
|
address internal deployer;
|
|
|
|
address[] internal accounts;
|
|
|
|
function setUp() public virtual {
|
|
Deploy deployment = new Deploy();
|
|
(deploymentConfig, minimeTokenFactory, minimeToken) = deployment.run();
|
|
(deployer,,,,,,) = deploymentConfig.activeNetworkConfig();
|
|
|
|
accounts = new address[](4);
|
|
accounts[0] = makeAddr("account0");
|
|
accounts[1] = makeAddr("account1");
|
|
accounts[2] = makeAddr("account2");
|
|
accounts[3] = makeAddr("account3");
|
|
}
|
|
|
|
function testDeployment() public {
|
|
(, address parentToken, uint256 parentSnapShotBlock, string memory name, uint8 decimals, string memory symbol,)
|
|
= deploymentConfig.activeNetworkConfig();
|
|
assertEq(minimeToken.name(), name);
|
|
assertEq(minimeToken.symbol(), symbol);
|
|
assertEq(minimeToken.decimals(), decimals);
|
|
assertEq(minimeToken.controller(), deployer);
|
|
assertEq(address(minimeToken.parentToken()), parentToken);
|
|
assertEq(minimeToken.parentSnapShotBlock(), parentSnapShotBlock);
|
|
}
|
|
|
|
function _generateTokens(address to, uint256 amount) internal {
|
|
vm.prank(deployer);
|
|
minimeToken.generateTokens(to, amount);
|
|
}
|
|
}
|
|
|
|
contract GenerateTokensTest is MiniMeTokenTest {
|
|
function setUp() public virtual override {
|
|
MiniMeTokenTest.setUp();
|
|
}
|
|
|
|
function test_RevertWhen_SenderIsNotController() public {
|
|
vm.expectRevert();
|
|
minimeToken.generateTokens(accounts[0], 10);
|
|
}
|
|
|
|
function testGenerateTokens() public {
|
|
_generateTokens(accounts[0], 10);
|
|
|
|
assertEq(minimeToken.totalSupply(), 10);
|
|
assertEq(minimeToken.balanceOf(accounts[0]), 10);
|
|
}
|
|
}
|
|
|
|
contract TransferTest is MiniMeTokenTest {
|
|
function setUp() public virtual override {
|
|
MiniMeTokenTest.setUp();
|
|
}
|
|
|
|
function testTransfer() public {
|
|
uint256 currentBlock = block.number;
|
|
uint256 nextBlock = currentBlock + 1;
|
|
|
|
_generateTokens(accounts[0], 10);
|
|
|
|
// enforce the next block
|
|
vm.roll(nextBlock);
|
|
|
|
vm.prank(accounts[0]);
|
|
minimeToken.transfer(accounts[1], 2);
|
|
|
|
assertEq(minimeToken.totalSupply(), 10);
|
|
assertEq(minimeToken.balanceOf(accounts[0]), 8);
|
|
assertEq(minimeToken.balanceOf(accounts[1]), 2);
|
|
|
|
// check balance at original block
|
|
assertEq(minimeToken.balanceOfAt(accounts[0], currentBlock), 10);
|
|
}
|
|
}
|
|
|
|
contract AllowanceTest is MiniMeTokenTest {
|
|
function setUp() public virtual override {
|
|
MiniMeTokenTest.setUp();
|
|
}
|
|
|
|
function testAllowance() public {
|
|
vm.prank(accounts[0]);
|
|
minimeToken.approve(accounts[1], 2);
|
|
|
|
uint256 allowed = minimeToken.allowance(accounts[0], accounts[1]);
|
|
assertEq(allowed, 2);
|
|
|
|
uint256 currentBlock = block.number;
|
|
uint256 nextBlock = currentBlock + 1;
|
|
|
|
// ensure `accounts[0]` has tokens
|
|
_generateTokens(accounts[0], 10);
|
|
|
|
// enforce the next block
|
|
vm.roll(nextBlock);
|
|
|
|
vm.prank(accounts[1]);
|
|
minimeToken.transferFrom(accounts[0], accounts[2], 1);
|
|
|
|
allowed = minimeToken.allowance(accounts[0], accounts[1]);
|
|
assertEq(allowed, 1);
|
|
|
|
assertEq(minimeToken.totalSupply(), 10);
|
|
assertEq(minimeToken.balanceOf(accounts[0]), 9);
|
|
assertEq(minimeToken.balanceOf(accounts[2]), 1);
|
|
|
|
// check balance at blocks
|
|
assertEq(minimeToken.balanceOfAt(accounts[0], currentBlock), 10);
|
|
assertEq(minimeToken.balanceOfAt(accounts[0], nextBlock), 9);
|
|
assertEq(minimeToken.balanceOfAt(accounts[2], nextBlock), 1);
|
|
}
|
|
}
|
|
|
|
contract DestroyTokensTest is MiniMeTokenTest {
|
|
function setUp() public virtual override {
|
|
MiniMeTokenTest.setUp();
|
|
}
|
|
|
|
function testDestroyTokens() public {
|
|
// ensure `accounts[0]` has tokens
|
|
_generateTokens(accounts[0], 10);
|
|
|
|
vm.prank(deployer);
|
|
minimeToken.destroyTokens(accounts[0], 3);
|
|
|
|
assertEq(minimeToken.totalSupply(), 7);
|
|
assertEq(minimeToken.balanceOf(accounts[0]), 7);
|
|
}
|
|
}
|
|
|
|
contract CreateCloneTokenTest is MiniMeTokenTest {
|
|
function setUp() public virtual override {
|
|
MiniMeTokenTest.setUp();
|
|
}
|
|
|
|
function _createClone() internal returns (MiniMeToken) {
|
|
address cloneAddress = minimeToken.createCloneToken("Clone Token 1", 18, "MMTc", 0, true);
|
|
MiniMeToken clone = MiniMeToken(payable(cloneAddress));
|
|
return clone;
|
|
}
|
|
|
|
function testCreateCloneToken() public {
|
|
// fund some accounts to later check if cloned token has same balances
|
|
uint256 currentBlock = block.number;
|
|
_generateTokens(accounts[0], 7);
|
|
uint256 nextBlock = block.number + 1;
|
|
vm.roll(nextBlock);
|
|
_generateTokens(accounts[1], 3);
|
|
uint256 secondNextBlock = block.number + 2;
|
|
vm.roll(secondNextBlock);
|
|
_generateTokens(accounts[2], 5);
|
|
|
|
MiniMeToken clone = _createClone();
|
|
|
|
assertEq(address(clone.parentToken()), address(minimeToken));
|
|
assertEq(clone.parentSnapShotBlock(), block.number);
|
|
assertEq(clone.totalSupply(), 15);
|
|
assertEq(clone.balanceOf(accounts[0]), 7);
|
|
assertEq(clone.balanceOf(accounts[1]), 3);
|
|
assertEq(clone.balanceOf(accounts[2]), 5);
|
|
|
|
assertEq(clone.totalSupplyAt(currentBlock), 7);
|
|
assertEq(clone.totalSupplyAt(nextBlock), 10);
|
|
|
|
assertEq(clone.balanceOfAt(accounts[0], currentBlock), 7);
|
|
assertEq(clone.balanceOfAt(accounts[1], currentBlock), 0);
|
|
assertEq(clone.balanceOfAt(accounts[2], currentBlock), 0);
|
|
|
|
assertEq(clone.balanceOfAt(accounts[0], nextBlock), 7);
|
|
assertEq(clone.balanceOfAt(accounts[1], nextBlock), 3);
|
|
assertEq(clone.balanceOfAt(accounts[2], nextBlock), 0);
|
|
|
|
assertEq(clone.balanceOfAt(accounts[0], secondNextBlock), 7);
|
|
assertEq(clone.balanceOfAt(accounts[1], secondNextBlock), 3);
|
|
assertEq(clone.balanceOfAt(accounts[2], secondNextBlock), 5);
|
|
}
|
|
|
|
function testGenerateTokens() public {
|
|
_generateTokens(accounts[0], 10);
|
|
|
|
vm.prank(deployer);
|
|
MiniMeToken clone = _createClone();
|
|
assertEq(clone.totalSupply(), 10);
|
|
|
|
vm.prank(deployer);
|
|
clone.generateTokens(accounts[0], 5);
|
|
}
|
|
}
|