feat: update token test

mektigboy · Feb 20, 2023 · fe1d15d · fe1d15d
1 parent bdb8b8e
commit fe1d15d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/Token.t.sol b/test/Token.t.sol
@@ -16,12 +16,16 @@ contract Token is Test {
     IToken tokenInterface;
 
     function setUp() public {
-        vm.prank(OWNER);
+        vm.startPrank(OWNER);
+
         address tokenAddress = HuffDeployer
             .config()
             .with_args(abi.encodePacked(OWNER, abi.encode(21000000)))
             .deploy("Token");
         tokenInterface = IToken(tokenAddress);
+
+        tokenInterface.transfer(ATTACKER, 20);
+        vm.stopPrank();
     }
 
     function test_totalSupply() public {