From 05b16edc7fba60f6c05b12fde67fab523569a0e9 Mon Sep 17 00:00:00 2001
From: 0xcivita <0xcivita@proton.me>
Date: Sun, 28 Jan 2024 12:00:52 -0500
Subject: [PATCH] test integration manager ignite with univ3

---
 tests/integration/conftest.py                 |  15 ++-
 .../test_manager_ignite_with_univ3.py         | 125 ++++++++++++++++++
 2 files changed, 138 insertions(+), 2 deletions(-)
 create mode 100644 tests/integration/test_manager_ignite_with_univ3.py

diff --git a/tests/integration/conftest.py b/tests/integration/conftest.py
index 7f6aa68..813a263 100644
--- a/tests/integration/conftest.py
+++ b/tests/integration/conftest.py
@@ -40,6 +40,11 @@ def univ3_pool(assert_mainnet_fork, Contract):
     return Contract("0x8ad599c3A0ff1De082011EFDDc58f1908eb6e6D8")
 
 
+@pytest.fixture(scope="module")
+def univ3_static_quoter(assert_mainnet_fork, Contract):
+    return Contract("0xc80f61d1bdAbD8f5285117e1558fDDf8C64870FE")
+
+
 @pytest.fixture(scope="module")
 def mrglv1_factory(project, accounts, univ3_factory):
     deployer = project.MarginalV1PoolDeployer.deploy(sender=accounts[0])
@@ -66,9 +71,15 @@ def mrglv1_router(project, accounts, mrglv1_factory, WETH9):
 
 
 @pytest.fixture(scope="module")
-def mrglv1_quoter(project, accounts, mrglv1_factory, WETH9):
+def mrglv1_quoter(
+    project, accounts, mrglv1_factory, WETH9, mrglv1_manager, univ3_static_quoter
+):
     return project.Quoter.deploy(
-        mrglv1_factory.address, WETH9.address, sender=accounts[0]
+        mrglv1_factory.address,
+        WETH9.address,
+        mrglv1_manager.address,
+        univ3_static_quoter.address,
+        sender=accounts[0],
     )
 
 
diff --git a/tests/integration/test_manager_ignite_with_univ3.py b/tests/integration/test_manager_ignite_with_univ3.py
new file mode 100644
index 0000000..989e924
--- /dev/null
+++ b/tests/integration/test_manager_ignite_with_univ3.py
@@ -0,0 +1,125 @@
+import pytest
+
+from utils.constants import (
+    MIN_SQRT_RATIO,
+    MAX_SQRT_RATIO,
+    MAINTENANCE_UNIT,
+    BASE_FEE_MIN,
+    GAS_LIQUIDATE,
+    TICK_CUMULATIVE_RATE_MAX,
+    FUNDING_PERIOD,
+)
+from utils.utils import calc_amounts_from_liquidity_sqrt_price_x96, get_position_key
+
+
+@pytest.fixture
+def mint_position(
+    mrglv1_pool_initialized_with_liquidity,
+    position_lib,
+    chain,
+    mrglv1_manager,
+    sender,
+):
+    def mint(zero_for_one: bool) -> int:
+        state = mrglv1_pool_initialized_with_liquidity.state()
+        maintenance = mrglv1_pool_initialized_with_liquidity.maintenance()
+        oracle = mrglv1_pool_initialized_with_liquidity.oracle()
+
+        sqrt_price_limit_x96 = (
+            MIN_SQRT_RATIO + 1 if zero_for_one else MAX_SQRT_RATIO - 1
+        )
+        (reserve0, reserve1) = calc_amounts_from_liquidity_sqrt_price_x96(
+            state.liquidity, state.sqrtPriceX96
+        )
+        reserve = reserve1 if zero_for_one else reserve0
+
+        size = reserve * 1 // 100  # 1% of reserves
+        margin = (size * maintenance * 125) // (MAINTENANCE_UNIT * 100)
+        size_min = (size * 80) // 100
+        debt_max = 2**128 - 1
+        amount_in_max = 2**256 - 1
+        deadline = chain.pending_timestamp + 3600
+
+        mint_params = (
+            mrglv1_pool_initialized_with_liquidity.token0(),
+            mrglv1_pool_initialized_with_liquidity.token1(),
+            maintenance,
+            oracle,
+            zero_for_one,
+            size,
+            size_min,
+            debt_max,
+            amount_in_max,
+            sqrt_price_limit_x96,
+            margin,
+            sender.address,
+            deadline,
+        )
+
+        premium = mrglv1_pool_initialized_with_liquidity.rewardPremium()
+        base_fee = chain.blocks[-1].base_fee
+        rewards = position_lib.liquidationRewards(
+            base_fee,
+            BASE_FEE_MIN,
+            GAS_LIQUIDATE,
+            premium,
+        )
+
+        tx = mrglv1_manager.mint(mint_params, sender=sender, value=rewards)
+        token_id = tx.decode_logs(mrglv1_manager.Mint)[0].tokenId
+        return int(token_id)
+
+    yield mint
+
+
+@pytest.mark.integration
+@pytest.mark.parametrize("zero_for_one", [True, False])
+def test_manager_ignite_with_univ3__settles_position(
+    mrglv1_pool_initialized_with_liquidity,
+    mrglv1_manager,
+    univ3_pool,
+    zero_for_one,
+    sender,
+    alice,
+    chain,
+    position_lib,
+    position_amounts_lib,
+    mint_position,
+):
+    token_id = mint_position(zero_for_one)
+
+    position_id = mrglv1_pool_initialized_with_liquidity.state().totalPositions - 1
+    key = get_position_key(mrglv1_manager.address, position_id)
+    position = mrglv1_pool_initialized_with_liquidity.positions(key)
+
+    block_timestamp_next = chain.pending_timestamp
+    deadline = chain.pending_timestamp + 3600
+    amount_out_min = 0
+
+    ignite_params = (
+        mrglv1_pool_initialized_with_liquidity.token0(),
+        mrglv1_pool_initialized_with_liquidity.token1(),
+        mrglv1_pool_initialized_with_liquidity.maintenance(),
+        mrglv1_pool_initialized_with_liquidity.oracle(),
+        token_id,
+        amount_out_min,
+        alice.address,
+        deadline,
+    )
+    mrglv1_manager.ignite(ignite_params, sender=sender)
+
+    state = mrglv1_pool_initialized_with_liquidity.state()
+    tick_cumulative_last = state.tickCumulative
+    oracle_tick_cumulatives, _ = univ3_pool.observe([0])
+
+    # sync then settle position
+    position = position_lib.sync(
+        position,
+        block_timestamp_next,
+        tick_cumulative_last,
+        oracle_tick_cumulatives[0],
+        TICK_CUMULATIVE_RATE_MAX,
+        FUNDING_PERIOD,
+    )
+    position = position_lib.settle(position)
+    assert mrglv1_pool_initialized_with_liquidity.positions(key) == position