ERC20.t.sol.gyb

%{ 
  import yaml
  with open('../test-cases.yml', 'r') as file:
    config = yaml.safe_load(file)
    contracts = config["contracts"]
}%
// SPDX-License-Identifier: UNLICENSED
pragma solidity >=0.8.4;

import {DSTest} from "@ds-test/test.sol";
import {Vm} from "@forge-std/Vm.sol";
% for variation in contracts["erc20"]["variations"]:
import {ERC20_${variation}} from "$/ERC20_${variation}.sol";
% end
% for variation in contracts["erc20"]["variations"]:

// deploy
contract ERC20_${variation}_deploy_Test is DSTest {
    function test_deploy() public {
        ERC20_${variation} sut = new ERC20_${variation}();
    }
}

// transferToNonOwner
contract ERC20_${variation}_transferToNonOwner_Test is DSTest {
    Vm internal constant HEVM = Vm(HEVM_ADDRESS);

    ERC20_${variation} internal sut;

    function setUp() public {
        sut = new ERC20_${variation}();
        sut.mint(address(0xAAAA), 1000 ether);
    }

    function test_transferToNonOwner() public {
        HEVM.prank(address(0xAAAA));
        sut.transfer(address(0xBBBB), 500 ether);
    }
}

// transferToOwner
contract ERC20_${variation}_transferToOwner_Test is DSTest {
    Vm internal constant HEVM = Vm(HEVM_ADDRESS);

    ERC20_${variation} internal sut;

    function setUp() public {
        sut = new ERC20_${variation}();
        sut.mint(address(0xAAAA), 1000 ether);
        sut.mint(address(0xBBBB), 1000 ether);
    }

    function test_transferToOwner() public {
        HEVM.prank(address(0xAAAA));
        sut.transfer(address(0xBBBB), 500 ether);
    }
}

// transferFromToNonOwner
contract ERC20_${variation}_transferFromToNonOwner_Test is DSTest {
    Vm internal constant HEVM = Vm(HEVM_ADDRESS);

    ERC20_${variation} internal sut;

    function setUp() public {
        sut = new ERC20_${variation}();
        sut.mint(address(0xAAAA), 1000 ether);
        HEVM.prank(address(0xAAAA));
        sut.approve(address(0xBBBB), 1000 ether);
    }

    function test_transferFromToNonOwner() public {
        HEVM.prank(address(0xBBBB));
        sut.transferFrom(address(0xAAAA), address(0xCCCC), 500 ether);
    }
}

// transferFromToOwner
contract ERC20_${variation}_transferFromToOwner_Test is DSTest {
    Vm internal constant HEVM = Vm(HEVM_ADDRESS);

    ERC20_${variation} internal sut;

    function setUp() public {
        sut = new ERC20_${variation}();
        sut.mint(address(0xAAAA), 1000 ether);
        sut.mint(address(0xCCCC), 1000 ether);
        HEVM.prank(address(0xAAAA));
        sut.approve(address(0xBBBB), 1000 ether);
    }

    function test_transferFromToOwner() public {
        HEVM.prank(address(0xBBBB));
        sut.transferFrom(address(0xAAAA), address(0xCCCC), 500 ether);
    }
}

// approve
contract ERC20_${variation}_approve_Test is DSTest {
    Vm internal constant HEVM = Vm(HEVM_ADDRESS);

    ERC20_${variation} internal sut;

    function setUp() public {
        sut = new ERC20_${variation}();
        sut.mint(address(0xAAAA), 1000 ether);
    }

    function test_approve() public {
        HEVM.prank(address(0xAAAA));
        sut.approve(address(0xBBBB), 1000 ether);
    }
}

// totalSupply
contract ERC20_${variation}_totalSupply_Test is DSTest {
    Vm internal constant HEVM = Vm(HEVM_ADDRESS);

    ERC20_${variation} internal sut;

    function setUp() public {
        sut = new ERC20_${variation}();
        sut.mint(address(0xAAAA), 1000 ether);
    }

    function test_totalSupply() public view {
        sut.totalSupply();
    }
}

// balanceOf
contract ERC20_${variation}_balanceOf_Test is DSTest {
    Vm internal constant HEVM = Vm(HEVM_ADDRESS);

    ERC20_${variation} internal sut;

    function setUp() public {
        sut = new ERC20_${variation}();
        sut.mint(address(0xAAAA), 1000 ether);
    }

    function test_balanceOf() public view {
        sut.balanceOf(address(0xAAAA));
    }
}

// allowance
contract ERC20_${variation}_allowance_Test is DSTest {
    Vm internal constant HEVM = Vm(HEVM_ADDRESS);

    ERC20_${variation} internal sut;

    function setUp() public {
        sut = new ERC20_${variation}();
        sut.mint(address(0xAAAA), 1000 ether);
        HEVM.prank(address(0xAAAA));
        sut.approve(address(0xBBBB), 1000 ether);
    }

    function test_allowance() public view {
        sut.allowance(address(0xAAAA), address(0xBBBB));
    }
}
% end