snekmate-erc20-mocks

tests/fixtures/tokens.py

@@ -2,46 +2,40 @@
 import pytest
 
 
-@pytest.fixture(scope="module")
-def weth(deployer):
-    with boa.env.prank(deployer):
-        return boa.load("contracts/mocks/WETH.vy")
-
-
 @pytest.fixture(scope="module")
 def usd(deployer):
     with boa.env.prank(deployer):
-        return boa.load("contracts/mocks/ERC20Mock.vy", "USD", "USD", 18)
+        return boa.load("tests/mocks/MockERC20.vy", "USD", "USD", 18)
 
 
 @pytest.fixture(scope="module")
 def btc(deployer):
     with boa.env.prank(deployer):
-        return boa.load("contracts/mocks/ERC20Mock.vy", "BTC", "BTC", 18)
+        return boa.load("tests/mocks/MockERC20.vy", "BTC", "BTC", 18)
 
 
 @pytest.fixture(scope="module")
 def stg(deployer):
     with boa.env.prank(deployer):
-        return boa.load("contracts/mocks/ERC20Mock.vy", "STG", "STG", 18)
+        return boa.load("tests/mocks/MockERC20.vy", "STG", "STG", 18)
 
 
 @pytest.fixture(scope="module")
 def usdt(deployer):
     with boa.env.prank(deployer):
-        return boa.load("contracts/mocks/ERC20Mock.vy", "USDT", "USDT", 6)
+        return boa.load("tests/mocks/MockERC20.vy", "USDT", "USDT", 6)
 
 
 @pytest.fixture(scope="module")
 def usdc(deployer):
     with boa.env.prank(deployer):
-        return boa.load("contracts/mocks/ERC20Mock.vy", "USDC", "USDC", 6)
+        return boa.load("tests/mocks/MockERC20.vy", "USDC", "USDC", 6)
 
 
 @pytest.fixture(scope="module")
 def dai(deployer):
     with boa.env.prank(deployer):
-        return boa.load("contracts/mocks/ERC20Mock.vy", "DAI", "DAI", 18)
+        return boa.load("tests/mocks/MockERC20.vy", "DAI", "DAI", 18)
 
 
 @pytest.fixture(scope="module")

tests/integration/test_create2_deployment.py

@@ -159,7 +159,7 @@ def factory(
 
 @pytest.fixture(scope="module")
 def coins():
-    erc20_mock = boa.load_partial("./contracts/mocks/ERC20Mock.vy")
+    erc20_mock = boa.load_partial("./tests/mocks/MockERC20.vy")
     return [
         erc20_mock.at("0xf939E0A03FB07F59A73314E73794Be0E57ac1b4E"),
         erc20_mock.at("0xD533a949740bb3306d119CC777fa900bA034cd52"),

tests/mocks/MockERC20.vy

@@ -0,0 +1,15 @@
+# pragma version ~=0.4.0
+
+from snekmate.tokens import erc20
+from snekmate.auth import ownable
+
+initializes: ownable
+initializes: erc20[ownable := ownable]
+
+exports: erc20.__interface__
+
+
+@deploy
+def __init__(decimals: uint8):
+    ownable.__init__()
+    erc20.__init__("mock", "mock", decimals, "mock", "mock")

tests/mocks/MockERC20Fee.vy

@@ -0,0 +1,49 @@
+# pragma version ~=0.4.0
+
+from snekmate.tokens import erc20
+from snekmate.auth import ownable
+
+from ethereum.ercs import IERC20
+
+initializes: ownable
+initializes: erc20[ownable := ownable]
+
+exports: (erc20.approve, erc20.transferFrom)
+
+MAX_FEE: constant(uint256) = 10_000
+FEE_PERCENTAGE: public(immutable(uint256)) 
+
+@deploy
+def __init__(decimals: uint8, fee: uint256):
+    ownable.__init__()
+    erc20.__init__("mock", "mock", decimals, "mock", "mock")
+    assert fee <= MAX_FEE, "fee must be less than or equal to 100%"
+    FEE_PERCENTAGE = fee
+
+
+@internal
+def _transfer(owner: address, to: address, amount: uint256):
+    # Adapted from snekmate to have fee on transfer
+
+    assert owner != empty(address), "erc20: transfer from the zero address"
+    assert to != empty(address), "erc20: transfer to the zero address"
+
+    fee: uint256 = amount * FEE_PERCENTAGE // MAX_FEE
+    amount -= fee
+
+    erc20._burn(owner, fee)
+
+    owner_balance: uint256 = erc20.balanceOf[owner]
+    assert owner_balance >= amount, "erc20: transfer amount exceeds balance"
+    # Deduct full amount from sender
+    erc20.balanceOf[owner] = unsafe_sub(owner_balance, amount)
+    # Credit recipient with full amount first
+    erc20.balanceOf[to] = unsafe_add(erc20.balanceOf[to], amount)
+    log IERC20.Transfer(owner, to, amount)
+
+
+@external
+def transfer(to: address, amount: uint256) -> bool:
+    # Adapted from snekmate to have fee on transfer
+    self._transfer(msg.sender, to, amount)
+    return True