89 lines
3.3 KiB
Solidity
89 lines
3.3 KiB
Solidity
// 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();
|
|
}
|
|
}
|