@layerzerolabs/protocol-stellar-v2 0.2.13 → 0.2.18

package/contracts/oapps/oapp/src/macro_tests/test_macros.rs

@@ -1,304 +1,401 @@
-extern crate self as oapp;
-
-use endpoint_v2::Origin;
-use oapp::{
-    oapp_core::OAppCore,
-    oapp_options_type3::{EnforcedOptionParam, OAppOptionsType3},
-    oapp_receiver::OAppReceiver,
-    oapp_sender::OAppSender,
-};
-use oapp_macros::{oapp, oapp_manual_impl};
-use soroban_sdk::{contractimpl, Address, Bytes, BytesN, Env, Vec};
-
 // ============================================================================
 // Test 1: Full OApp with all defaults
 // ============================================================================
-#[oapp]
-struct TestFullDefault;
-
-impl TestFullDefault {
-    fn __lz_receive(
-        _env: &Env,
-        _executor: &Address,
-        _origin: &Origin,
-        _guid: &BytesN<32>,
-        _message: &Bytes,
-        _extra_data: &Bytes,
-        _value: i128,
-    ) {
-        // Default behavior
+mod test_full_default {
+    extern crate self as oapp;
+
+    use endpoint_v2::Origin;
+    use oapp::oapp_receiver::LzReceiveInternal;
+    use oapp_macros::oapp;
+    use soroban_sdk::{Address, Bytes, BytesN, Env};
+
+    #[oapp]
+    struct TestFullDefault;
+
+    impl LzReceiveInternal for TestFullDefault {
+        fn __lz_receive(
+            _env: &Env,
+            _origin: &Origin,
+            _guid: &BytesN<32>,
+            _message: &Bytes,
+            _extra_data: &Bytes,
+            _executor: &Address,
+            _value: i128,
+        ) {
+            // Default behavior
+        }
     }
 }
 
 // ============================================================================
 // Test 2: Full OApp with manual core implementation
 // ============================================================================
-#[oapp]
-#[oapp_manual_impl(core)]
-struct TestFullManualCore;
-
-#[contractimpl(contracttrait)]
-impl OAppCore for TestFullManualCore {
-    fn oapp_version(_env: &Env) -> (u64, u64) {
-        (2, 0)
+mod test_full_manual_core {
+    extern crate self as oapp;
+
+    use endpoint_v2::Origin;
+    use oapp::oapp_core::OAppCore;
+    use oapp::oapp_receiver::LzReceiveInternal;
+    use oapp_macros::oapp;
+    use soroban_sdk::{contractimpl, Address, Bytes, BytesN, Env};
+
+    #[oapp(custom = [core])]
+    struct TestFullManualCore;
+
+    #[contractimpl(contracttrait)]
+    impl OAppCore for TestFullManualCore {
+        fn oapp_version(_env: &Env) -> (u64, u64) {
+            (2, 0)
+        }
     }
-}
 
-impl TestFullManualCore {
-    fn __lz_receive(
-        _env: &Env,
-        _executor: &Address,
-        _origin: &Origin,
-        _guid: &BytesN<32>,
-        _message: &Bytes,
-        _extra_data: &Bytes,
-        _value: i128,
-    ) {
-        // Custom logic
+    impl LzReceiveInternal for TestFullManualCore {
+        fn __lz_receive(
+            _env: &Env,
+            _origin: &Origin,
+            _guid: &BytesN<32>,
+            _message: &Bytes,
+            _extra_data: &Bytes,
+            _executor: &Address,
+            _value: i128,
+        ) {
+            // Custom logic
+        }
     }
 }
 
 // ============================================================================
 // Test 3: Full OApp with manual sender implementation
 // ============================================================================
-#[oapp]
-#[oapp_manual_impl(sender)]
-struct TestFullManualSender;
+mod test_full_manual_sender {
+    extern crate self as oapp;
 
-impl OAppSender for TestFullManualSender {
-    // Custom sender implementation
-}
+    use endpoint_v2::Origin;
+    use oapp::oapp_receiver::LzReceiveInternal;
+    use oapp::oapp_sender::OAppSenderInternal;
+    use oapp_macros::oapp;
+    use soroban_sdk::{Address, Bytes, BytesN, Env};
+
+    #[oapp(custom = [sender])]
+    struct TestFullManualSender;
 
-impl TestFullManualSender {
-    fn __lz_receive(
-        _env: &Env,
-        _executor: &Address,
-        _origin: &Origin,
-        _guid: &BytesN<32>,
-        _message: &Bytes,
-        _extra_data: &Bytes,
-        _value: i128,
-    ) {
-        // Implementation
+    impl OAppSenderInternal for TestFullManualSender {
+        // Custom sender implementation
+    }
+
+    impl LzReceiveInternal for TestFullManualSender {
+        fn __lz_receive(
+            _env: &Env,
+            _origin: &Origin,
+            _guid: &BytesN<32>,
+            _message: &Bytes,
+            _extra_data: &Bytes,
+            _executor: &Address,
+            _value: i128,
+        ) {
+            // Implementation
+        }
     }
 }
 
 // ============================================================================
 // Test 4: Full OApp with manual receiver implementation
 // ============================================================================
-#[oapp]
-#[oapp_manual_impl(receiver)]
-struct TestFullManualReceiver;
-
-#[contractimpl(contracttrait)]
-impl OAppReceiver for TestFullManualReceiver {
-    fn is_compose_msg_sender(_env: &Env, _origin: &Origin, _message: &Bytes, _sender: &Address) -> bool {
-        true
+mod test_full_manual_receiver {
+    extern crate self as oapp;
+
+    use endpoint_v2::Origin;
+    use oapp::oapp_receiver::{LzReceiveInternal, OAppReceiver};
+    use oapp_macros::oapp;
+    use soroban_sdk::{contractimpl, Address, Bytes, BytesN, Env};
+
+    #[oapp(custom = [receiver])]
+    struct TestFullManualReceiver;
+
+    impl LzReceiveInternal for TestFullManualReceiver {
+        fn __lz_receive(
+            _env: &Env,
+            _origin: &Origin,
+            _guid: &BytesN<32>,
+            _message: &Bytes,
+            _extra_data: &Bytes,
+            _executor: &Address,
+            _value: i128,
+        ) {
+            // Custom implementation
+        }
     }
 
-    fn allow_initialize_path(_env: &Env, _origin: &Origin) -> bool {
-        true
-    }
+    #[contractimpl(contracttrait)]
+    impl OAppReceiver for TestFullManualReceiver {
+        fn is_compose_msg_sender(_env: &Env, _origin: &Origin, _message: &Bytes, _sender: &Address) -> bool {
+            true
+        }
 
-    fn next_nonce(_env: &Env, _src_eid: u32, _sender: &BytesN<32>) -> u64 {
-        1 // Ordered delivery
-    }
+        fn allow_initialize_path(_env: &Env, _origin: &Origin) -> bool {
+            true
+        }
 
-    fn lz_receive(
-        _env: &Env,
-        _executor: &Address,
-        _origin: &Origin,
-        _guid: &BytesN<32>,
-        _message: &Bytes,
-        _extra_data: &Bytes,
-        _value: i128,
-    ) {
-        // Custom implementation - verify_and_clear_payload must be called manually
+        fn next_nonce(_env: &Env, _src_eid: u32, _sender: &BytesN<32>) -> u64 {
+            1 // Ordered delivery
+        }
     }
 }
 
 // ============================================================================
 // Test 5: Full OApp with manual options_type3 implementation
 // ============================================================================
-#[oapp]
-#[oapp_manual_impl(options_type3)]
-struct TestFullManualOptions;
-
-impl TestFullManualOptions {
-    fn __lz_receive(
-        _env: &Env,
-        _executor: &Address,
-        _origin: &Origin,
-        _guid: &BytesN<32>,
-        _message: &Bytes,
-        _extra_data: &Bytes,
-        _value: i128,
-    ) {
-        // Implementation
+mod test_full_manual_options {
+    extern crate self as oapp;
+
+    use endpoint_v2::Origin;
+    use oapp::oapp_options_type3::{EnforcedOptionParam, OAppOptionsType3};
+    use oapp::oapp_receiver::LzReceiveInternal;
+    use oapp_macros::oapp;
+    use soroban_sdk::{contractimpl, Address, Bytes, BytesN, Env, Vec};
+
+    #[oapp(custom = [options_type3])]
+    struct TestFullManualOptions;
+
+    impl LzReceiveInternal for TestFullManualOptions {
+        fn __lz_receive(
+            _env: &Env,
+            _origin: &Origin,
+            _guid: &BytesN<32>,
+            _message: &Bytes,
+            _extra_data: &Bytes,
+            _executor: &Address,
+            _value: i128,
+        ) {
+            // Implementation
+        }
     }
-}
 
-#[contractimpl(contracttrait)]
-impl OAppOptionsType3 for TestFullManualOptions {
-    // Custom options implementation
+    #[contractimpl(contracttrait)]
+    impl OAppOptionsType3 for TestFullManualOptions {
+        // Custom options implementation
+    }
 }
 
 // ============================================================================
 // Test 6: Full OApp with manual core + sender
 // ============================================================================
-#[oapp]
-#[oapp_manual_impl(core, sender)]
-struct TestFullManualCoreSender;
-
-#[contractimpl(contracttrait)]
-impl OAppCore for TestFullManualCoreSender {
-    fn oapp_version(_env: &Env) -> (u64, u64) {
-        (3, 0)
+mod test_full_manual_core_sender {
+    extern crate self as oapp;
+
+    use endpoint_v2::Origin;
+    use oapp::{oapp_core::OAppCore, oapp_receiver::LzReceiveInternal, oapp_sender::OAppSenderInternal};
+    use oapp_macros::oapp;
+    use soroban_sdk::{contractimpl, Address, Bytes, BytesN, Env};
+
+    #[oapp(custom = [core, sender])]
+    struct TestFullManualCoreSender;
+
+    #[contractimpl(contracttrait)]
+    impl OAppCore for TestFullManualCoreSender {
+        fn oapp_version(_env: &Env) -> (u64, u64) {
+            (3, 0)
+        }
     }
-}
 
-impl OAppSender for TestFullManualCoreSender {
-    // Custom sender implementation
-}
+    impl OAppSenderInternal for TestFullManualCoreSender {
+        // Custom sender implementation
+    }
 
-impl TestFullManualCoreSender {
-    fn __lz_receive(
-        _env: &Env,
-        _executor: &Address,
-        _origin: &Origin,
-        _guid: &BytesN<32>,
-        _message: &Bytes,
-        _extra_data: &Bytes,
-        _value: i128,
-    ) {
-        // Implementation
+    impl LzReceiveInternal for TestFullManualCoreSender {
+        fn __lz_receive(
+            _env: &Env,
+            _origin: &Origin,
+            _guid: &BytesN<32>,
+            _message: &Bytes,
+            _extra_data: &Bytes,
+            _executor: &Address,
+            _value: i128,
+        ) {
+            // Implementation
+        }
     }
 }
 
 // ============================================================================
 // Test 7: Full OApp with manual core + receiver
 // ============================================================================
-#[oapp]
-#[oapp_manual_impl(core, receiver)]
-struct TestFullManualCoreReceiver;
-
-#[contractimpl(contracttrait)]
-impl OAppCore for TestFullManualCoreReceiver {
-    fn oapp_version(_env: &Env) -> (u64, u64) {
-        (4, 0)
+mod test_full_manual_core_receiver {
+    extern crate self as oapp;
+
+    use endpoint_v2::Origin;
+    use oapp::{
+        oapp_core::OAppCore,
+        oapp_receiver::{LzReceiveInternal, OAppReceiver},
+    };
+    use oapp_macros::oapp;
+    use soroban_sdk::{contractimpl, Address, Bytes, BytesN, Env};
+
+    #[oapp(custom = [core, receiver])]
+    struct TestFullManualCoreReceiver;
+
+    #[contractimpl(contracttrait)]
+    impl OAppCore for TestFullManualCoreReceiver {
+        fn oapp_version(_env: &Env) -> (u64, u64) {
+            (4, 0)
+        }
     }
-}
 
-#[contractimpl(contracttrait)]
-impl OAppReceiver for TestFullManualCoreReceiver {
-    fn lz_receive(
-        _env: &Env,
-        _executor: &Address,
-        _origin: &Origin,
-        _guid: &BytesN<32>,
-        _message: &Bytes,
-        _extra_data: &Bytes,
-        _value: i128,
-    ) {
-        // Custom receiver implementation
+    impl LzReceiveInternal for TestFullManualCoreReceiver {
+        fn __lz_receive(
+            _env: &Env,
+            _origin: &Origin,
+            _guid: &BytesN<32>,
+            _message: &Bytes,
+            _extra_data: &Bytes,
+            _executor: &Address,
+            _value: i128,
+        ) {
+            // Custom receiver implementation
+        }
     }
+
+    #[contractimpl(contracttrait)]
+    impl OAppReceiver for TestFullManualCoreReceiver {}
 }
 
 // ============================================================================
 // Test 8: Full OApp with manual sender + receiver
 // ============================================================================
-#[oapp]
-#[oapp_manual_impl(sender, receiver)]
-struct TestFullManualSenderReceiver;
-
-impl OAppSender for TestFullManualSenderReceiver {
-    // Custom sender implementation
-}
+mod test_full_manual_sender_receiver {
+    extern crate self as oapp;
+
+    use endpoint_v2::Origin;
+    use oapp::{
+        oapp_receiver::{LzReceiveInternal, OAppReceiver},
+        oapp_sender::OAppSenderInternal,
+    };
+    use oapp_macros::oapp;
+    use soroban_sdk::{contractimpl, Address, Bytes, BytesN, Env};
+
+    #[oapp(custom = [sender, receiver])]
+    struct TestFullManualSenderReceiver;
+
+    impl OAppSenderInternal for TestFullManualSenderReceiver {
+        // Custom sender implementation
+    }
 
-#[contractimpl(contracttrait)]
-impl OAppReceiver for TestFullManualSenderReceiver {
-    fn lz_receive(
-        _env: &Env,
-        _executor: &Address,
-        _origin: &Origin,
-        _guid: &BytesN<32>,
-        _message: &Bytes,
-        _extra_data: &Bytes,
-        _value: i128,
-    ) {
-        // Custom receiver implementation
+    impl LzReceiveInternal for TestFullManualSenderReceiver {
+        fn __lz_receive(
+            _env: &Env,
+            _origin: &Origin,
+            _guid: &BytesN<32>,
+            _message: &Bytes,
+            _extra_data: &Bytes,
+            _executor: &Address,
+            _value: i128,
+        ) {
+            // Custom receiver implementation
+        }
     }
+
+    #[contractimpl(contracttrait)]
+    impl OAppReceiver for TestFullManualSenderReceiver {}
 }
 
 // ============================================================================
 // Test 9: Full OApp with manual core + sender + receiver (all except options)
 // ============================================================================
-#[oapp]
-#[oapp_manual_impl(core, sender, receiver)]
-struct TestFullManualAllExceptOptions;
-
-#[contractimpl(contracttrait)]
-impl OAppCore for TestFullManualAllExceptOptions {
-    fn oapp_version(_env: &Env) -> (u64, u64) {
-        (5, 0)
+mod test_full_manual_all_except_options {
+    extern crate self as oapp;
+
+    use endpoint_v2::Origin;
+    use oapp::{
+        oapp_core::OAppCore,
+        oapp_receiver::{LzReceiveInternal, OAppReceiver},
+        oapp_sender::OAppSenderInternal,
+    };
+    use oapp_macros::oapp;
+    use soroban_sdk::{contractimpl, Address, Bytes, BytesN, Env};
+
+    #[oapp(custom = [core, sender, receiver])]
+    struct TestFullManualAllExceptOptions;
+
+    #[contractimpl(contracttrait)]
+    impl OAppCore for TestFullManualAllExceptOptions {
+        fn oapp_version(_env: &Env) -> (u64, u64) {
+            (5, 0)
+        }
     }
-}
 
-impl OAppSender for TestFullManualAllExceptOptions {
-    // Custom sender implementation
-}
+    impl OAppSenderInternal for TestFullManualAllExceptOptions {
+        // Custom sender implementation
+    }
 
-#[contractimpl(contracttrait)]
-impl OAppReceiver for TestFullManualAllExceptOptions {
-    fn lz_receive(
-        _env: &Env,
-        _executor: &Address,
-        _origin: &Origin,
-        _guid: &BytesN<32>,
-        _message: &Bytes,
-        _extra_data: &Bytes,
-        _value: i128,
-    ) {
-        // Custom receiver implementation
+    impl LzReceiveInternal for TestFullManualAllExceptOptions {
+        fn __lz_receive(
+            _env: &Env,
+            _origin: &Origin,
+            _guid: &BytesN<32>,
+            _message: &Bytes,
+            _extra_data: &Bytes,
+            _executor: &Address,
+            _value: i128,
+        ) {
+            // Custom receiver implementation
+        }
     }
+
+    #[contractimpl(contracttrait)]
+    impl OAppReceiver for TestFullManualAllExceptOptions {}
 }
 
 // ============================================================================
 // Test 10: Full OApp with all manual implementations
 // ============================================================================
-#[oapp]
-#[oapp_manual_impl(core, sender, receiver, options_type3)]
-struct TestFullManualAll;
-
-#[contractimpl(contracttrait)]
-impl OAppCore for TestFullManualAll {
-    fn oapp_version(_env: &Env) -> (u64, u64) {
-        (6, 0)
+mod test_full_manual_all {
+    extern crate self as oapp;
+
+    use endpoint_v2::Origin;
+    use oapp::{
+        oapp_core::OAppCore,
+        oapp_options_type3::{EnforcedOptionParam, OAppOptionsType3},
+        oapp_receiver::{LzReceiveInternal, OAppReceiver},
+        oapp_sender::OAppSenderInternal,
+    };
+    use oapp_macros::oapp;
+    use soroban_sdk::{contractimpl, Address, Bytes, BytesN, Env, Vec};
+
+    #[oapp(custom = [core, sender, receiver, options_type3])]
+    struct TestFullManualAll;
+
+    #[contractimpl(contracttrait)]
+    impl OAppCore for TestFullManualAll {
+        fn oapp_version(_env: &Env) -> (u64, u64) {
+            (6, 0)
+        }
     }
-}
 
-impl OAppSender for TestFullManualAll {
-    // Custom sender implementation
-}
+    impl OAppSenderInternal for TestFullManualAll {
+        // Custom sender implementation
+    }
 
-#[contractimpl(contracttrait)]
-impl OAppReceiver for TestFullManualAll {
-    fn lz_receive(
-        _env: &Env,
-        _executor: &Address,
-        _origin: &Origin,
-        _guid: &BytesN<32>,
-        _message: &Bytes,
-        _extra_data: &Bytes,
-        _value: i128,
-    ) {
-        // Custom receiver implementation
+    impl LzReceiveInternal for TestFullManualAll {
+        fn __lz_receive(
+            _env: &Env,
+            _origin: &Origin,
+            _guid: &BytesN<32>,
+            _message: &Bytes,
+            _extra_data: &Bytes,
+            _executor: &Address,
+            _value: i128,
+        ) {
+            // Custom receiver implementation
+        }
     }
-}
 
-#[contractimpl(contracttrait)]
-impl OAppOptionsType3 for TestFullManualAll {
-    // Custom options implementation
+    #[contractimpl(contracttrait)]
+    impl OAppReceiver for TestFullManualAll {}
+
+    #[contractimpl(contracttrait)]
+    impl OAppOptionsType3 for TestFullManualAll {
+        // Custom options implementation
+    }
 }
 
 // ============================================================================