@layerzerolabs/protocol-stellar-v2 0.2.22 → 0.2.23

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

@@ -0,0 +1,240 @@
+extern crate std;
+
+use crate::extensions::oft_fee::{OFTFee, OFTFeeError, OFTFeeInternal};
+use soroban_sdk::{
+    contract, contractimpl,
+    testutils::{Address as _, MockAuth, MockAuthInvoke},
+    token::{StellarAssetClient, TokenClient},
+    Address, Env, IntoVal,
+};
+use utils::auth::Auth;
+
+// ============================================================================
+// Test Contract
+// ============================================================================
+
+#[contract]
+struct FeeTestContract;
+
+impl Auth for FeeTestContract {
+    fn authorizer(env: &Env) -> Address {
+        env.current_contract_address()
+    }
+}
+
+impl OFTFeeInternal for FeeTestContract {}
+
+#[contractimpl(contracttrait)]
+impl OFTFee for FeeTestContract {}
+
+#[contractimpl]
+impl FeeTestContract {
+    pub fn fee_view(env: Env, dst_eid: u32, amount_ld: i128) -> i128 {
+        <Self as OFTFeeInternal>::__fee_view(&env, dst_eid, amount_ld)
+    }
+
+    pub fn charge_fee(env: Env, token: Address, from: Address, fee_amount: i128) {
+        <Self as OFTFeeInternal>::__charge_fee(&env, &token, &from, fee_amount)
+    }
+}
+
+// ============================================================================
+// Test Setup
+// ============================================================================
+
+struct TestSetup {
+    env: Env,
+    client: FeeTestContractClient<'static>,
+    contract_id: Address,
+    token: Address,
+    from: Address,
+    fee_deposit: Address,
+}
+
+fn setup() -> TestSetup {
+    let env = Env::default();
+    // Enable mock_all_auths_allowing_non_root_auth for all tests to bypass authorization checks
+    // including sub-contract invocations like token transfers
+    env.mock_all_auths_allowing_non_root_auth();
+
+    let contract_id = env.register(FeeTestContract, ());
+    let client = FeeTestContractClient::new(&env, &contract_id);
+
+    let token_admin = Address::generate(&env);
+    let sac = env.register_stellar_asset_contract_v2(token_admin.clone());
+    let token = sac.address();
+
+    let from = Address::generate(&env);
+    let fee_deposit = Address::generate(&env);
+
+    // Mint to `from` - auth is mocked globally
+    StellarAssetClient::new(&env, &token).mint(&from, &1_000_000i128);
+
+    TestSetup { env, client, contract_id, token, from, fee_deposit }
+}
+
+// ============================================================================
+// Fee View Tests
+// ============================================================================
+
+#[test]
+fn test_fee_view_zero_fee_returns_zero() {
+    let TestSetup { client, .. } = setup();
+
+    let fee = client.fee_view(&1u32, &1_000_000i128);
+    assert_eq!(fee, 0);
+}
+
+#[test]
+fn test_fee_deposit_address_errors_when_unset() {
+    let TestSetup { client, .. } = setup();
+
+    let res = client.try_fee_deposit_address();
+    assert_eq!(res.err().unwrap().ok().unwrap(), OFTFeeError::InvalidFeeDepositAddress.into());
+}
+
+#[test]
+fn test_fee_view_nonzero_fee_errors_when_deposit_address_unset() {
+    let TestSetup { client, .. } = setup();
+
+    client.set_default_fee_bps(&100u64);
+
+    let res = client.try_fee_view(&7u32, &1_000_000i128);
+    assert_eq!(res.err().unwrap().ok().unwrap(), OFTFeeError::InvalidFeeDepositAddress.into());
+}
+
+// ============================================================================
+// Set Default Fee BPS Tests
+// ============================================================================
+
+#[test]
+fn test_set_default_fee_bps_rejects_invalid_value() {
+    let TestSetup { client, .. } = setup();
+
+    let res = client.try_set_default_fee_bps(&10_001u64);
+    assert_eq!(res.err().unwrap().ok().unwrap(), OFTFeeError::InvalidFeeBps.into());
+}
+
+#[test]
+fn test_set_default_fee_bps_rejects_same_value() {
+    let TestSetup { client, .. } = setup();
+
+    let res = client.try_set_default_fee_bps(&0u64);
+    assert_eq!(res.err().unwrap().ok().unwrap(), OFTFeeError::SameValue.into());
+
+    client.set_default_fee_bps(&123u64);
+    let res2 = client.try_set_default_fee_bps(&123u64);
+    assert_eq!(res2.err().unwrap().ok().unwrap(), OFTFeeError::SameValue.into());
+}
+
+// ============================================================================
+// Set Fee BPS Tests
+// ============================================================================
+
+#[test]
+fn test_set_fee_bps_rejects_invalid_and_same_value() {
+    let TestSetup { client, .. } = setup();
+    let dst_eid = 101u32;
+
+    let res = client.try_set_fee_bps(&dst_eid, &None);
+    assert_eq!(res.err().unwrap().ok().unwrap(), OFTFeeError::SameValue.into());
+
+    let res2 = client.try_set_fee_bps(&dst_eid, &Some(10_001u64));
+    assert_eq!(res2.err().unwrap().ok().unwrap(), OFTFeeError::InvalidFeeBps.into());
+}
+
+#[test]
+fn test_set_fee_bps_set_and_remove() {
+    let TestSetup { client, .. } = setup();
+    let dst_eid = 101u32;
+
+    client.set_fee_bps(&dst_eid, &Some(200u64));
+    assert_eq!(client.fee_bps(&dst_eid), Some(200u64));
+    assert_eq!(client.effective_fee_bps(&dst_eid), 200u64);
+
+    client.set_default_fee_bps(&111u64);
+    client.set_fee_bps(&dst_eid, &None);
+    assert_eq!(client.fee_bps(&dst_eid), None);
+    assert_eq!(client.effective_fee_bps(&dst_eid), 111u64);
+}
+
+// ============================================================================
+// Charge Fee Tests
+// ============================================================================
+
+#[test]
+fn test_charge_fee_errors_without_deposit_address() {
+    let TestSetup { env, client, contract_id, token, from, .. } = setup();
+
+    // Mock auth for token transfer (will fail before transfer due to missing deposit address)
+    env.mock_auths(&[MockAuth {
+        address: &from,
+        invoke: &MockAuthInvoke {
+            contract: &contract_id,
+            fn_name: "charge_fee",
+            args: (&token, &from, &1i128).into_val(&env),
+            sub_invokes: &[MockAuthInvoke {
+                contract: &token,
+                fn_name: "transfer",
+                args: (&from, &Address::generate(&env), &1i128).into_val(&env),
+                sub_invokes: &[],
+            }],
+        },
+    }]);
+
+    let res = client.try_charge_fee(&token, &from, &1i128);
+    assert_eq!(res.err().unwrap().ok().unwrap(), OFTFeeError::InvalidFeeDepositAddress.into());
+}
+
+#[test]
+fn test_set_fee_deposit_address_same_value() {
+    let TestSetup { client, fee_deposit, .. } = setup();
+
+    client.set_fee_deposit_address(&fee_deposit);
+    let res = client.try_set_fee_deposit_address(&fee_deposit);
+    assert_eq!(res.err().unwrap().ok().unwrap(), OFTFeeError::SameValue.into());
+}
+
+#[test]
+fn test_charge_fee_zero_amount_no_transfer() {
+    let TestSetup { env, client, token, from, fee_deposit, .. } = setup();
+
+    client.set_fee_deposit_address(&fee_deposit);
+
+    let token_client = TokenClient::new(&env, &token);
+    let from_before = token_client.balance(&from);
+    let dep_before = token_client.balance(&fee_deposit);
+
+    // fee_amount == 0 => no transfer
+    client.charge_fee(&token, &from, &0i128);
+
+    assert_eq!(token_client.balance(&from), from_before);
+    assert_eq!(token_client.balance(&fee_deposit), dep_before);
+}
+
+#[test]
+fn test_charge_fee_transfers() {
+    let TestSetup { env, client, token, from, fee_deposit, .. } = setup();
+
+    client.set_fee_deposit_address(&fee_deposit);
+
+    let token_client = TokenClient::new(&env, &token);
+    let from_before = token_client.balance(&from);
+    let dep_before = token_client.balance(&fee_deposit);
+
+    client.charge_fee(&token, &from, &123i128);
+
+    assert_eq!(token_client.balance(&from), from_before - 123);
+    assert_eq!(token_client.balance(&fee_deposit), dep_before + 123);
+}
+
+#[test]
+fn test_fee_view_computes_correct_fee() {
+    let TestSetup { client, fee_deposit, .. } = setup();
+
+    client.set_fee_deposit_address(&fee_deposit);
+    client.set_default_fee_bps(&100u64); // 1%
+
+    let fee = client.fee_view(&999u32, &10_000i128);
+    assert_eq!(fee, 100); // 1% of 10,000
+}

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

@@ -0,0 +1,141 @@
+extern crate std;
+
+use crate::extensions::pausable::{OFTPausable, OFTPausableError, OFTPausableInternal};
+use soroban_sdk::{contract, contractimpl, Address, Env};
+use utils::auth::Auth;
+
+// ============================================================================
+// Test Contract
+// ============================================================================
+
+#[contract]
+struct PausableTestContract;
+
+impl Auth for PausableTestContract {
+    fn authorizer(env: &Env) -> Address {
+        env.current_contract_address()
+    }
+}
+
+impl OFTPausableInternal for PausableTestContract {}
+
+#[contractimpl(contracttrait)]
+impl OFTPausable for PausableTestContract {}
+
+#[contractimpl]
+impl PausableTestContract {
+    pub fn assert_not_paused(env: Env) {
+        <Self as OFTPausableInternal>::__assert_not_paused(&env)
+    }
+}
+
+// ============================================================================
+// Test Setup
+// ============================================================================
+
+struct TestSetup {
+    client: PausableTestContractClient<'static>,
+}
+
+fn setup() -> TestSetup {
+    let env = Env::default();
+    env.mock_all_auths_allowing_non_root_auth();
+
+    let contract_id = env.register(PausableTestContract, ());
+    let client = PausableTestContractClient::new(&env, &contract_id);
+    TestSetup { client }
+}
+
+// ============================================================================
+// Initial State Tests
+// ============================================================================
+
+#[test]
+fn test_initial_state_is_not_paused() {
+    let TestSetup { client, .. } = setup();
+
+    assert!(!client.is_paused());
+}
+
+// ============================================================================
+// Set Paused Tests
+// ============================================================================
+
+#[test]
+fn test_set_paused_to_true() {
+    let TestSetup { client, .. } = setup();
+
+    client.set_paused(&true);
+    assert!(client.is_paused());
+}
+
+#[test]
+fn test_set_paused_to_false_after_pausing() {
+    let TestSetup { client, .. } = setup();
+
+    client.set_paused(&true);
+    assert!(client.is_paused());
+
+    client.set_paused(&false);
+    assert!(!client.is_paused());
+}
+
+#[test]
+fn test_set_paused_rejects_same_value_when_unpaused() {
+    let TestSetup { client, .. } = setup();
+
+    // Initially not paused, setting to false should fail
+    let res = client.try_set_paused(&false);
+    assert_eq!(res.err().unwrap().ok().unwrap(), OFTPausableError::PauseStatusUnchanged.into());
+}
+
+#[test]
+fn test_set_paused_rejects_same_value_when_paused() {
+    let TestSetup { client, .. } = setup();
+
+    client.set_paused(&true);
+
+    // Already paused, setting to true should fail
+    let res = client.try_set_paused(&true);
+    assert_eq!(res.err().unwrap().ok().unwrap(), OFTPausableError::PauseStatusUnchanged.into());
+}
+
+// ============================================================================
+// Assert Not Paused Tests
+// ============================================================================
+
+#[test]
+fn test_assert_not_paused_succeeds_when_not_paused() {
+    let TestSetup { client, .. } = setup();
+
+    // Should not panic when not paused
+    client.assert_not_paused();
+}
+
+#[test]
+fn test_assert_not_paused_fails_when_paused() {
+    let TestSetup { client, .. } = setup();
+
+    client.set_paused(&true);
+
+    let res = client.try_assert_not_paused();
+    assert_eq!(res.err().unwrap().ok().unwrap(), OFTPausableError::Paused.into());
+}
+
+// ============================================================================
+// Toggle Tests
+// ============================================================================
+
+#[test]
+fn test_pause_unpause_toggle_multiple_times() {
+    let TestSetup { client, .. } = setup();
+
+    // Toggle multiple times
+    for _ in 0..3 {
+        assert!(!client.is_paused());
+        client.set_paused(&true);
+        assert!(client.is_paused());
+        client.set_paused(&false);
+    }
+    assert!(!client.is_paused());
+}