@layerzerolabs/protocol-stellar-v2 0.2.12 → 0.2.15

package/contracts/utils/src/tests/ownable.rs

@@ -1,13 +1,9 @@
 use crate::{
-    errors::OwnableError,
-    ownable::{
-        self, DefaultOwnable, DefaultOwnableStorage, Ownable, OwnableInitializer, OwnershipRenounced,
-        OwnershipTransferred,
-    },
+    ownable::{self, Ownable, OwnableInitializer, OwnableStorage, OwnershipRenounced, OwnershipTransferred},
     testing_utils::assert_event,
 };
 use soroban_sdk::{
-    contract, contractimpl, panic_with_error,
+    contract, contractimpl,
     testutils::{Address as _, MockAuth, MockAuthInvoke},
     Address, Env, IntoVal,
 };
@@ -30,47 +26,18 @@ impl Contract {
         ownable::enforce_owner_auth::<Self>(env)
     }
 
-    pub fn init_owner(env: &Env, owner: &Address) {
-        if DefaultOwnableStorage::has_owner(env) {
-            panic_with_error!(env, OwnableError::OwnerAlreadySet);
-        }
-        DefaultOwnableStorage::set_owner(env, owner);
-    }
-
     pub fn remove_owner(env: &Env) {
-        DefaultOwnableStorage::remove_owner(env);
+        OwnableStorage::remove_owner(env);
     }
-}
 
-#[contractimpl]
-impl Ownable for Contract {
-    fn owner(env: &Env) -> Option<Address> {
-        DefaultOwnable::owner(env)
-    }
-
-    fn transfer_ownership(env: &Env, new_owner: &Address) {
-        DefaultOwnable::transfer_ownership(env, new_owner);
-    }
-
-    fn renounce_ownership(env: &Env) {
-        DefaultOwnable::renounce_ownership(env);
+    pub fn reinit_owner(env: &Env, owner: &Address) {
+        Self::init_owner(env, owner);
     }
 }
 
-/// Contract that does not implement Ownable trait - for testing raw init_owner
-#[contract]
-pub struct NonOwnableContract;
-
-#[contractimpl]
-impl NonOwnableContract {
-    pub fn init(env: &Env, owner: &Address) {
-        DefaultOwnable::init_owner(env, owner);
-    }
-
-    pub fn owner(env: &Env) -> Option<Address> {
-        DefaultOwnable::owner(env)
-    }
-}
+#[contractimpl(contracttrait)]
+impl Ownable for Contract {}
+impl OwnableInitializer for Contract {}
 
 // ============================================
 // auth: Owner authentication tests
@@ -329,42 +296,16 @@ fn chain_new_owner_can_transfer() {
 // init: DefaultOwnable::init_owner tests
 // ============================================
 
-#[test]
-fn init_owner_success() {
-    let env = Env::default();
-    let owner = Address::generate(&env);
-    let contract = env.register(NonOwnableContract, ());
-    let client = NonOwnableContractClient::new(&env, &contract);
-
-    assert_eq!(client.owner(), None);
-    client.init(&owner);
-    assert_eq!(client.owner(), Some(owner));
-}
-
-#[test]
-#[should_panic(expected = "Error(Contract, #1300)")]
-fn init_owner_already_set_fails() {
-    let env = Env::default();
-    let owner = Address::generate(&env);
-    let new_owner = Address::generate(&env);
-    let contract = env.register(NonOwnableContract, ());
-    let client = NonOwnableContractClient::new(&env, &contract);
-
-    client.init(&owner);
-    client.init(&new_owner);
-}
-
 #[test]
 #[should_panic(expected = "Error(Contract, #1300)")]
-fn init_owner_when_already_set_fails() {
+fn reinit_owner_fails() {
     let env = Env::default();
     let owner = Address::generate(&env);
     let new_owner = Address::generate(&env);
-
     let contract = env.register(Contract, (&owner,));
     let client = ContractClient::new(&env, &contract);
 
-    client.init_owner(&new_owner);
+    client.reinit_owner(&new_owner);
 }
 
 // ============================================