@zoralabs/coins 0.9.0 → 1.0.1

package/test/CoinDopplerUniV3.t.sol

@@ -9,15 +9,10 @@ import {CoinConstants} from "../src/libs/CoinConstants.sol";
 import {IUniswapV3Pool} from "../src/interfaces/IUniswapV3Pool.sol";
 import {LpPosition} from "../src/types/LpPosition.sol";
 import {IDopplerErrors} from "../src/interfaces/IDopplerErrors.sol";
-import {CoinDopplerUniV3} from "../src/libs/CoinDopplerUniV3.sol";
+import {DopplerMath, CoinDopplerUniV3} from "../src/libs/CoinDopplerUniV3.sol";
 import {TickMath} from "../src/utils/uniswap/TickMath.sol";
 
 contract DopplerUniswapV3Test is BaseTest {
-    int24 internal constant DEFAULT_DISCOVERY_TICK_LOWER = -777000;
-    int24 internal constant DEFAULT_DISCOVERY_TICK_UPPER = 222000;
-    uint16 internal constant DEFAULT_NUM_DISCOVERY_POSITIONS = 10; // will be 11 total with tail position
-    uint256 internal constant DEFAULT_DISCOVERY_SUPPLY_SHARE = 0.495e18; // half of the 990m total pool supply
-
     function _deployCoin(bytes memory poolConfig_) internal {
         vm.prank(users.creator);
         (address coinAddress, ) = factory.deploy(
@@ -42,134 +37,7 @@ contract DopplerUniswapV3Test is BaseTest {
         super.setUp();
     }
 
-    function test_deploy_legacy_eth_config() public {
-        bytes memory poolConfig = _generatePoolConfig(
-            CoinConfigurationVersions.LEGACY_POOL_VERSION,
-            address(weth),
-            MarketConstants.LP_TICK_LOWER_WETH,
-            0,
-            1,
-            0
-        );
-
-        _deployCoin(poolConfig);
-
-        assertEq(coin.currency(), address(weth), "currency");
-        assertEq(coin.totalSupply(), 1_000_000_000e18, "totalSupply");
-        assertEq(coin.balanceOf(users.creator), 10_000_000e18, "balanceOf creator");
-        assertGt(coin.balanceOf(coin.poolAddress()), 989_999_999e18, "balanceOf pool");
-
-        (
-            address asset,
-            address numeraire,
-            int24 tickLower,
-            int24 tickUpper,
-            uint16 numPositions,
-            bool isInitialized,
-            bool isExited,
-            uint256 maxShareToBeSold,
-            uint256 totalTokensOnBondingCurve
-        ) = coin.poolState();
-
-        assertEq(asset, address(coin));
-        assertEq(numeraire, address(weth));
-        assertEq(numPositions, 1);
-        assertTrue(isInitialized);
-        assertFalse(isExited);
-        assertEq(maxShareToBeSold, 0);
-        assertEq(totalTokensOnBondingCurve, CoinConstants.POOL_LAUNCH_SUPPLY);
-
-        bool isCoinToken0 = address(coin) < address(weth);
-
-        if (isCoinToken0) {
-            assertEq(tickLower, MarketConstants.LP_TICK_LOWER_WETH);
-            assertEq(tickUpper, MarketConstants.LP_TICK_UPPER);
-        } else {
-            assertEq(tickLower, -MarketConstants.LP_TICK_UPPER);
-            assertEq(tickUpper, -MarketConstants.LP_TICK_LOWER_WETH);
-        }
-    }
-
-    function test_invalid_tick_range() public {
-        bytes memory poolConfig = _generatePoolConfig(CoinConfigurationVersions.LEGACY_POOL_VERSION, address(weth), -100, 100, 100, 0.5e18);
-
-        vm.expectRevert();
-        factory.deploy(users.creator, _getDefaultOwners(), "https://test.com", "Testcoin", "TEST", poolConfig, users.platformReferrer, 0);
-    }
-
-    function test_inverted_tick_range_revert() public {
-        // These tick ranges are flipped
-        bytes memory poolConfig = _generatePoolConfig(CoinConfigurationVersions.LEGACY_POOL_VERSION, address(weth), 10000, -10000, 100, 0.5e18);
-
-        vm.expectRevert(abi.encodeWithSignature("InvalidWethLowerTick()"));
-        factory.deploy(users.creator, _getDefaultOwners(), "https://test.com", "Testcoin", "TEST", poolConfig, users.platformReferrer, 0);
-    }
-
-    function test_deploy_legacy_eth_config_with_prebuy(uint256 initialOrderSize) public {
-        vm.assume(initialOrderSize > CoinConstants.MIN_ORDER_SIZE);
-        vm.assume(initialOrderSize < 10 ether);
-
-        vm.deal(users.creator, initialOrderSize);
-
-        bytes memory poolConfig = _generatePoolConfig(
-            CoinConfigurationVersions.LEGACY_POOL_VERSION,
-            address(weth),
-            MarketConstants.LP_TICK_LOWER_WETH,
-            0,
-            0,
-            0
-        );
-
-        vm.prank(users.creator);
-        (address coinAddress, ) = factory.deploy{value: initialOrderSize}(
-            users.creator,
-            _getDefaultOwners(),
-            "https://test.com",
-            "Testcoin",
-            "TEST",
-            poolConfig,
-            users.platformReferrer,
-            initialOrderSize
-        );
-
-        coin = Coin(payable(coinAddress));
-        pool = IUniswapV3Pool(coin.poolAddress());
-
-        assertGt(coin.balanceOf(users.creator), 10_000_000e18, "balanceOf creator");
-    }
-
-    function test_deploy_legacy_usdc_config_with_prebuy() public {
-        address[] memory owners = new address[](1);
-        owners[0] = users.creator;
-
-        uint256 orderSize = dealUSDC(users.creator, 100);
-
-        vm.prank(users.creator);
-        usdc.approve(address(factory), orderSize);
-
-        bytes memory poolConfig = _generatePoolConfig(CoinConfigurationVersions.LEGACY_POOL_VERSION, address(usdc), USDC_TICK_LOWER, 0, 0, 0);
-
-        vm.prank(users.creator);
-        (address coinAddress, uint256 coinsPurchased) = factory.deploy(
-            users.creator,
-            owners,
-            "https://testcoinusdcpair.com",
-            "Testcoinusdcpair",
-            "TESTCOINUSDCPAIR",
-            poolConfig,
-            users.platformReferrer,
-            orderSize
-        );
-        coin = Coin(payable(coinAddress));
-        pool = IUniswapV3Pool(coin.poolAddress());
-        vm.label(address(coin), "COIN");
-        vm.label(address(pool), "POOL");
-
-        assertEq(coin.currency(), address(usdc), "currency");
-        assertEq(coin.balanceOf(users.creator), CoinConstants.CREATOR_LAUNCH_REWARD + coinsPurchased);
-    }
-
-    function test_deploy_doppler_eth() public {
+    function test_supply_constants() public {
         bytes memory poolConfig = _generatePoolConfig(
             CoinConfigurationVersions.DOPPLER_UNI_V3_POOL_VERSION,
             address(weth),
@@ -180,26 +48,15 @@ contract DopplerUniswapV3Test is BaseTest {
         );
 
         _deployCoin(poolConfig);
+        assertEq(CoinConstants.MAX_TOTAL_SUPPLY, CoinConstants.POOL_LAUNCH_SUPPLY + CoinConstants.CREATOR_LAUNCH_REWARD);
+
+        assertEq(CoinConstants.MAX_TOTAL_SUPPLY, 1_000_000_000e18);
+        assertEq(CoinConstants.POOL_LAUNCH_SUPPLY, 990_000_000e18);
+        assertEq(CoinConstants.CREATOR_LAUNCH_REWARD, 10_000_000e18);
 
-        (
-            address asset,
-            address numeraire,
-            ,
-            ,
-            uint16 numPositions,
-            bool isInitialized,
-            bool isExited,
-            uint256 maxShareToBeSold,
-            uint256 totalTokensOnBondingCurve
-        ) = coin.poolState();
-
-        assertEq(asset, address(coin), "poolState.asset");
-        assertEq(numeraire, address(weth), "poolState.numeraire");
-        assertEq(numPositions, DEFAULT_NUM_DISCOVERY_POSITIONS, "poolState.numPositions");
-        assertTrue(isInitialized, "poolState.isInitialized");
-        assertFalse(isExited, "poolState.isExited");
-        assertEq(maxShareToBeSold, DEFAULT_DISCOVERY_SUPPLY_SHARE, "poolState.maxShareToBeSold");
-        assertEq(totalTokensOnBondingCurve, CoinConstants.POOL_LAUNCH_SUPPLY, "poolState.totalTokensOnBondingCurve");
+        assertEq(coin.totalSupply(), CoinConstants.MAX_TOTAL_SUPPLY);
+        assertEq(coin.balanceOf(coin.payoutRecipient()), CoinConstants.CREATOR_LAUNCH_REWARD);
+        assertApproxEqAbs(coin.balanceOf(address(pool)), CoinConstants.POOL_LAUNCH_SUPPLY, 1e18);
     }
 
     function test_deploy_doppler_eth_with_prebuy(uint256 initialOrderSize) public {
@@ -264,40 +121,40 @@ contract DopplerUniswapV3Test is BaseTest {
         int24 TICK_SPACING = 60;
         int24 expected = 12300;
 
-        assertEq(CoinDopplerUniV3.alignTickToTickSpacing(true, tick, TICK_SPACING), expected, "Align positive tick (token0)");
+        assertEq(DopplerMath.alignTickToTickSpacing(true, tick, TICK_SPACING), expected, "Align positive tick (token0)");
     }
 
     function test_alignTick_isToken0_negative() public pure {
         int24 tick = -12345;
         int24 TICK_SPACING = 60;
         int24 expected = -12360;
-        assertEq(CoinDopplerUniV3.alignTickToTickSpacing(true, tick, TICK_SPACING), expected, "Align negative tick (token0)");
+        assertEq(DopplerMath.alignTickToTickSpacing(true, tick, TICK_SPACING), expected, "Align negative tick (token0)");
     }
 
     function test_alignTick_isToken1_negative() public pure {
         int24 tick = -12345;
         int24 TICK_SPACING = 60;
         int24 expected = -12300;
-        assertEq(CoinDopplerUniV3.alignTickToTickSpacing(false, tick, TICK_SPACING), expected, "Align negative tick (token1)");
+        assertEq(DopplerMath.alignTickToTickSpacing(false, tick, TICK_SPACING), expected, "Align negative tick (token1)");
     }
 
     function test_alignTick_isToken1_zero() public pure {
         int24 tick = 0;
         int24 expected = 0;
-        assertEq(CoinDopplerUniV3.alignTickToTickSpacing(false, tick, MarketConstants.TICK_SPACING), expected, "Align zero tick (token1)");
+        assertEq(DopplerMath.alignTickToTickSpacing(false, tick, MarketConstants.TICK_SPACING), expected, "Align zero tick (token1)");
     }
 
     // Additional tick alignment test for full branch coverage
     function test_alignTick_isToken0_zero() public pure {
         int24 tick = 0;
         int24 expected = 0;
-        assertEq(CoinDopplerUniV3.alignTickToTickSpacing(true, tick, MarketConstants.TICK_SPACING), expected, "Align zero tick (token0)");
+        assertEq(DopplerMath.alignTickToTickSpacing(true, tick, MarketConstants.TICK_SPACING), expected, "Align zero tick (token0)");
     }
 
     function test_alignTick_isToken1_positive() public pure {
         int24 tick = 12345;
         int24 expected = 12400; // Round up for token1
-        assertEq(CoinDopplerUniV3.alignTickToTickSpacing(false, tick, MarketConstants.TICK_SPACING), expected, "Align positive tick (token1)");
+        assertEq(DopplerMath.alignTickToTickSpacing(false, tick, MarketConstants.TICK_SPACING), expected, "Align positive tick (token1)");
     }
 
     function test_calculateLpTail_isToken0() public pure {
@@ -306,10 +163,10 @@ contract DopplerUniswapV3Test is BaseTest {
         uint256 tailSupply = 1e18;
         bool isToken0 = true;
 
-        LpPosition memory tail = CoinDopplerUniV3.calculateLpTail(tickLower, tickUpper, isToken0, tailSupply, MarketConstants.TICK_SPACING);
+        LpPosition memory tail = DopplerMath.calculateLpTail(tickLower, tickUpper, isToken0, tailSupply, MarketConstants.TICK_SPACING);
 
         int24 expectedPosTickLower = tickUpper;
-        int24 expectedPosTickUpper = CoinDopplerUniV3.alignTickToTickSpacing(true, TickMath.MAX_TICK, MarketConstants.TICK_SPACING);
+        int24 expectedPosTickUpper = DopplerMath.alignTickToTickSpacing(true, TickMath.MAX_TICK, MarketConstants.TICK_SPACING);
 
         assertEq(tail.tickLower, expectedPosTickLower, "Tail tickLower (token0)");
         assertEq(tail.tickUpper, expectedPosTickUpper, "Tail tickUpper (token0)");
@@ -322,9 +179,9 @@ contract DopplerUniswapV3Test is BaseTest {
         uint256 tailSupply = 1e18;
         bool isToken0 = false;
 
-        LpPosition memory tail = CoinDopplerUniV3.calculateLpTail(tickLower, tickUpper, isToken0, tailSupply, MarketConstants.TICK_SPACING);
+        LpPosition memory tail = DopplerMath.calculateLpTail(tickLower, tickUpper, isToken0, tailSupply, MarketConstants.TICK_SPACING);
 
-        int24 expectedPosTickLower = CoinDopplerUniV3.alignTickToTickSpacing(false, TickMath.MIN_TICK, MarketConstants.TICK_SPACING);
+        int24 expectedPosTickLower = DopplerMath.alignTickToTickSpacing(false, TickMath.MIN_TICK, MarketConstants.TICK_SPACING);
         int24 expectedPosTickUpper = tickLower;
 
         assertEq(tail.tickLower, expectedPosTickLower, "Tail tickLower (token1)");
@@ -339,14 +196,15 @@ contract DopplerUniswapV3Test is BaseTest {
         uint256 discoverySupply = 100e18;
         LpPosition[] memory newPositions = new LpPosition[](DEFAULT_NUM_DISCOVERY_POSITIONS);
 
-        (LpPosition[] memory positions, uint256 totalAssetsSold) = CoinDopplerUniV3.calculateLogNormalDistribution(
+        (LpPosition[] memory positions, uint256 totalAssetsSold) = DopplerMath.calculateLogNormalDistribution(
             tickLower,
             tickUpper,
             MarketConstants.TICK_SPACING,
             isToken0,
             discoverySupply,
             DEFAULT_NUM_DISCOVERY_POSITIONS,
-            newPositions
+            newPositions,
+            0
         );
 
         assertEq(positions.length, DEFAULT_NUM_DISCOVERY_POSITIONS, "Correct number of positions (token0)");
@@ -370,14 +228,15 @@ contract DopplerUniswapV3Test is BaseTest {
         uint256 discoverySupply = 100e18;
         LpPosition[] memory newPositions = new LpPosition[](DEFAULT_NUM_DISCOVERY_POSITIONS);
 
-        (LpPosition[] memory positions, uint256 totalAssetsSold) = CoinDopplerUniV3.calculateLogNormalDistribution(
+        (LpPosition[] memory positions, uint256 totalAssetsSold) = DopplerMath.calculateLogNormalDistribution(
             tickLower,
             tickUpper,
             MarketConstants.TICK_SPACING,
             isToken0,
             discoverySupply,
             DEFAULT_NUM_DISCOVERY_POSITIONS,
-            newPositions
+            newPositions,
+            0
         );
 
         assertEq(positions.length, DEFAULT_NUM_DISCOVERY_POSITIONS, "Correct number of positions (token1)");
@@ -401,14 +260,15 @@ contract DopplerUniswapV3Test is BaseTest {
         uint256 discoverySupply = 0;
         LpPosition[] memory newPositions = new LpPosition[](DEFAULT_NUM_DISCOVERY_POSITIONS);
 
-        (LpPosition[] memory positions, uint256 totalAssetsSold) = CoinDopplerUniV3.calculateLogNormalDistribution(
+        (LpPosition[] memory positions, uint256 totalAssetsSold) = DopplerMath.calculateLogNormalDistribution(
             tickLower,
             tickUpper,
             MarketConstants.TICK_SPACING,
             isToken0,
             discoverySupply,
             DEFAULT_NUM_DISCOVERY_POSITIONS,
-            newPositions
+            newPositions,
+            0
         );
 
         assertEq(positions.length, DEFAULT_NUM_DISCOVERY_POSITIONS, "Correct number of positions (zero supply)");
@@ -432,14 +292,15 @@ contract DopplerUniswapV3Test is BaseTest {
         uint16 totalPositions = 1;
         LpPosition[] memory newPositions = new LpPosition[](totalPositions);
 
-        (LpPosition[] memory positions, uint256 totalAssetsSold) = CoinDopplerUniV3.calculateLogNormalDistribution(
+        (LpPosition[] memory positions, uint256 totalAssetsSold) = DopplerMath.calculateLogNormalDistribution(
             tickLower,
             tickUpper,
             MarketConstants.TICK_SPACING,
             isToken0,
             discoverySupply,
             totalPositions,
-            newPositions
+            newPositions,
+            0
         );
 
         assertEq(positions.length, totalPositions, "Correct number of positions");