Update coin tests to use non forked environment - significantly speeding up test time. (#1426)

Migrated coin tests to use non-forked environment when testing, significantly speeding up test time.

Comparison:
* CoinUniV4.t.sol: before 214.26s, after 9.27s
* CreatorCoin.t.sol: before 36.68s, after 677.10ms

Author: oveddan

packages/coins/test/Coin.t.sol

@@ -16,7 +16,7 @@ contract CoinTest is BaseTest {
     using stdJson for string;
 
     function setUp() public override {
-        super.setUp();
+        super.setUpNonForked();
     }
 
     function test_contract_ierc165_support() public {
@@ -107,23 +107,35 @@ contract CoinTest is BaseTest {
     }
 
     function test_burn() public {
-        _deployV4Coin();
-        vm.deal(users.buyer, 1 ether);
-        vm.prank(users.buyer);
-        _swapSomeCurrencyForCoin(coinV4, address(weth), 1 ether, users.coinRecipient);
+        // Deploy a mock ERC20 currency
+        MockERC20 mockCurrency = new MockERC20("Mock Currency", "MOCK");
+        mockCurrency.mint(users.buyer, 1000 ether);
+        // Pool manager needs currency for liquidity operations
+        mockCurrency.mint(address(poolManager), 1000000 ether);
+
+        // Deploy coin with mock currency
+        coinV4 = ContentCoin(payable(address(_deployV4Coin(address(mockCurrency), address(0), bytes32(0)))));
 
-        uint256 beforeBalance = coinV4.balanceOf(users.coinRecipient);
+        // Approve with permit2 and swap
+        uint128 swapAmount = 1 ether;
+        vm.startPrank(users.buyer);
+        UniV4SwapHelper.approveTokenWithPermit2(permit2, address(router), address(mockCurrency), swapAmount, uint48(block.timestamp + 1 days));
+        vm.stopPrank();
+
+        _swapSomeCurrencyForCoin(coinV4, address(mockCurrency), swapAmount, users.buyer);
+
+        uint256 beforeBalance = coinV4.balanceOf(users.buyer);
         uint256 beforeTotalSupply = coinV4.totalSupply();
 
         uint256 burnAmount = beforeBalance / 2;
 
-        vm.prank(users.coinRecipient);
+        vm.prank(users.buyer);
         coinV4.burn(burnAmount);
 
-        uint256 afterBalance = coinV4.balanceOf(users.coinRecipient);
+        uint256 afterBalance = coinV4.balanceOf(users.buyer);
         uint256 afterTotalSupply = coinV4.totalSupply();
 
-        assertEq(beforeBalance - afterBalance, burnAmount, "coinRecipient coin balance");
+        assertEq(beforeBalance - afterBalance, burnAmount, "buyer coin balance");
         assertEq(beforeTotalSupply - afterTotalSupply, burnAmount, "coin total supply");
     }
 

packages/coins/test/CoinUniV4.t.sol

@@ -38,9 +38,8 @@ contract CoinUniV4Test is BaseTest {
     MockERC20 internal mockERC20B;
 
     function setUp() public override {
-        super.setUpWithBlockNumber(30267794);
+        super.setUpNonForked();
 
-        quoter = IV4Quoter(V4_QUOTER);
         mockERC20A = new MockERC20("MockERC20A", "MCKA");
         mockERC20B = new MockERC20("MockERC20B", "MCKB");
 

packages/coins/test/ContentCoinRewards.t.sol

@@ -21,9 +21,7 @@ contract ContentCoinRewardsTest is BaseTest {
     address internal tradeReferrer;
 
     function setUp() public override {
-        super.setUpWithBlockNumber(30267794);
-
-        deal(address(zoraToken), address(poolManager), 1_000_000_000e18);
+        super.setUpNonForked();
 
         backingCreatorCoin = CreatorCoin(_deployCreatorCoin());
 

packages/coins/test/CreatorCoin.t.sol

@@ -13,10 +13,7 @@ contract CreatorCoinTest is BaseTest {
     CreatorCoin internal creatorCoin;
 
     function setUp() public override {
-        super.setUpWithBlockNumber(30267794);
-
-        deal(address(zoraToken), address(poolManager), 1_000_000_000e18);
-
+        super.setUpNonForked();
         _deployCreatorCoin();
     }
 

packages/coins/test/CreatorCoinRewards.t.sol

@@ -22,9 +22,7 @@ contract CreatorCoinRewardsTest is BaseTest {
     address internal tradeReferrer;
 
     function setUp() public override {
-        super.setUpWithBlockNumber(30267794);
-
-        deal(address(zoraToken), address(poolManager), 1_000_000_000e18);
+        super.setUpNonForked();
 
         // Set up referrer addresses for all tests
         platformReferrer = makeAddr("platformReferrer");

packages/coins/test/Factory.t.sol

@@ -8,7 +8,7 @@ import {IZoraFactory} from "../src/interfaces/IZoraFactory.sol";
 
 contract FactoryTest is BaseTest {
     function setUp() public override {
-        super.setUp();
+        super.setUpNonForked();
     }
 
     function test_factory_constructor_and_proxy_setup() public {
@@ -115,8 +115,8 @@ contract FactoryTest is BaseTest {
 
         address platformReferrer = users.platformReferrer;
 
-        bytes memory poolConfig = CoinConfigurationVersions.defaultDopplerMultiCurveUniV4(address(weth));
-        bytes memory poolConfigForGettingAddress = poolConfigChanged ? CoinConfigurationVersions.defaultDopplerMultiCurveUniV4(address(0)) : poolConfig;
+        bytes memory poolConfig = CoinConfigurationVersions.defaultDopplerMultiCurveUniV4(address(0));
+        bytes memory poolConfigForGettingAddress = poolConfigChanged ? CoinConfigurationVersions.defaultDopplerMultiCurveUniV4(ZORA_TOKEN_ADDRESS) : poolConfig;
 
         address expectedCoinAddress = factory.coinAddress(msgSender, name, symbol, poolConfigForGettingAddress, platformReferrer, salt);
 

packages/coins/test/MultiOwnable.t.sol

@@ -5,7 +5,7 @@ import "./utils/BaseTest.sol";
 
 contract MultiOwnableTest is BaseTest {
     function setUp() public override {
-        super.setUp();
+        super.setUpNonForked();
 
         _deployV4Coin();
     }
@@ -135,15 +135,15 @@ contract MultiOwnableTest is BaseTest {
 
     function test_revert_init_with_zero_owners() public {
         address[] memory emptyOwners = new address[](0);
-        bytes memory poolConfig_ = _generatePoolConfig(address(weth));
+        bytes memory poolConfig_ = _generatePoolConfig(address(0));
         vm.expectRevert(MultiOwnable.OneOwnerRequired.selector);
         factory.deploy(users.creator, emptyOwners, "https://test.com", "Test Token", "TEST", poolConfig_, users.platformReferrer, 0);
     }
 
     function test_revert_init_with_zero_address() public {
         address[] memory owners = new address[](1);
         owners[0] = address(0);
-        bytes memory poolConfig_ = _generatePoolConfig(address(weth));
+        bytes memory poolConfig_ = _generatePoolConfig(address(0));
         vm.expectRevert(MultiOwnable.OwnerCannotBeAddressZero.selector);
         factory.deploy(users.creator, owners, "https://test.com", "Test Token", "TEST", poolConfig_, users.platformReferrer, 0);
     }
@@ -152,7 +152,7 @@ contract MultiOwnableTest is BaseTest {
         address[] memory owners = new address[](2);
         owners[0] = users.creator;
         owners[1] = users.creator;
-        bytes memory poolConfig_ = _generatePoolConfig(address(weth));
+        bytes memory poolConfig_ = _generatePoolConfig(address(0));
         vm.expectRevert(MultiOwnable.AlreadyOwner.selector);
         factory.deploy(users.creator, owners, "https://test.com", "Test Token", "TEST", poolConfig_, users.platformReferrer, 0);
     }

packages/coins/test/ZoraHookRegistry.t.sol

@@ -7,20 +7,28 @@ import {IZoraHookRegistry} from "../src/interfaces/IZoraHookRegistry.sol";
 import {ZoraHookRegistry} from "../src/hook-registry/ZoraHookRegistry.sol";
 
 contract MockHook {
-    function contractVersion() public pure returns (string memory) {
-        return "0.0.0";
+    string private _version;
+
+    constructor(string memory version_) {
+        _version = version_;
+    }
+
+    function contractVersion() public view returns (string memory) {
+        return _version;
     }
 }
 
+contract MockHookNoVersion {
+    // No contractVersion function
+}
+
 contract ZoraHookRegistryTest is Test {
-    uint256 internal forkId;
     address internal owner;
 
     ZoraHookRegistry internal zoraHookRegistry;
     MockHook internal mockHook;
 
     function setUp() public {
-        forkId = vm.createSelectFork("base", 34509280);
         owner = makeAddr("owner");
 
         address[] memory initialOwners = new address[](1);
@@ -29,7 +37,7 @@ contract ZoraHookRegistryTest is Test {
         zoraHookRegistry = new ZoraHookRegistry();
         zoraHookRegistry.initialize(initialOwners);
 
-        mockHook = new MockHook();
+        mockHook = new MockHook("0.0.0");
     }
 
     function test_register_hooks() public {
@@ -179,7 +187,8 @@ contract ZoraHookRegistryTest is Test {
         address[] memory hooks = new address[](1);
         string[] memory tags = new string[](1);
 
-        hooks[0] = 0x81542dC43Aff247eff4a0eceFC286A2973aE1040;
+        MockHook hookWithVersion = new MockHook("1.1.1");
+        hooks[0] = address(hookWithVersion);
         tags[0] = "CONTENT";
 
         vm.prank(owner);
@@ -192,7 +201,8 @@ contract ZoraHookRegistryTest is Test {
         address[] memory hooks = new address[](1);
         string[] memory tags = new string[](1);
 
-        hooks[0] = 0xA1eBdD5cA6470Bbd67114331387f2dDa7bfad040;
+        MockHookNoVersion hookNoVersion = new MockHookNoVersion();
+        hooks[0] = address(hookNoVersion);
         tags[0] = "CONTENT";
 
         vm.prank(owner);
@@ -207,8 +217,8 @@ contract ZoraHookRegistryTest is Test {
 
     function test_get_hook_addresses_multiple_and_remove_middle() public {
         address a = address(mockHook);
-        address b = address(new MockHook());
-        address c = address(new MockHook());
+        address b = address(new MockHook("0.0.0"));
+        address c = address(new MockHook("0.0.0"));
 
         address[] memory hooks = new address[](3);
         string[] memory tags = new string[](3);

packages/coins/test/mocks/MockAirlock.sol

@@ -0,0 +1,22 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.13;
+
+import {IAirlock} from "../../src/interfaces/IAirlock.sol";
+
+/// @title MockAirlock
+/// @notice Mock implementation of IAirlock for testing purposes
+contract MockAirlock is IAirlock {
+    address private _owner;
+
+    constructor(address owner_) {
+        _owner = owner_;
+    }
+
+    function owner() external view override returns (address) {
+        return _owner;
+    }
+
+    function setOwner(address newOwner) external {
+        _owner = newOwner;
+    }
+}

packages/coins/test/mocks/SimpleERC20.sol

@@ -0,0 +1,8 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.23;
+
+import {ERC20} from "@openzeppelin/contracts/token/ERC20/ERC20.sol";
+
+contract SimpleERC20 is ERC20 {
+    constructor(string memory name, string memory symbol) ERC20(name, symbol) {}
+}