@layerzerolabs/protocol-stellar-v2 0.2.18 → 0.2.19

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

@@ -1,434 +0,0 @@
-//! Tests for the pausable extension functionality.
-//!
-//! Tests verify that when paused:
-//! - `send` is blocked
-//! - `lz_receive` is blocked
-//! - `quote_send` is blocked
-//! - `quote_oft` works (view function, not blocked)
-
-use crate::{
-    extensions::pausable::OFTPausableError,
-    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 _, Address, Bytes, BytesN, Env, IntoVal};
-
-use super::setup::ExtensiveOFTTestSetup;
-
-// ==================== Pausable State Tests ====================
-
-#[test]
-fn test_initial_state_not_paused() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    assert!(!setup.is_paused());
-}
-
-#[test]
-fn test_set_paused_true() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    setup.set_paused(true);
-    assert!(setup.is_paused());
-}
-
-#[test]
-fn test_set_paused_false_after_true() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    setup.set_paused(true);
-    assert!(setup.is_paused());
-
-    setup.set_paused(false);
-    assert!(!setup.is_paused());
-}
-
-#[test]
-fn test_set_paused_same_value_fails() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Already not paused, setting to false should fail with PauseStatusUnchanged
-    let result = setup.try_set_paused(false);
-    assert_eq!(result.err().unwrap().ok().unwrap(), OFTPausableError::PauseStatusUnchanged.into());
-}
-
-#[test]
-fn test_set_paused_true_when_already_paused_fails() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // First pause the contract
-    setup.set_paused(true);
-    assert!(setup.is_paused());
-
-    // Try to pause again - should fail with PauseStatusUnchanged
-    let result = setup.try_set_paused(true);
-    assert_eq!(result.err().unwrap().ok().unwrap(), OFTPausableError::PauseStatusUnchanged.into());
-}
-
-// ==================== Send Tests (Pausable) ====================
-
-#[test]
-fn test_send_succeeds_when_not_paused() {
-    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);
-
-    let amount_ld = 1_000_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);
-
-    // Should succeed when not paused
-    let (msg_receipt, _) = setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-    assert!(msg_receipt.nonce > 0);
-}
-
-#[test]
-fn test_send_fails_when_paused() {
-    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);
-
-    let amount_ld = 1_000_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);
-
-    // Pause the contract
-    setup.set_paused(true);
-
-    // Should fail when paused
-    let result = setup.try_send(&sender, &send_param, &fee, &sender, &oft_receipt);
-    assert_eq!(result.err().unwrap().ok().unwrap(), OFTPausableError::Paused.into());
-}
-
-#[test]
-fn test_send_succeeds_after_unpause() {
-    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);
-
-    let amount_ld = 1_000_000i128;
-    setup.fund_tokens(&sender, amount_ld);
-    setup.fund_native_fees(&sender, setup.native_fee);
-
-    // Pause then unpause
-    setup.set_paused(true);
-    setup.set_paused(false);
-
-    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 after unpause
-    let (msg_receipt, _) = setup.send(&sender, &send_param, &fee, &sender, &oft_receipt);
-    assert!(msg_receipt.nonce > 0);
-}
-
-// ==================== lz_receive Tests (Pausable) ====================
-
-#[test]
-fn test_lz_receive_succeeds_when_not_paused() {
-    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);
-
-    let amount_sd = 1_000_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);
-
-    let initial_balance = setup.token_client.balance(&recipient);
-
-    // Should succeed when not paused
-    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_fails_when_paused() {
-    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);
-
-    // Pause the contract
-    setup.set_paused(true);
-
-    let amount_sd = 1_000_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 when paused
-    let result = setup.try_lz_receive(&executor, &origin, &guid, &message, &extra_data, 0);
-    assert_eq!(result.err().unwrap().ok().unwrap(), OFTPausableError::Paused.into());
-}
-
-#[test]
-fn test_lz_receive_succeeds_after_unpause() {
-    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);
-
-    // Pause then unpause
-    setup.set_paused(true);
-    setup.set_paused(false);
-
-    let amount_sd = 1_000_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);
-
-    let initial_balance = setup.token_client.balance(&recipient);
-
-    // Should succeed after unpause
-    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);
-}
-
-// ==================== quote_send Tests (Pausable) ====================
-
-#[test]
-fn test_quote_send_succeeds_when_not_paused() {
-    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);
-
-    let amount_ld = 1_000_000i128;
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-
-    // Should succeed when not paused
-    let fee = setup.quote_send(&sender, &send_param, false);
-    assert!(fee.native_fee > 0);
-}
-
-#[test]
-fn test_quote_send_fails_when_paused() {
-    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);
-
-    // Pause the contract
-    setup.set_paused(true);
-
-    let amount_ld = 1_000_000i128;
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-
-    // Should fail when paused
-    let result = setup.try_quote_send(&sender, &send_param, false);
-    assert_eq!(result.err().unwrap().ok().unwrap(), OFTPausableError::Paused.into());
-}
-
-// ==================== quote_oft Tests (Pausable) ====================
-
-#[test]
-fn test_quote_oft_succeeds_when_not_paused() {
-    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);
-
-    let amount_ld = 1_000_000i128;
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-
-    // Should succeed when not paused
-    let receipt = setup.quote_oft(&send_param);
-    assert_eq!(receipt.amount_sent_ld, amount_ld);
-    assert_eq!(receipt.amount_received_ld, amount_ld);
-}
-
-#[test]
-fn test_quote_oft_fails_when_paused() {
-    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);
-
-    // Pause the contract
-    setup.set_paused(true);
-
-    let amount_ld = 1_000_000i128;
-    let send_param = setup.create_send_param(dst_eid, amount_ld, amount_ld);
-
-    // Should fail when paused
-    let result = setup.try_quote_oft(&send_param);
-    assert_eq!(result.err().unwrap().ok().unwrap(), OFTPausableError::Paused.into());
-}
-
-// ==================== Authentication Tests ====================
-
-#[test]
-fn test_set_paused_requires_owner_auth() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // Try to pause without auth
-    let result = setup.oft.try_set_paused(&true);
-    assert!(result.is_err());
-
-    // Should still be unpaused
-    assert!(!setup.is_paused());
-}
-
-#[test]
-fn test_set_paused_wrong_auth_fails() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let non_owner = Address::generate(&env);
-
-    // Try to pause 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_paused",
-            args: (true,).into_val(&env),
-            sub_invokes: &[],
-        },
-    }]);
-    let result = setup.oft.try_set_paused(&true);
-    assert!(result.is_err());
-
-    // Should still be unpaused
-    assert!(!setup.is_paused());
-}
-
-#[test]
-fn test_unpause_requires_owner_auth() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    // First pause as owner
-    setup.set_paused(true);
-    assert!(setup.is_paused());
-
-    // Try to unpause without auth
-    let result = setup.oft.try_set_paused(&false);
-    assert!(result.is_err());
-
-    // Should still be paused
-    assert!(setup.is_paused());
-}
-
-#[test]
-fn test_unpause_wrong_auth_fails() {
-    let env = Env::default();
-    let setup = ExtensiveOFTTestSetup::new(&env);
-
-    let non_owner = Address::generate(&env);
-
-    // First pause as owner
-    setup.set_paused(true);
-    assert!(setup.is_paused());
-
-    // Try to unpause 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_paused",
-            args: (false,).into_val(&env),
-            sub_invokes: &[],
-        },
-    }]);
-    let result = setup.oft.try_set_paused(&false);
-    assert!(result.is_err());
-
-    // Should still be paused
-    assert!(setup.is_paused());
-}