Merge pull request #146 from nninkovicSQA/TESTCOV-Sputnik-Factory-Own…

…ership

TESTCOV: Sputnik Factory: Ownership

sputnikdao-factory2/src/lib.rs

@@ -394,6 +394,7 @@ pub extern "C" fn store() {
 
 #[cfg(test)]
 mod tests {
+    use near_sdk::test_utils::test_env::{alice, bob, carol};
     use near_sdk::test_utils::{accounts, VMContextBuilder};
     use near_sdk::{testing_env, PromiseResult};
 
@@ -448,4 +449,72 @@ mod tests {
             vec![format!("test.{}", accounts(0)).parse().unwrap()]
         );
     }
+
+    //              #################################              //
+    //              #    Factory ownership tests    #              //
+    //              #################################              //
+
+    #[test]
+    fn test_factory_can_get_current_owner() {
+        let mut context = VMContextBuilder::new();
+        testing_env!(context
+            .current_account_id(alice())
+            .predecessor_account_id(alice())
+            .attached_deposit(to_yocto("5"))
+            .build());
+        let factory = SputnikDAOFactory::new();
+
+        assert_eq!(factory.get_owner(), alice());
+    }
+
+    #[test]
+    #[should_panic]
+    fn test_factory_fails_setting_owner_from_not_owner_account() {
+        let mut context = VMContextBuilder::new();
+        testing_env!(context
+            .current_account_id(alice())
+            .predecessor_account_id(carol())
+            .attached_deposit(to_yocto("5"))
+            .build());
+        let factory = SputnikDAOFactory::new();
+
+        factory.set_owner(bob());
+    }
+
+    #[test]
+    fn test_owner_can_be_a_dao_account() {
+        let mut context = VMContextBuilder::new();
+        testing_env!(context
+            .current_account_id(bob())
+            .predecessor_account_id(bob())
+            .attached_deposit(to_yocto("6"))
+            .build());
+        let mut factory = SputnikDAOFactory::new();
+
+        factory.create(bob(), "{}".as_bytes().to_vec().into());
+
+        factory.set_owner(AccountId::new_unchecked("bob.sputnik-dao.near".to_string()));
+
+        assert_eq!(
+            factory.get_owner(),
+            AccountId::new_unchecked("bob.sputnik-dao.near".to_string())
+        )
+    }
+
+    #[test]
+    fn test_owner_gets_succesfully_updated() {
+        let mut context = VMContextBuilder::new();
+        testing_env!(context
+            .current_account_id(accounts(0))
+            .predecessor_account_id(accounts(0))
+            .attached_deposit(to_yocto("5"))
+            .build());
+        let factory = SputnikDAOFactory::new();
+
+        assert_ne!(factory.get_owner(), bob());
+
+        factory.set_owner(bob());
+
+        assert_eq!(factory.get_owner(), bob())
+    }
 }