@layerzerolabs/protocol-stellar-v2 0.2.21 → 0.2.23

package/contracts/workers/dvn/src/tests/multisig/set_signer.rs

@@ -1,132 +0,0 @@
-use crate::tests::setup::TestSetup;
-use soroban_sdk::{
-    testutils::{AuthorizedFunction, BytesN as _},
-    BytesN, Env, IntoVal, Symbol,
-};
-use utils::errors::MultisigError;
-
-fn random_signer(env: &Env) -> BytesN<20> {
-    BytesN::random(env)
-}
-
-fn clear_signers(setup: &TestSetup) {
-    setup.env.as_contract(&setup.contract_id, || {
-        utils::multisig::MultisigStorage::remove_signers(&setup.env);
-    });
-}
-
-#[test]
-fn test_set_signer() {
-    let setup = TestSetup::new(1);
-    let new_signer = random_signer(&setup.env);
-
-    setup.multisig_client.set_signer(&new_signer, &true);
-
-    let auths = setup.env.auths();
-    assert_eq!(auths.len(), 1);
-    let (auth_addr, auth_invocation) = &auths[0];
-    assert_eq!(auth_addr, &setup.contract_id);
-    match &auth_invocation.function {
-        AuthorizedFunction::Contract((contract_id, fn_name, args)) => {
-            assert_eq!(contract_id, &setup.contract_id);
-            assert_eq!(fn_name, &Symbol::new(&setup.env, "set_signer"));
-            assert_eq!(args, &(new_signer.clone(), true).into_val(&setup.env));
-        }
-        _ => panic!("Expected Contract auth"),
-    }
-
-    assert!(setup.multisig_client.is_signer(&new_signer));
-    assert_eq!(setup.multisig_client.get_signers().len(), 2);
-}
-
-#[test]
-fn test_set_signer_duplicate_fails() {
-    let setup = TestSetup::new(1);
-    let signer = random_signer(&setup.env);
-
-    setup.multisig_client.set_signer(&signer, &true);
-    let res = setup.multisig_client.try_set_signer(&signer, &true);
-
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::SignerAlreadyExists.into());
-}
-
-#[test]
-fn test_set_signer_remove_success() {
-    let setup = TestSetup::new(1);
-    let signer = random_signer(&setup.env);
-
-    setup.multisig_client.set_signer(&signer, &true);
-    assert!(setup.multisig_client.is_signer(&signer));
-
-    setup.multisig_client.set_signer(&signer, &false);
-    assert!(!setup.multisig_client.is_signer(&signer));
-}
-
-#[test]
-fn test_set_signer_remove_not_found() {
-    let setup = TestSetup::new(1);
-    let signer = random_signer(&setup.env);
-
-    let res = setup.multisig_client.try_set_signer(&signer, &false);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::SignerNotFound.into());
-}
-
-#[test]
-fn test_set_signer_invalid_signer_zero_address() {
-    let setup = TestSetup::new(1);
-    let zero = BytesN::from_array(&setup.env, &[0u8; 20]);
-
-    let res = setup.multisig_client.try_set_signer(&zero, &true);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::InvalidSigner.into());
-}
-
-#[test]
-fn test_get_signers_empty_map() {
-    let setup = TestSetup::new(1);
-
-    clear_signers(&setup);
-
-    let signers = setup.multisig_client.get_signers();
-    assert_eq!(signers.len(), 0);
-    let non_signer = random_signer(&setup.env);
-    assert!(!setup.multisig_client.is_signer(&non_signer));
-}
-
-#[test]
-fn test_total_signers() {
-    let setup = TestSetup::new(2);
-
-    assert_eq!(setup.multisig_client.total_signers(), 2);
-
-    let new_signer = random_signer(&setup.env);
-    setup.multisig_client.set_signer(&new_signer, &true);
-    assert_eq!(setup.multisig_client.total_signers(), 3);
-
-    setup.multisig_client.set_signer(&new_signer, &false);
-    assert_eq!(setup.multisig_client.total_signers(), 2);
-}
-
-#[test]
-fn test_remove_signer_when_no_signers() {
-    let setup = TestSetup::new(1);
-
-    clear_signers(&setup);
-
-    let signer = random_signer(&setup.env);
-    let res = setup.multisig_client.try_set_signer(&signer, &false);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::SignerNotFound.into());
-}
-
-#[test]
-fn test_remove_signer_violates_threshold() {
-    // Create DVN with 2 signers and threshold 2
-    let setup = TestSetup::new(2);
-
-    // Get one of the existing signers
-    let signers = setup.multisig_client.get_signers();
-    let signer_to_remove = signers.get(0).unwrap();
-
-    // Removing a signer should fail because it would leave only 1 signer with threshold 2
-    let res = setup.multisig_client.try_set_signer(&signer_to_remove, &false);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::TotalSignersLessThanThreshold.into());
-}

package/contracts/workers/dvn/src/tests/multisig/set_threshold.rs

@@ -1,109 +0,0 @@
-use crate::{
-    dvn::LzDVN,
-    tests::setup::{TestSetup, DEFAULT_MULTIPLIER_BPS, VID},
-};
-use soroban_sdk::{
-    testutils::{Address as _, AuthorizedFunction, BytesN as _},
-    vec, Address, BytesN, Env, IntoVal, Symbol, Vec,
-};
-use utils::errors::MultisigError;
-
-fn random_signer(env: &Env) -> BytesN<20> {
-    BytesN::random(env)
-}
-
-fn register_dvn(env: &Env, signer_count: usize, threshold: u32) {
-    let mut signers: Vec<BytesN<20>> = vec![env];
-    (0..signer_count).map(|_| random_signer(env)).for_each(|s| signers.push_back(s));
-    let admins: Vec<Address> = vec![env, Address::generate(env)];
-    let supported_msglibs: Vec<Address> = vec![env, Address::generate(env)];
-    let price_feed: Address = Address::generate(env);
-    let worker_fee_lib: Address = Address::generate(env);
-    let deposit_address: Address = Address::generate(env);
-
-    let _ = env.register(
-        LzDVN,
-        (
-            &VID,
-            &signers,
-            &threshold,
-            &admins,
-            &supported_msglibs,
-            &price_feed,
-            &DEFAULT_MULTIPLIER_BPS,
-            &worker_fee_lib,
-            &deposit_address,
-        ),
-    );
-}
-
-#[test]
-fn test_set_threshold_success() {
-    let setup = TestSetup::new(2);
-
-    setup.multisig_client.set_threshold(&2);
-
-    let auths = setup.env.auths();
-    assert_eq!(auths.len(), 1);
-    let (auth_addr, auth_invocation) = &auths[0];
-    assert_eq!(auth_addr, &setup.contract_id);
-    match &auth_invocation.function {
-        AuthorizedFunction::Contract((contract_id, fn_name, args)) => {
-            assert_eq!(contract_id, &setup.contract_id);
-            assert_eq!(fn_name, &Symbol::new(&setup.env, "set_threshold"));
-            assert_eq!(args, &(2u32,).into_val(&setup.env));
-        }
-        _ => panic!("Expected Contract auth"),
-    }
-
-    assert_eq!(setup.multisig_client.threshold(), 2);
-
-    setup.multisig_client.set_threshold(&1);
-    assert_eq!(setup.multisig_client.threshold(), 1);
-}
-
-#[test]
-fn test_set_threshold_zero_fails() {
-    let setup = TestSetup::new(1);
-
-    let res = setup.multisig_client.try_set_threshold(&0);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::ZeroThreshold.into());
-}
-
-#[test]
-fn test_set_threshold_above_signers_fails() {
-    let setup = TestSetup::new(2);
-
-    let res = setup.multisig_client.try_set_threshold(&3);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::TotalSignersLessThanThreshold.into());
-}
-
-#[test]
-fn test_set_threshold_with_no_signers_fails() {
-    let setup = TestSetup::new(1);
-
-    setup.env.as_contract(&setup.contract_id, || {
-        utils::multisig::MultisigStorage::remove_signers(&setup.env);
-    });
-
-    let res = setup.multisig_client.try_set_threshold(&1);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::TotalSignersLessThanThreshold.into());
-}
-
-#[test]
-#[should_panic(expected = "Error(Contract, #1067)")] // ZeroThreshold
-fn test_constructor_threshold_zero_panics() {
-    let env = Env::default();
-    env.mock_all_auths();
-
-    register_dvn(&env, 1, 0);
-}
-
-#[test]
-#[should_panic(expected = "Error(Contract, #1065)")] // TotalSignersLessThanThreshold
-fn test_constructor_signers_less_than_threshold_panics() {
-    let env = Env::default();
-    env.mock_all_auths();
-
-    register_dvn(&env, 1, 2);
-}

package/contracts/workers/dvn/src/tests/multisig/verify_signatures.rs

@@ -1,106 +0,0 @@
-extern crate std;
-
-use crate::tests::{key_pair::KeyPair, setup::TestSetup};
-use soroban_sdk::{vec, Bytes, BytesN, Vec};
-use utils::errors::MultisigError;
-
-fn hash(env: &soroban_sdk::Env, label: &[u8]) -> BytesN<32> {
-    env.crypto().keccak256(&Bytes::from_slice(env, label)).to_bytes()
-}
-
-fn sign(setup: &TestSetup, idx: usize, hash: &BytesN<32>) -> BytesN<65> {
-    setup.key_pairs[idx].sign_bytes(&setup.env, hash)
-}
-
-fn sorted_signatures(setup: &TestSetup, digest: &BytesN<32>) -> Vec<BytesN<65>> {
-    let mut pairs: std::vec::Vec<_> =
-        setup.key_pairs.iter().map(|kp| (kp.eth_address, kp.sign_bytes(&setup.env, digest))).collect();
-    pairs.sort_by(|a, b| a.0.cmp(&b.0));
-    let mut result: Vec<BytesN<65>> = vec![&setup.env];
-    pairs.into_iter().for_each(|(_, sig)| result.push_back(sig));
-    result
-}
-
-#[test]
-fn test_verify_signatures() {
-    let setup = TestSetup::new(1);
-    let digest = hash(&setup.env, b"test");
-    let signature = setup.key_pairs[0].sign(&digest.to_array());
-    let signatures: Vec<BytesN<65>> = vec![&setup.env, BytesN::from_array(&setup.env, &signature)];
-
-    setup.multisig_client.verify_signatures(&digest, &signatures);
-}
-
-#[test]
-fn test_verify_signatures_multiple_signers() {
-    let setup = TestSetup::new(2);
-    let digest = hash(&setup.env, b"multisig");
-
-    let signatures = sorted_signatures(&setup, &digest);
-
-    setup.multisig_client.verify_signatures(&digest, &signatures);
-}
-
-#[test]
-fn test_verify_signatures_raw_v_path() {
-    let setup = TestSetup::new(1);
-    let digest = hash(&setup.env, b"rawv");
-
-    // Sign normally (v in 27/28), then force v into raw 0/1 path
-    let mut sig_bytes = setup.key_pairs[0].sign(&digest.to_array());
-    sig_bytes[64] = sig_bytes[64].saturating_sub(27); // map 27->0 or 28->1
-    let signature = BytesN::from_array(&setup.env, &sig_bytes);
-    let signatures: Vec<BytesN<65>> = vec![&setup.env, signature];
-
-    setup.multisig_client.verify_signatures(&digest, &signatures);
-}
-
-#[test]
-fn test_verify_signatures_with_wrong_signer() {
-    let setup = TestSetup::new(1);
-    let digest = hash(&setup.env, b"test");
-    let wrong_key_pair = KeyPair::generate();
-    let signature = wrong_key_pair.sign(&digest.to_array());
-    let signatures: Vec<BytesN<65>> = vec![&setup.env, BytesN::from_array(&setup.env, &signature)];
-
-    let res = setup.multisig_client.try_verify_signatures(&digest, &signatures);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::SignerNotFound.into());
-}
-
-#[test]
-fn test_verify_signatures_insufficient_signatures() {
-    let setup = TestSetup::new(2);
-    let digest = hash(&setup.env, b"insufficient");
-
-    let signatures: Vec<BytesN<65>> = vec![&setup.env, sign(&setup, 0, &digest)];
-
-    let res = setup.multisig_client.try_verify_signatures(&digest, &signatures);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::SignatureError.into());
-}
-
-#[test]
-fn test_verify_signatures_duplicate_signer() {
-    let setup = TestSetup::new(2);
-    let digest = hash(&setup.env, b"duplicate");
-
-    let sig1 = sign(&setup, 0, &digest);
-    let signatures: Vec<BytesN<65>> = vec![&setup.env, sig1.clone(), sig1];
-
-    let res = setup.multisig_client.try_verify_signatures(&digest, &signatures);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::UnsortedSigners.into());
-}
-
-#[test]
-fn test_verify_signatures_threshold_zero() {
-    let setup = TestSetup::new(1);
-    // Force threshold to 0 via storage to hit ZeroThreshold branch
-    setup.env.as_contract(&setup.contract_id, || {
-        utils::multisig::MultisigStorage::set_threshold(&setup.env, &0);
-    });
-    let digest = hash(&setup.env, b"threshold_zero");
-
-    let signatures: Vec<BytesN<65>> = vec![&setup.env, sign(&setup, 0, &digest)];
-
-    let res = setup.multisig_client.try_verify_signatures(&digest, &signatures);
-    assert_eq!(res.err().unwrap().ok().unwrap(), MultisigError::ZeroThreshold.into());
-}