// SPDX-License-Identifier: MIT pragma solidity ^0.8.18; import {Test, console} from 'forge-std/Test.sol'; import {DeployTokenDivider} from 'script/DeployTokenDivider.s.sol'; import {TokenDivider} from 'src/TokenDivider.sol'; import {ERC721Mock} from '../mocks/ERC721Mock.sol'; import {ERC20Mock} from '@openzeppelin/contracts/mocks/token/ERC20Mock.sol'; import {Ownable} from '@openzeppelin/contracts/access/Ownable.sol'; contract TokenDiverTest is Test { DeployTokenDivider deployer; TokenDivider tokenDivider; ERC721Mock erc721Mock; address public USER = makeAddr("user"); address public USER2 = makeAddr("user2"); uint256 constant public STARTING_USER_BALANCE = 10e18; uint256 constant public AMOUNT = 2e18; uint256 constant public TOKEN_ID = 0; function setUp() public { deployer = new DeployTokenDivider(); tokenDivider = deployer.run(); erc721Mock = new ERC721Mock(); erc721Mock.mint(USER); // @auditor: user has the NFT vm.deal(USER2, STARTING_USER_BALANCE); // @auditor: user2 got 10 ether } function test_assertSetUp() public { // then what? // USER has no ERC721 NFT of TOKEN_ID assertTrue(address(erc721Mock.ownerOf(TOKEN_ID)) == address(USER2)); // USER has no balance // USER2 has STARTING_USER_BALANCE } /* function testDivideNft() public { vm.startPrank(USER); erc721Mock.approve(address(tokenDivider), TOKEN_ID); tokenDivider.divideNft(address(erc721Mock), TOKEN_ID, AMOUNT); vm.stopPrank(); ERC20Mock erc20Mock = ERC20Mock(tokenDivider.getErc20InfoFromNft(address(erc721Mock)).erc20Address); console.log("ERC20 Token name is: ", erc20Mock.name()); console.log("ERC20 Token symbol is: ", erc20Mock.symbol()); assertEq(tokenDivider.getErc20TotalMintedAmount(address(erc20Mock)), AMOUNT); assertEq(erc721Mock.ownerOf(TOKEN_ID), address(tokenDivider)); assertEq(tokenDivider.getBalanceOf(USER, address(erc20Mock)), AMOUNT); } modifier nftDivided() { vm.startPrank(USER); erc721Mock.approve(address(tokenDivider), TOKEN_ID); tokenDivider.divideNft(address(erc721Mock), TOKEN_ID, AMOUNT); vm.stopPrank(); _; } function testDivideNftFailsIsSenderIsNotNftOwner() public { vm.prank(USER); erc721Mock.approve(address(tokenDivider), TOKEN_ID); vm.startPrank(USER2); vm.expectRevert(TokenDivider.TokenDivider__NotFromNftOwner.selector); tokenDivider.divideNft(address(erc721Mock), TOKEN_ID, AMOUNT); vm.stopPrank(); } */ }