@layerzerolabs/protocol-stellar-v2 0.2.8

package/contracts/message-libs/uln-302/src/tests/send_uln302/effective_executor_config.rs

@@ -0,0 +1,132 @@
+extern crate std;
+
+use crate::{
+    events::ExecutorConfigSet,
+    interfaces::{ExecutorConfig, OAppExecutorConfig, UlnConfig},
+    tests::setup::{setup, TestSetup},
+    uln302::CONFIG_TYPE_EXECUTOR,
+};
+use endpoint_v2::SetConfigParam;
+use soroban_sdk::{log, testutils::Address as _, vec, xdr::ToXdr, Address, Env};
+use utils::testing_utils::assert_event;
+
+#[test]
+fn test_effective_executor_config_with_default_only() {
+    let setup = setup();
+
+    let executor = setup.register_executable_address();
+    let default_config = ExecutorConfig { max_message_size: 5000, executor };
+    let eid = 102;
+
+    setup.set_default_executor_config(eid, default_config.clone());
+
+    let TestSetup { env, uln302, .. } = setup;
+
+    let oapp = Address::generate(&env);
+    log!(&env, "oapp: {}", oapp);
+
+    let config = uln302.effective_executor_config(&oapp, &eid);
+    assert_eq!(config, default_config);
+}
+
+#[test]
+fn test_effective_executor_config_with_custom_executor() {
+    let setup = setup();
+
+    let eid = 102;
+    let default_executor = setup.register_executable_address();
+    let default_config = ExecutorConfig { max_message_size: 5000, executor: default_executor };
+
+    let default_uln_config = UlnConfig::generate(&setup.env, 10, 2, 3, 2);
+
+    setup.set_default_executor_config(eid, default_config.clone());
+
+    setup.set_default_receive_uln_config(eid, default_uln_config.clone());
+    setup.set_default_send_uln_config(eid, default_uln_config.clone());
+
+    let oapp = Address::generate(&setup.env);
+    log!(&setup.env, "oapp: {}", oapp);
+
+    // Set custom executor but use default message size
+    let custom_executor = setup.register_executable_address();
+    let custom_config = OAppExecutorConfig { max_message_size: 0, executor: Some(custom_executor.clone()) };
+
+    let config_bytes = custom_config.clone().to_xdr(&setup.env);
+    let params = vec![&setup.env, SetConfigParam { eid, config_type: CONFIG_TYPE_EXECUTOR, config: config_bytes }];
+    setup.endpoint.set_config(&Address::generate(&setup.env), &oapp, &setup.uln302.address, &params);
+
+    // Assert ExecutorConfigSet event was published immediately after the setter
+
+    assert_event(
+        &setup.env,
+        &setup.uln302.address,
+        ExecutorConfigSet { config: custom_config.clone(), dst_eid: eid, sender: oapp.clone() },
+    );
+
+    let config = setup.uln302.effective_executor_config(&oapp, &eid);
+    assert_eq!(config.executor, custom_executor);
+    assert_eq!(config.max_message_size, default_config.max_message_size);
+}
+
+#[test]
+fn test_effective_executor_config_with_custom_message_size() {
+    let setup = setup();
+
+    let eid = 102;
+    let default_executor = setup.register_executable_address();
+    let default_config = ExecutorConfig { max_message_size: 5000, executor: default_executor.clone() };
+
+    let default_uln_config = UlnConfig::generate(&setup.env, 10, 2, 3, 2);
+
+    setup.set_default_executor_config(eid, default_config.clone());
+
+    setup.set_default_receive_uln_config(eid, default_uln_config.clone());
+    setup.set_default_send_uln_config(eid, default_uln_config.clone());
+
+    let TestSetup { env, uln302, endpoint, .. } = setup;
+
+    let oapp = Address::generate(&env);
+    log!(&env, "oapp: {}", oapp);
+
+    // Set custom message size but use default executor
+    let custom_config = OAppExecutorConfig { max_message_size: 10000, executor: None };
+
+    let config_bytes = custom_config.clone().to_xdr(&env);
+    let params = vec![&env, SetConfigParam { eid, config_type: CONFIG_TYPE_EXECUTOR, config: config_bytes }];
+    endpoint.set_config(&Address::generate(&env), &oapp, &uln302.address, &params);
+
+    let config = uln302.effective_executor_config(&oapp, &eid);
+    assert_eq!(config.executor, default_executor);
+    assert_eq!(config.max_message_size, custom_config.max_message_size);
+}
+
+#[test]
+fn test_effective_executor_config_fully_custom() {
+    let setup = setup();
+
+    let eid = 102;
+    let default_executor = setup.register_executable_address();
+    let default_config = ExecutorConfig { max_message_size: 5000, executor: default_executor.clone() };
+
+    let default_uln_config = UlnConfig::generate(&setup.env, 10, 2, 3, 2);
+
+    setup.set_default_executor_config(eid, default_config.clone());
+
+    setup.set_default_receive_uln_config(eid, default_uln_config.clone());
+    setup.set_default_send_uln_config(eid, default_uln_config.clone());
+
+    let oapp = Address::generate(&setup.env);
+    log!(&setup.env, "oapp: {}", oapp);
+
+    // Set both custom executor and message size
+    let custom_executor = setup.register_executable_address();
+    let custom_config = OAppExecutorConfig { max_message_size: 8000, executor: Some(custom_executor.clone()) };
+
+    let config_bytes = custom_config.clone().to_xdr(&setup.env);
+    let params = vec![&setup.env, SetConfigParam { eid, config_type: CONFIG_TYPE_EXECUTOR, config: config_bytes }];
+    setup.endpoint.set_config(&Address::generate(&setup.env), &oapp, &setup.uln302.address, &params);
+
+    let config = setup.uln302.effective_executor_config(&oapp, &eid);
+    assert_eq!(config.executor, custom_executor);
+    assert_eq!(config.max_message_size, custom_config.max_message_size);
+}

package/contracts/message-libs/uln-302/src/tests/send_uln302/effective_send_uln_config.rs

@@ -0,0 +1,117 @@
+extern crate std;
+
+use crate::{
+    errors::Uln302Error,
+    events::UlnSendConfigSet,
+    interfaces::{OAppUlnConfig, UlnConfig},
+    tests::setup::{setup, TestSetup},
+    uln302::CONFIG_TYPE_SEND_ULN,
+};
+use endpoint_v2::SetConfigParam;
+use soroban_sdk::{log, testutils::Address as _, vec, xdr::ToXdr, Address, Env};
+use utils::testing_utils::assert_event;
+
+#[test]
+fn test_effective_send_uln_config_with_default_only() {
+    // setup default send & receive uln config to enable `is_supported_eid`
+    let setup = setup();
+
+    // default send & receive uln config
+    let default_config = UlnConfig::generate(&setup.env, 10, 2, 3, 2);
+    let eid = 100;
+
+    setup.set_default_send_uln_config(eid, default_config.clone());
+    setup.set_default_receive_uln_config(eid, default_config.clone());
+
+    let TestSetup { env, uln302, .. } = setup;
+
+    let oapp = Address::generate(&env);
+
+    log!(&env, "oapp: {}", oapp);
+    // get send uln config for oapp without custom config - should return default
+    let config = uln302.effective_send_uln_config(&oapp, &eid);
+    assert_eq!(config, default_config);
+}
+
+#[test]
+fn test_effective_send_uln_config_with_custom_config() {
+    let setup = setup();
+
+    let default_config = UlnConfig::generate(&setup.env, 10, 2, 3, 2);
+    let eid = 100;
+
+    setup.set_default_send_uln_config(eid, default_config.clone());
+    setup.set_default_receive_uln_config(eid, default_config.clone());
+
+    let TestSetup { env, uln302, endpoint, .. } = setup;
+
+    let oapp = Address::generate(&env);
+    log!(&env, "oapp: {}", oapp);
+
+    // Create custom config that overrides confirmations but uses default DVNs
+    let custom_config = OAppUlnConfig {
+        use_default_confirmations: false,
+        use_default_required_dvns: true,
+        use_default_optional_dvns: true,
+        uln_config: UlnConfig {
+            confirmations: 20,
+            required_dvns: vec![&env],
+            optional_dvns: vec![&env],
+            optional_dvn_threshold: 0,
+        },
+    };
+
+    // Set custom config via set_config
+    let config_bytes = custom_config.clone().to_xdr(&env);
+    let params = vec![&env, SetConfigParam { eid, config_type: CONFIG_TYPE_SEND_ULN, config: config_bytes }];
+    endpoint.set_config(&Address::generate(&env), &oapp, &uln302.address, &params);
+
+    // Assert UlnSendConfigSet event was published immediately after the setter
+    assert_event(
+        &env,
+        &uln302.address,
+        UlnSendConfigSet { config: custom_config.clone(), dst_eid: eid, sender: oapp.clone() },
+    );
+
+    // Get aggregated config - should have custom confirmations but default DVNs
+    let config = uln302.effective_send_uln_config(&oapp, &eid);
+    assert_eq!(config.confirmations, 20);
+    assert_eq!(config.required_dvns, default_config.required_dvns);
+    assert_eq!(config.optional_dvns, default_config.optional_dvns);
+    assert_eq!(config.optional_dvn_threshold, default_config.optional_dvn_threshold);
+}
+
+#[test]
+fn test_effective_send_uln_config_must_have_at_least_one_dvn() {
+    let setup = setup();
+
+    // Create default with DVNs
+    let default_config = UlnConfig::generate(&setup.env, 10, 2, 0, 0);
+    let eid = 100;
+
+    setup.set_default_send_uln_config(eid, default_config.clone());
+    setup.set_default_receive_uln_config(eid, default_config);
+
+    let TestSetup { env, uln302, endpoint, .. } = setup;
+
+    let oapp = Address::generate(&env);
+    log!(&env, "oapp: {}", oapp);
+
+    // Create custom config that uses NO default DVNs and provides NO custom DVNs
+    let custom_config = OAppUlnConfig {
+        use_default_confirmations: true,
+        use_default_required_dvns: false,
+        use_default_optional_dvns: false,
+        uln_config: UlnConfig {
+            confirmations: 0,
+            required_dvns: vec![&env],
+            optional_dvns: vec![&env],
+            optional_dvn_threshold: 0,
+        },
+    };
+
+    let config_bytes = custom_config.to_xdr(&env);
+    let params = vec![&env, SetConfigParam { eid, config_type: CONFIG_TYPE_SEND_ULN, config: config_bytes }];
+    let result = endpoint.try_set_config(&Address::generate(&env), &oapp, &uln302.address, &params);
+    assert_eq!(result.err().unwrap().ok().unwrap(), Uln302Error::UlnAtLeastOneDVN.into());
+}

package/contracts/message-libs/uln-302/src/tests/send_uln302/mod.rs

@@ -0,0 +1,6 @@
+mod effective_executor_config;
+mod effective_send_uln_config;
+mod quote;
+mod send;
+mod set_default_executor_configs;
+mod set_default_send_uln_configs;