Added cross tests

src/e2e/cross.rs

@@ -0,0 +1,181 @@
+use crate::contracts::PoolKey;
+use crate::math::fee_growth::FeeGrowth;
+use crate::math::liquidity::Liquidity;
+use crate::math::percentage::Percentage;
+use crate::math::sqrt_price::calculate_sqrt_price;
+use crate::math::sqrt_price::SqrtPrice;
+use crate::math::token_amount::TokenAmount;
+use crate::math::MIN_SQRT_PRICE;
+use crate::token::TokenDeployer;
+use crate::FeeTier;
+use crate::InvariantDeployer;
+use alloc::string::String;
+use decimal::*;
+use odra::test_env;
+use odra::types::{U128, U256};
+
+#[test]
+fn test_cross() {
+    let deployer = test_env::get_account(0);
+    test_env::set_caller(deployer);
+    // Init basic dex and tokens
+    let mint_amount = U256::from(10u128.pow(10));
+    let mut token_x = TokenDeployer::init(String::from(""), String::from(""), 0, &mint_amount);
+    let mut token_y = TokenDeployer::init(String::from(""), String::from(""), 0, &mint_amount);
+    let mut invariant = InvariantDeployer::init(Percentage::from_scale(1, 2));
+    let fee_tier = FeeTier::new(Percentage::from_scale(6, 3), 10).unwrap();
+    let pool_key = PoolKey::new(*token_x.address(), *token_y.address(), fee_tier).unwrap();
+    // Init basic pool
+    {
+        invariant.add_fee_tier(fee_tier).unwrap();
+        let exist = invariant.fee_tier_exist(fee_tier);
+        assert!(exist);
+
+        let init_tick = 0;
+        let init_sqrt_price = calculate_sqrt_price(init_tick).unwrap();
+        invariant
+            .create_pool(
+                pool_key.token_x,
+                pool_key.token_y,
+                fee_tier,
+                init_sqrt_price,
+                init_tick,
+            )
+            .unwrap();
+    }
+    // Init basic positio
+    {
+        token_x.approve(invariant.address(), &mint_amount);
+        token_y.approve(invariant.address(), &mint_amount);
+
+        let lower_tick = -20;
+        let upper_tick = 10;
+        let liquidity = Liquidity::from_integer(1000000);
+
+        let pool_before = invariant
+            .get_pool(pool_key.token_x, pool_key.token_y, fee_tier)
+            .unwrap();
+
+        let slippage_limit_lower = pool_before.sqrt_price;
+        let slippage_limit_upper = pool_before.sqrt_price;
+
+        invariant
+            .create_position(
+                pool_key,
+                lower_tick,
+                upper_tick,
+                liquidity,
+                slippage_limit_lower,
+                slippage_limit_upper,
+            )
+            .unwrap();
+
+        let pool_after = invariant
+            .get_pool(pool_key.token_x, pool_key.token_y, fee_tier)
+            .unwrap();
+
+        assert_eq!(pool_after.liquidity, liquidity)
+    }
+    // Init cross position
+    {
+        let mint_amount = U256::from(10u128.pow(10));
+        token_x.approve(invariant.address(), &mint_amount);
+        token_y.approve(invariant.address(), &mint_amount);
+
+        let lower_tick = -40;
+        let upper_tick = -10;
+        let liquidity = Liquidity::from_integer(1000000);
+
+        let pool_before = invariant
+            .get_pool(pool_key.token_x, pool_key.token_y, fee_tier)
+            .unwrap();
+
+        let slippage_limit_lower = pool_before.sqrt_price;
+        let slippage_limit_upper = pool_before.sqrt_price;
+
+        invariant
+            .create_position(
+                pool_key,
+                lower_tick,
+                upper_tick,
+                liquidity,
+                slippage_limit_lower,
+                slippage_limit_upper,
+            )
+            .unwrap();
+
+        let pool_after = invariant
+            .get_pool(pool_key.token_x, pool_key.token_y, fee_tier)
+            .unwrap();
+
+        assert_eq!(pool_after.liquidity, pool_before.liquidity)
+    }
+    // init cross swap
+    {
+        let caller = test_env::get_account(1);
+        let amount = U256::from(1000);
+        token_x.mint(&caller, &amount);
+
+        let amount_x = token_x.balance_of(&caller);
+        assert_eq!(amount_x, U256::from(1000));
+        test_env::set_caller(caller);
+        token_x.approve(invariant.address(), &amount);
+
+        let amount_x = token_x.balance_of(invariant.address());
+        let amount_y = token_y.balance_of(invariant.address());
+        assert_eq!(amount_x, U256::from(500));
+        assert_eq!(amount_y, U256::from(2499));
+
+        let pool_before = invariant
+            .get_pool(pool_key.token_x, pool_key.token_y, fee_tier)
+            .unwrap();
+
+        let slippage = SqrtPrice::new(U128::from(MIN_SQRT_PRICE));
+        let swap_amount = TokenAmount::new(amount);
+        invariant
+            .swap(pool_key, true, swap_amount, true, slippage)
+            .unwrap();
+
+        let pool_after = invariant
+            .get_pool(pool_key.token_x, pool_key.token_y, fee_tier)
+            .unwrap();
+
+        let position_liquidity = Liquidity::from_integer(1000000);
+        let expected_tick = -20;
+
+        assert_eq!(
+            pool_after.liquidity - position_liquidity,
+            pool_before.liquidity
+        );
+        assert_eq!(pool_after.current_tick_index, expected_tick);
+        assert_ne!(pool_after.sqrt_price, pool_before.sqrt_price);
+
+        let amount_x = token_x.balance_of(&caller);
+        let amount_y = token_y.balance_of(&caller);
+        assert_eq!(amount_x, U256::from(0));
+        assert_eq!(amount_y, U256::from(990));
+
+        let amount_x = token_x.balance_of(invariant.address());
+        let amount_y = token_y.balance_of(invariant.address());
+        assert_eq!(amount_x, U256::from(1500));
+        assert_eq!(amount_y, U256::from(1509));
+
+        assert_eq!(
+            pool_after.fee_growth_global_x,
+            FeeGrowth::new(U128::from(40000000000000000000000u128))
+        );
+        assert_eq!(
+            pool_after.fee_growth_global_y,
+            FeeGrowth::new(U128::from(0))
+        );
+
+        assert_eq!(
+            pool_after.fee_protocol_token_x,
+            TokenAmount::new(U256::from(2))
+        );
+        assert_eq!(
+            pool_after.fee_protocol_token_y,
+            TokenAmount::new(U256::from(0))
+        );
+    }
+}

src/e2e/mod.rs

@@ -3,6 +3,7 @@ pub mod change_fee_receiver;
 pub mod change_protocol_fee;
 pub mod constructor;
 pub mod create_pool;
+pub mod cross;
 pub mod position;
 pub mod protocol_fee;
 pub mod remove_fee_tier;