@layerzerolabs/protocol-stellar-v2 0.2.18 → 0.2.20

package/contracts/oapps/oft/src/tests/extensions/test_rate_limiter.rs

@@ -1,1080 +0,0 @@
-//! Tests for the rate limiter extension functionality.
-//!
-//! Tests verify:
-//! - Rate limit configuration (set/unset)
-//! - Capacity tracking and decay over time
-//! - Rate limit enforcement on send (outbound)
-//! - Rate limit enforcement on lz_receive (inbound)
-//! - Capacity release on opposite direction
-
-use crate::{
-    extensions::rate_limiter::{Direction, RateLimitError},
-    tests::test_utils::{create_origin, create_recipient_address, encode_oft_message},
-    utils::address_to_bytes32,
-};
-use endpoint_v2::MessagingFee;
-use soroban_sdk::{
-    testutils::{Address as _, Ledger as _},
-    Address, Bytes, BytesN, Env, IntoVal,
-};
-
-use super::setup::ExtensiveOFTTestSetup;
-
-// ==================== Rate Limit Configuration Tests ====================
-
-#[test]
-fn test_initial_no_rate_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-
-    // No rate limit configured, capacity should be MAX
-    let capacity = setup.rate_limit_capacity(&Direction::Outbound, dst_eid);
-    assert_eq!(capacity, i128::MAX);
-
-    let (limit, window) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(limit, 0);
-    assert_eq!(window, 0);
-}
-
-#[test]
-fn test_set_rate_limit_outbound() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-    let limit = 1_000_000i128;
-    let window_seconds = 3600u64; // 1 hour
-
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, window_seconds);
-
-    let (stored_limit, stored_window) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(stored_limit, limit);
-    assert_eq!(stored_window, window_seconds);
-}
-
-#[test]
-fn test_set_rate_limit_inbound() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let src_eid = 100u32;
-    let limit = 2_000_000i128;
-    let window_seconds = 7200u64; // 2 hours
-
-    setup.set_rate_limit(&Direction::Inbound, src_eid, limit, window_seconds);
-
-    let (stored_limit, stored_window) = setup.rate_limit_config(&Direction::Inbound, src_eid);
-    assert_eq!(stored_limit, limit);
-    assert_eq!(stored_window, window_seconds);
-}
-
-#[test]
-fn test_set_rate_limit_both_directions() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let eid = 100u32;
-
-    setup.set_rate_limit(&Direction::Outbound, eid, 1_000_000, 3600);
-    setup.set_rate_limit(&Direction::Inbound, eid, 2_000_000, 7200);
-
-    let (outbound_limit, outbound_window) = setup.rate_limit_config(&Direction::Outbound, eid);
-    let (inbound_limit, inbound_window) = setup.rate_limit_config(&Direction::Inbound, eid);
-
-    assert_eq!(outbound_limit, 1_000_000);
-    assert_eq!(outbound_window, 3600);
-    assert_eq!(inbound_limit, 2_000_000);
-    assert_eq!(inbound_window, 7200);
-}
-
-#[test]
-fn test_set_rate_limit_zero_limit_fails() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let result = setup.try_set_rate_limit(&Direction::Outbound, 100, 0, 3600);
-    assert_eq!(result.err().unwrap().ok().unwrap(), RateLimitError::InvalidLimit.into());
-}
-
-#[test]
-fn test_set_rate_limit_zero_window_fails() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let result = setup.try_set_rate_limit(&Direction::Outbound, 100, 1_000_000, 0);
-    assert_eq!(result.err().unwrap().ok().unwrap(), RateLimitError::InvalidWindowSeconds.into());
-}
-
-#[test]
-fn test_set_rate_limit_same_values_fails() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-    let limit = 1_000_000i128;
-    let window_seconds = 3600u64;
-
-    // Set rate limit
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, window_seconds);
-
-    // Try to set same values again - should fail with SameValue
-    let result = setup.try_set_rate_limit(&Direction::Outbound, dst_eid, limit, window_seconds);
-    assert_eq!(result.err().unwrap().ok().unwrap(), RateLimitError::SameValue.into());
-}
-
-#[test]
-fn test_unset_rate_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 1_000_000, 3600);
-
-    let (limit, _) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(limit, 1_000_000);
-
-    setup.unset_rate_limit(&Direction::Outbound, dst_eid);
-
-    let (limit_after, _) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(limit_after, 0);
-}
-
-#[test]
-fn test_unset_rate_limit_not_found() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-
-    // Try to unset rate limit that doesn't exist - should fail with SameValue
-    let result = setup.try_unset_rate_limit(&Direction::Outbound, dst_eid);
-    assert_eq!(result.err().unwrap().ok().unwrap(), RateLimitError::SameValue.into());
-}
-
-#[test]
-fn test_update_rate_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 1_000_000, 3600);
-
-    // Update to new values
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 2_000_000, 7200);
-
-    let (limit, window) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(limit, 2_000_000);
-    assert_eq!(window, 7200);
-}
-
-#[test]
-fn test_update_rate_limit_only_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 1_000_000, 3600);
-
-    // Update only the limit (keep same window)
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 2_000_000, 3600);
-
-    let (limit, window) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(limit, 2_000_000);
-    assert_eq!(window, 3600);
-}
-
-#[test]
-fn test_update_rate_limit_only_window() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 1_000_000, 3600);
-
-    // Update only the window (keep same limit)
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 1_000_000, 7200);
-
-    let (limit, window) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(limit, 1_000_000);
-    assert_eq!(window, 7200);
-}
-
-// ==================== Capacity Tests ====================
-
-#[test]
-fn test_initial_capacity_equals_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, 3600);
-
-    let capacity = setup.rate_limit_capacity(&Direction::Outbound, dst_eid);
-    assert_eq!(capacity, limit);
-}
-
-#[test]
-fn test_in_flight_initially_zero() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 1_000_000, 3600);
-
-    let in_flight = setup.rate_limit_in_flight(&Direction::Outbound, dst_eid);
-    assert_eq!(in_flight, 0);
-}
-
-#[test]
-fn test_rate_limit_in_flight_no_rate_limit_returns_zero() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-
-    // No rate limit set - in_flight should return 0
-    let in_flight = setup.rate_limit_in_flight(&Direction::Outbound, dst_eid);
-    assert_eq!(in_flight, 0);
-}
-
-#[test]
-fn test_rate_limit_capacity_zero_when_in_flight_exceeds_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, 3600);
-
-    // Use all capacity
-    let amount_ld = limit;
-    setup.fund_tokens(&sender, amount_ld);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt = setup.quote_oft(&send_param);
-    setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-
-    // Capacity should be exactly 0 after using all of it
-    let capacity = setup.rate_limit_capacity(&Direction::Outbound, dst_eid);
-    assert_eq!(capacity, 0);
-
-    // In-flight should equal limit
-    let in_flight = setup.rate_limit_in_flight(&Direction::Outbound, dst_eid);
-    assert_eq!(in_flight, limit);
-}
-
-#[test]
-fn test_update_rate_limit_preserves_in_flight() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set initial rate limit
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 1_000_000, 3600);
-
-    // Use some capacity
-    let amount_ld = 300_000i128;
-    setup.fund_tokens(&sender, amount_ld);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt = setup.quote_oft(&send_param);
-    setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-
-    // In-flight should be 300_000
-    let in_flight_before = setup.rate_limit_in_flight(&Direction::Outbound, dst_eid);
-    assert_eq!(in_flight_before, amount_ld);
-
-    // Update rate limit to higher value
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 2_000_000, 3600);
-
-    // In-flight should be preserved (checkpointed)
-    let in_flight_after = setup.rate_limit_in_flight(&Direction::Outbound, dst_eid);
-    assert_eq!(in_flight_after, amount_ld);
-
-    // New capacity should be new_limit - in_flight
-    let capacity = setup.rate_limit_capacity(&Direction::Outbound, dst_eid);
-    assert_eq!(capacity, 2_000_000 - amount_ld);
-}
-
-// ==================== Capacity Decay Tests ====================
-
-#[test]
-fn test_rate_limit_invalid_timestamp_fails() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-
-    // Start with a timestamp in the future
-    setup.advance_time(10000);
-
-    // Set rate limit (last_update will be set to current timestamp: 10000)
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, 1_000_000, 3600);
-
-    // Move time backwards by modifying ledger info directly
-    env.ledger().with_mut(|li| {
-        li.timestamp = 5000; // Earlier than last_update
-    });
-
-    // Querying in_flight should fail because timestamp < last_update
-    let result = setup.try_rate_limit_in_flight(&Direction::Outbound, dst_eid);
-    assert!(result.is_err());
-    let err = result.err().unwrap().ok().unwrap();
-    assert_eq!(err, RateLimitError::InvalidTimestamp.into());
-}
-
-#[test]
-fn test_consume_rate_limit_when_capacity_exactly_zero() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, 3600);
-
-    // Use all capacity
-    setup.fund_tokens(&sender, limit);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param = setup.create_send_param(dst_eid, limit, limit);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt = setup.quote_oft(&send_param);
-    setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-
-    // Capacity should be 0 now (in_flight equals limit)
-    assert_eq!(setup.rate_limit_capacity(&Direction::Outbound, dst_eid), 0);
-
-    // Try to send a small amount - should fail because capacity is exactly 0
-    // This tests the branch where capacity = 0 (limit <= in_flight) in __consume_rate_limit_capacity
-    // Use amount that survives dust removal (conversion_rate = 10)
-    let small_amount = 100i128;
-    setup.fund_tokens(&sender, small_amount);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    // quote_oft will report the limit in OFTLimit but won't enforce
-    let send_param_2 = setup.create_send_param(dst_eid, small_amount, small_amount);
-    let oft_receipt_2 = setup.quote_oft(&send_param_2);
-
-    // send will fail because capacity is 0
-    let result = setup.try_send(&sender, &send_param_2, &fee, &sender, &oft_receipt_2);
-    assert_eq!(result.err().unwrap().ok().unwrap(), RateLimitError::ExceededRateLimit.into());
-}
-
-#[test]
-fn test_capacity_recovers_over_time() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set outbound rate limit: 1_000_000 per hour (3600 seconds)
-    let limit = 1_000_000i128;
-    let window_seconds = 3600u64;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, window_seconds);
-
-    // Use all capacity
-    let amount_ld = limit;
-    setup.fund_tokens(&sender, amount_ld);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt = setup.quote_oft(&send_param);
-    setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-
-    // Capacity should be 0
-    assert_eq!(setup.rate_limit_capacity(&Direction::Outbound, dst_eid), 0);
-
-    // Advance time by half the window (1800 seconds)
-    setup.advance_time(1800);
-
-    // Capacity should recover to ~50%
-    let capacity_after_half = setup.rate_limit_capacity(&Direction::Outbound, dst_eid);
-    assert!(capacity_after_half >= 490_000 && capacity_after_half <= 510_000);
-
-    // Advance time by another half (full window elapsed)
-    setup.advance_time(1800);
-
-    // Capacity should be fully recovered
-    let capacity_after_full = setup.rate_limit_capacity(&Direction::Outbound, dst_eid);
-    assert_eq!(capacity_after_full, limit);
-}
-
-#[test]
-fn test_send_after_capacity_recovery() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    let window_seconds = 3600u64;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, window_seconds);
-
-    // Use all capacity
-    setup.fund_tokens(&sender, limit);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param = setup.create_send_param(dst_eid, limit, limit);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt = setup.quote_oft(&send_param);
-    setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-
-    // Advance time to fully recover
-    setup.advance_time(window_seconds);
-
-    // Should be able to send again
-    setup.fund_tokens(&sender, limit);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param_2 = setup.create_send_param(dst_eid, limit, limit);
-    let oft_receipt_2 = setup.quote_oft(&send_param_2);
-    let (msg_receipt, _) = setup.send(&sender, &send_param_2, &fee, &sender, &oft_receipt_2);
-    assert!(msg_receipt.nonce > 0);
-}
-
-// ==================== Send Tests (Outbound Rate Limit) ====================
-
-#[test]
-fn test_send_within_rate_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, 3600);
-
-    let amount_ld = 500_000i128; // Within limit
-    setup.fund_tokens(&sender, amount_ld);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt = setup.quote_oft(&send_param);
-
-    // Should succeed within limit
-    let (msg_receipt, _) = setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-    assert!(msg_receipt.nonce > 0);
-
-    // Check capacity decreased
-    let capacity = setup.rate_limit_capacity(&Direction::Outbound, dst_eid);
-    assert_eq!(capacity, limit - amount_ld);
-}
-
-#[test]
-fn test_send_exactly_at_rate_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, 3600);
-
-    let amount_ld = limit; // Exactly at limit
-    setup.fund_tokens(&sender, amount_ld);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt = setup.quote_oft(&send_param);
-
-    // Should succeed at exactly limit
-    let (msg_receipt, _) = setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-    assert!(msg_receipt.nonce > 0);
-
-    // Capacity should be zero
-    let capacity = setup.rate_limit_capacity(&Direction::Outbound, dst_eid);
-    assert_eq!(capacity, 0);
-}
-
-#[test]
-fn test_send_exceeds_rate_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, 3600);
-
-    // Use amount that exceeds limit and use min_amount_ld = 0 to avoid slippage errors
-    let amount_ld = limit + 10; // Exceeds limit (use +10 to account for dust)
-    setup.fund_tokens(&sender, amount_ld);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    // Use min_amount_ld = 0 to avoid slippage check interference
-    let send_param = setup.create_send_param(dst_eid, amount_ld, 0);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-
-    // quote_oft returns the limit but doesn't enforce - the send will fail
-    let (oft_limit, _, _) = setup.oft.quote_oft(&send_param);
-    assert_eq!(oft_limit.max_amount_ld, limit); // Capacity is 1_000_000
-
-    let oft_receipt = setup.quote_oft(&send_param);
-
-    // Should fail at send - amount exceeds rate limit
-    let result = setup.try_send(&sender, &send_param, &fee, &sender, &oft_receipt);
-    assert_eq!(result.err().unwrap().ok().unwrap(), RateLimitError::ExceededRateLimit.into());
-}
-
-#[test]
-fn test_send_multiple_within_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, 3600);
-
-    // First send
-    let amount_1 = 300_000i128;
-    setup.fund_tokens(&sender, amount_1);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param_1 = setup.create_send_param(dst_eid, amount_1, amount_1);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt_1 = setup.quote_oft(&send_param_1);
-    setup.send(&sender, &send_param_1, &fee, &sender, &oft_receipt_1);
-
-    // Second send
-    let amount_2 = 400_000i128;
-    setup.fund_tokens(&sender, amount_2);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param_2 = setup.create_send_param(dst_eid, amount_2, amount_2);
-    let oft_receipt_2 = setup.quote_oft(&send_param_2);
-    setup.send(&sender, &send_param_2, &fee, &sender, &oft_receipt_2);
-
-    // Check remaining capacity
-    let capacity = setup.rate_limit_capacity(&Direction::Outbound, dst_eid);
-    assert_eq!(capacity, limit - amount_1 - amount_2);
-}
-
-#[test]
-fn test_send_multiple_exceeds_cumulative_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, 3600);
-
-    // First send - uses most of the limit
-    let amount_1 = 800_000i128;
-    setup.fund_tokens(&sender, amount_1);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param_1 = setup.create_send_param(dst_eid, amount_1, amount_1);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt_1 = setup.quote_oft(&send_param_1);
-    setup.send(&sender, &send_param_1, &fee, &sender, &oft_receipt_1);
-
-    // Verify capacity was consumed
-    let remaining_capacity = setup.rate_limit_capacity(&Direction::Outbound, dst_eid);
-    assert_eq!(remaining_capacity, limit - amount_1); // 200_000 remaining
-
-    // Second send - would exceed remaining capacity
-    let amount_2 = 300_000i128; // Only 200_000 capacity left
-    setup.fund_tokens(&sender, amount_2);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param_2 = setup.create_send_param(dst_eid, amount_2, amount_2);
-    let oft_receipt_2 = setup.quote_oft(&send_param_2);
-
-    // Should fail at send - cumulative exceeds rate limit
-    let result = setup.try_send(&sender, &send_param_2, &fee, &sender, &oft_receipt_2);
-    let err = result.err().unwrap().ok().unwrap();
-    assert_eq!(err, RateLimitError::ExceededRateLimit.into());
-}
-
-// ==================== lz_receive Tests (Inbound Rate Limit) ====================
-
-#[test]
-fn test_lz_receive_within_rate_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let executor = Address::generate(&env);
-    let recipient = create_recipient_address(&env);
-    let src_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(src_eid, &peer);
-
-    // Set inbound rate limit
-    let limit = 10_000_000i128; // 10M in local decimals
-    setup.set_rate_limit(&Direction::Inbound, src_eid, limit, 3600);
-
-    let amount_sd = 500_000u64; // Within limit (will be converted to ld)
-    let recipient_bytes32 = address_to_bytes32(&recipient);
-    let message = encode_oft_message(&env, &recipient_bytes32, amount_sd);
-
-    let guid = BytesN::from_array(&env, &[1u8; 32]);
-    let origin = create_origin(src_eid, &peer, 1);
-    let extra_data = Bytes::new(&env);
-
-    let initial_balance = setup.token_client.balance(&recipient);
-
-    // Should succeed within limit
-    setup.lz_receive(&executor, &origin, &guid, &message, &extra_data, 0);
-
-    let conversion_rate = setup.oft.decimal_conversion_rate();
-    let expected_amount_ld = (amount_sd as i128) * conversion_rate;
-    assert_eq!(setup.token_client.balance(&recipient), initial_balance + expected_amount_ld);
-}
-
-#[test]
-fn test_lz_receive_exceeds_rate_limit() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let executor = Address::generate(&env);
-    let recipient = create_recipient_address(&env);
-    let src_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(src_eid, &peer);
-
-    // Set small inbound rate limit
-    let limit = 100_000i128; // Small limit
-    setup.set_rate_limit(&Direction::Inbound, src_eid, limit, 3600);
-
-    // Try to receive more than limit
-    // conversion_rate is 10, so 100_000 sd = 1_000_000 ld > limit of 100_000
-    let amount_sd = 100_000u64;
-    let recipient_bytes32 = address_to_bytes32(&recipient);
-    let message = encode_oft_message(&env, &recipient_bytes32, amount_sd);
-
-    let guid = BytesN::from_array(&env, &[1u8; 32]);
-    let origin = create_origin(src_eid, &peer, 1);
-    let extra_data = Bytes::new(&env);
-
-    // Should fail - exceeds limit
-    let result = setup.try_lz_receive(&executor, &origin, &guid, &message, &extra_data, 0);
-    assert!(result.is_err());
-    let err = result.err().unwrap().ok().unwrap();
-    assert_eq!(err, RateLimitError::ExceededRateLimit.into());
-}
-
-// ==================== Capacity Release Tests (Bidirectional) ====================
-
-#[test]
-fn test_inbound_release_outbound_capacity() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let recipient = create_recipient_address(&env);
-    let eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, eid, limit, 3600);
-
-    // Use half the outbound capacity
-    let outbound_amount = 500_000i128;
-    setup.fund_tokens(&sender, outbound_amount);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param = setup.create_send_param(eid, outbound_amount, outbound_amount);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt = setup.quote_oft(&send_param);
-    setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-
-    // Outbound in-flight should be 500_000
-    let in_flight_before = setup.rate_limit_in_flight(&Direction::Outbound, eid);
-    assert_eq!(in_flight_before, outbound_amount);
-
-    // Now receive tokens (which releases outbound capacity)
-    let executor = Address::generate(&env);
-    let amount_sd = 30_000u64; // 300_000 in ld (conversion rate is 10)
-    let recipient_bytes32 = address_to_bytes32(&recipient);
-    let message = encode_oft_message(&env, &recipient_bytes32, amount_sd);
-
-    let guid = BytesN::from_array(&env, &[1u8; 32]);
-    let origin = create_origin(eid, &peer, 1);
-    let extra_data = Bytes::new(&env);
-
-    setup.lz_receive(&executor, &origin, &guid, &message, &extra_data, 0);
-
-    // Outbound in-flight should be reduced by inbound amount
-    let conversion_rate = setup.oft.decimal_conversion_rate();
-    let inbound_amount_ld = (amount_sd as i128) * conversion_rate;
-    let in_flight_after = setup.rate_limit_in_flight(&Direction::Outbound, eid);
-    assert_eq!(in_flight_after, outbound_amount - inbound_amount_ld);
-}
-
-#[test]
-fn test_release_rate_limit_more_than_in_flight_resets_to_zero() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-    let recipient = create_recipient_address(&env);
-    let eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, eid, limit, 3600);
-
-    // Use small amount of outbound capacity
-    let outbound_amount = 100_000i128;
-    setup.fund_tokens(&sender, outbound_amount);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param = setup.create_send_param(eid, outbound_amount, outbound_amount);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt = setup.quote_oft(&send_param);
-    setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-
-    // Outbound in-flight should be 100_000
-    let in_flight_before = setup.rate_limit_in_flight(&Direction::Outbound, eid);
-    assert_eq!(in_flight_before, outbound_amount);
-
-    // Receive MORE tokens than were sent (releases more than in-flight)
-    let executor = Address::generate(&env);
-    let amount_sd = 50_000u64; // 500_000 in ld (more than 100_000 in flight)
-    let recipient_bytes32 = address_to_bytes32(&recipient);
-    let message = encode_oft_message(&env, &recipient_bytes32, amount_sd);
-
-    let guid = BytesN::from_array(&env, &[1u8; 32]);
-    let origin = create_origin(eid, &peer, 1);
-    let extra_data = Bytes::new(&env);
-
-    setup.lz_receive(&executor, &origin, &guid, &message, &extra_data, 0);
-
-    // Outbound in-flight should reset to 0 (not go negative)
-    let in_flight_after = setup.rate_limit_in_flight(&Direction::Outbound, eid);
-    assert_eq!(in_flight_after, 0);
-
-    // Capacity should be back to full limit
-    let capacity = setup.rate_limit_capacity(&Direction::Outbound, eid);
-    assert_eq!(capacity, limit);
-}
-
-// ==================== quote_oft Tests (Rate Limit in OFTLimit) ====================
-
-#[test]
-fn test_quote_oft_returns_rate_limit_capacity() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // Set outbound rate limit
-    let limit = 1_000_000i128;
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, 3600);
-
-    let amount_ld = 500_000i128;
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-
-    let (oft_limit, _, _) = setup.oft.quote_oft(&send_param);
-
-    // max_amount_ld should reflect rate limit capacity
-    assert_eq!(oft_limit.max_amount_ld, limit);
-}
-
-#[test]
-fn test_quote_oft_no_rate_limit_returns_max() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let dst_eid = 100u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid, &peer);
-
-    // No rate limit set
-
-    let amount_ld = 500_000i128;
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-
-    let (oft_limit, _, _) = setup.oft.quote_oft(&send_param);
-
-    // max_amount_ld should be i128::MAX when no rate limit
-    assert_eq!(oft_limit.max_amount_ld, i128::MAX);
-}
-
-// ==================== Different Destinations Tests ====================
-
-#[test]
-fn test_rate_limits_independent_per_destination() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Set fee deposit address (required by ExtensiveOFT)
-    let fee_collector = create_recipient_address(&env);
-    setup.set_fee_deposit_address(&fee_collector);
-
-    let sender = Address::generate(&env);
-
-    let dst_eid_1 = 100u32;
-    let dst_eid_2 = 200u32;
-    let peer = BytesN::from_array(&env, &[2u8; 32]);
-    setup.set_peer(dst_eid_1, &peer);
-    setup.set_peer(dst_eid_2, &peer);
-
-    // Set different limits for different destinations
-    setup.set_rate_limit(&Direction::Outbound, dst_eid_1, 1_000_000, 3600);
-    setup.set_rate_limit(&Direction::Outbound, dst_eid_2, 2_000_000, 3600);
-
-    // Use all capacity for dst_eid_1
-    let amount_1 = 1_000_000i128;
-    setup.fund_tokens(&sender, amount_1);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param_1 = setup.create_send_param(dst_eid_1, amount_1, amount_1);
-    let fee = MessagingFee { native_fee: setup.native_fee, zro_fee: 0 };
-    let oft_receipt_1 = setup.quote_oft(&send_param_1);
-    setup.send(&sender, &send_param_1, &fee, &sender, &oft_receipt_1);
-
-    // dst_eid_1 should have no capacity
-    assert_eq!(setup.rate_limit_capacity(&Direction::Outbound, dst_eid_1), 0);
-
-    // dst_eid_2 should still have full capacity
-    assert_eq!(setup.rate_limit_capacity(&Direction::Outbound, dst_eid_2), 2_000_000);
-
-    // Should be able to send to dst_eid_2
-    let amount_2 = 1_500_000i128;
-    setup.fund_tokens(&sender, amount_2);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    let send_param_2 = setup.create_send_param(dst_eid_2, amount_2, amount_2);
-    let oft_receipt_2 = setup.quote_oft(&send_param_2);
-    let (msg_receipt, _) = setup.send(&sender, &send_param_2, &fee, &sender, &oft_receipt_2);
-    assert!(msg_receipt.nonce > 0);
-}
-
-// ==================== Authentication Tests ====================
-
-#[test]
-fn test_set_rate_limit_requires_owner_auth() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-    let limit = 1_000_000i128;
-    let window_seconds = 3600u64;
-
-    // Try to set rate limit without auth
-    let result = setup.oft.try_set_rate_limit(&Direction::Outbound, &dst_eid, &limit, &window_seconds);
-    assert!(result.is_err());
-
-    // Rate limit should not be set
-    let (stored_limit, stored_window) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(stored_limit, 0);
-    assert_eq!(stored_window, 0);
-}
-
-#[test]
-fn test_set_rate_limit_wrong_auth_fails() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let non_owner = Address::generate(&env);
-    let dst_eid = 100u32;
-    let limit = 1_000_000i128;
-    let window_seconds = 3600u64;
-
-    // Try to set rate limit with wrong auth (non-owner)
-    env.mock_auths(&[soroban_sdk::testutils::MockAuth {
-        address: &non_owner,
-        invoke: &soroban_sdk::testutils::MockAuthInvoke {
-            contract: &setup.oft.address,
-            fn_name: "set_rate_limit",
-            args: (&Direction::Outbound, dst_eid, limit, window_seconds).into_val(&env),
-            sub_invokes: &[],
-        },
-    }]);
-    let result = setup.oft.try_set_rate_limit(&Direction::Outbound, &dst_eid, &limit, &window_seconds);
-    assert!(result.is_err());
-
-    // Rate limit should not be set
-    let (stored_limit, stored_window) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(stored_limit, 0);
-    assert_eq!(stored_window, 0);
-}
-
-#[test]
-fn test_set_rate_limit_inbound_requires_owner_auth() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let src_eid = 100u32;
-    let limit = 1_000_000i128;
-    let window_seconds = 3600u64;
-
-    // Try to set inbound rate limit without auth
-    let result = setup.oft.try_set_rate_limit(&Direction::Inbound, &src_eid, &limit, &window_seconds);
-    assert!(result.is_err());
-
-    // Rate limit should not be set
-    let (stored_limit, stored_window) = setup.rate_limit_config(&Direction::Inbound, src_eid);
-    assert_eq!(stored_limit, 0);
-    assert_eq!(stored_window, 0);
-}
-
-#[test]
-fn test_unset_rate_limit_requires_owner_auth() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let dst_eid = 100u32;
-    let limit = 1_000_000i128;
-    let window_seconds = 3600u64;
-
-    // First set rate limit as owner
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, window_seconds);
-    let (stored_limit, _) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(stored_limit, limit);
-
-    // Try to unset rate limit without auth
-    let result = setup.oft.try_unset_rate_limit(&Direction::Outbound, &dst_eid);
-    assert!(result.is_err());
-
-    // Rate limit should still be set
-    let (stored_limit, _) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(stored_limit, limit);
-}
-
-#[test]
-fn test_unset_rate_limit_wrong_auth_fails() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let non_owner = Address::generate(&env);
-    let dst_eid = 100u32;
-    let limit = 1_000_000i128;
-    let window_seconds = 3600u64;
-
-    // First set rate limit as owner
-    setup.set_rate_limit(&Direction::Outbound, dst_eid, limit, window_seconds);
-
-    // Try to unset rate limit with wrong auth (non-owner)
-    env.mock_auths(&[soroban_sdk::testutils::MockAuth {
-        address: &non_owner,
-        invoke: &soroban_sdk::testutils::MockAuthInvoke {
-            contract: &setup.oft.address,
-            fn_name: "unset_rate_limit",
-            args: (&Direction::Outbound, dst_eid).into_val(&env),
-            sub_invokes: &[],
-        },
-    }]);
-    let result = setup.oft.try_unset_rate_limit(&Direction::Outbound, &dst_eid);
-    assert!(result.is_err());
-
-    // Rate limit should still be set
-    let (stored_limit, _) = setup.rate_limit_config(&Direction::Outbound, dst_eid);
-    assert_eq!(stored_limit, limit);
-}