wip

onchain/test/BaseLineLP.t.sol

@@ -0,0 +1,89 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+pragma solidity ^0.8.19;
+
+import "forge-std/Test.sol";
+import "forge-std/console.sol";
+import {TwabController} from "pt-v5-twab-controller/TwabController.sol";
+import {PoolAddress, PoolKey} from "@aperture/uni-v3-lib/PoolAddress.sol";
+import "@uniswap-v3-core/interfaces/IUniswapV3Factory.sol";
+import "@uniswap-v3-core/interfaces/IUniswapV3Pool.sol";
+import "../src/interfaces/IWETH9.sol";
+import "../src/Harb.sol";
+import {BaseLineLP} from "../src/BaseLineLP.sol";
+import {Stake, ExceededAvailableStake} from "../src/Stake.sol";
+
+address constant WETH = 0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2;
+address constant V3_FACTORY = 0x1F98431c8aD98523631AE4a59f267346ea31F984;
+address constant TAX_POOL = address(2);
+// default fee of 1%
+uint24 constant FEE = uint24(10_000);
+
+contract BaseLineLPTest is Test {
+    uint256 mainnetFork;
+    IWETH9 weth;
+    Harb harb;
+    IUniswapV3Factory factory;
+    Stake stake;
+    BaseLineLP liquidityManager;
+
+    function sqrt(uint256 y) internal pure returns (uint256 z) {
+        if (y > 3) {
+            z = y;
+            uint256 x = y / 2 + 1;
+            while (x < z) {
+                z = x;
+                x = (y / x + x) / 2;
+            }
+        } else if (y != 0) {
+            z = 1;
+        }
+        // z is now the integer square root of y, or the closest integer to the square root of y.
+    }
+
+    function initializePoolFor1Cent(bool isEthToken0, address pool) public {
+        uint256 price;
+        if (isEthToken0) {
+            // ETH as token0, so we are setting the price of 1 ETH in terms of token1 (USD cent)
+            price = 3700 * 10**20; // 1 ETH = 3700 USD, scaled by 10^18 for precision
+        } else {
+            // Token (valued at 1 USD cent) as token0, ETH as token1
+            // We invert the logic to represent the price of 1 token in terms of ETH
+            price = uint256(10**16) / 3700; // Adjust for 18 decimal places
+        }
+
+        uint160 sqrtPriceX96 = uint160(sqrt(price) * 2**96 / 10**18); // Adjust sqrt value to 96-bit precision
+
+        // Initialize pool with the calculated sqrtPriceX96
+        IUniswapV3Pool(pool).initialize(sqrtPriceX96);
+    }
+
+    function setUp() public {
+        mainnetFork = vm.createFork(vm.envString("ETH_NODE_URI_MAINNET"), 19615864);
+        vm.selectFork(mainnetFork);
+        weth = IWETH9(WETH);
+        TwabController tc = new TwabController(60 * 60 * 24, uint32(block.timestamp));
+        harb = new Harb("HARB", "HARB", V3_FACTORY, WETH, tc);
+        factory = IUniswapV3Factory(V3_FACTORY);
+        address pool = factory.createPool(address(weth), address(harb), FEE);
+
+        initializePoolFor1Cent(address(weth) < address(harb), pool);
+
+        stake = new Stake(address(harb));
+        harb.setStakingPool(address(stake));
+        liquidityManager = new BaseLineLP(V3_FACTORY, WETH, address(harb));
+        harb.setLiquidityManager(address(liquidityManager));
+    }
+
+    function testLP(address account, uint256 amount) public {
+        vm.deal(account, 10 ether);
+        vm.prank(account);
+        (bool sent, ) = address(liquidityManager).call{value: 10 ether}("");
+        require(sent, "Failed to send Ether");
+
+        vm.expectRevert();
+        liquidityManager.shift();
+
+        int24 startTick = (address(weth) < address(harb)) ? int24(128219) : int24(-128219); //initialize at 1 cent
+        liquidityManager.slide();
+    }
+}