feat: updating ZOS over the chain (#385)

Author: renauter

substrate-node/pallets/pallet-tfgrid/src/lib.rs

@@ -180,6 +180,10 @@ pub mod pallet {
     #[pallet::getter(fn pallet_version)]
     pub type PalletVersion<T> = StorageValue<_, types::StorageVersion, ValueQuery>;
 
+    #[pallet::storage]
+    #[pallet::getter(fn zos_version)]
+    pub type ZosVersion<T> = StorageValue<_, Vec<u8>, ValueQuery>;
+
     #[pallet::config]
     pub trait Config: frame_system::Config + pallet_timestamp::Config {
         type Event: From<Event<Self>> + IsType<<Self as frame_system::Config>::Event>;
@@ -250,6 +254,8 @@ pub mod pallet {
         FarmingPolicyUpdated(types::FarmingPolicy<T::BlockNumber>),
         FarmingPolicySet(u32, Option<FarmingPolicyLimit>),
         FarmCertificationSet(u32, FarmCertification),
+
+        ZosVersionUpdated(Vec<u8>),
     }
 
     #[pallet::error]
@@ -315,6 +321,8 @@ pub mod pallet {
         FarmNameTooShort,
         FarmNameTooLong,
         MethodIsDeprecated,
+
+        InvalidZosVersion,
     }
 
     #[pallet::genesis_config]
@@ -1724,6 +1732,25 @@ pub mod pallet {
 
             Ok(().into())
         }
+
+        #[pallet::weight(100_000_000 + T::DbWeight::get().writes(1) + T::DbWeight::get().reads(1))]
+        pub fn set_zos_version(
+            origin: OriginFor<T>,
+            zos_version: Vec<u8>,
+        ) -> DispatchResultWithPostInfo {
+            T::RestrictedOrigin::ensure_origin(origin)?;
+
+            ensure!(
+                ZosVersion::<T>::get() != zos_version,
+                Error::<T>::InvalidZosVersion
+            );
+
+            ZosVersion::<T>::put(&zos_version);
+
+            Self::deposit_event(Event::ZosVersionUpdated(zos_version));
+
+            Ok(().into())
+        }
     }
 }
 

substrate-node/pallets/pallet-tfgrid/src/tests.rs

@@ -1,6 +1,8 @@
-use crate::{mock::*, Error};
+use super::Event as TfgridEvent;
+use crate::{mock::Event as MockEvent, mock::*, Error};
 use frame_support::{assert_noop, assert_ok, BoundedVec};
-use frame_system::RawOrigin;
+use frame_system::{EventRecord, Phase, RawOrigin};
+use sp_core::H256;
 use tfchain_support::types::{
     FarmCertification, FarmingPolicyLimit, Location, NodeCertification, PublicConfig, PublicIP,
     Resources,
@@ -1308,6 +1310,28 @@ fn test_create_and_update_policy() {
     });
 }
 
+#[test]
+ fn test_set_zos_version() {
+     ExternalityBuilder::build().execute_with(|| {
+         let zos_version = "1.0.0".as_bytes().to_vec();
+         assert_ok!(TfgridModule::set_zos_version(
+             RawOrigin::Root.into(),
+             zos_version.clone(),
+         ));
+
+         let saved_zos_version = TfgridModule::zos_version();
+         assert_eq!(saved_zos_version, zos_version);
+
+         let our_events = System::events();
+         assert_eq!(
+             our_events.contains(&record(MockEvent::TfgridModule(
+                 TfgridEvent::<TestRuntime>::ZosVersionUpdated(zos_version)
+             ))),
+             true
+         );
+     })
+ }
+
 fn create_entity() {
     let name = "foobar".as_bytes().to_vec();
     let country = "Belgium".as_bytes().to_vec();
@@ -1488,3 +1512,11 @@ fn create_farming_policies() {
         FarmCertification::NotCertified,
     ));
 }
+
+fn record(event: Event) -> EventRecord<Event, H256> {
+    EventRecord {
+        phase: Phase::Initialization,
+        event,
+        topics: vec![],
+    }
+}