diff --git a/test/unit/TokenDividerTest.audit.divideNft.t.sol b/test/unit/TokenDividerTest.audit.divideNft.t.sol index a03332e..54d50b8 100644 --- a/test/unit/TokenDividerTest.audit.divideNft.t.sol +++ b/test/unit/TokenDividerTest.audit.divideNft.t.sol @@ -31,13 +31,14 @@ contract TokenDiverTest is Test { vm.deal(USER2, STARTING_USER_BALANCE); // @auditor: user2 got 10 ether } - function test_assertSetUp() public { + function test_assertSetUp() public view { // 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 + // USER has the ERC721 NFT of TOKEN_ID + assertTrue(address(erc721Mock.ownerOf(TOKEN_ID)) == address(USER)); + + // USER2 has no ERC721 NFT of TOKEN_ID + assertTrue(address(erc721Mock.ownerOf(TOKEN_ID)) != address(USER2)); } /*