-
Notifications
You must be signed in to change notification settings - Fork 248
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
8 changed files
with
267 additions
and
0 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,35 @@ | ||
| [package] | ||
| name = "pallet-storage-overlay-checks" | ||
| version = "0.1.0" | ||
| authors = ["Subspace Labs <https://subspace.network>"] | ||
| edition = "2021" | ||
| license = "0BSD" | ||
| homepage = "https://subspace.network" | ||
| repository = "https://github.com/autonomys/subspace" | ||
| description = "Test pallet to check the storage overlay changes" | ||
| include = [ | ||
| "/src", | ||
| "/Cargo.toml", | ||
| ] | ||
|
|
||
| [dependencies] | ||
| codec = { package = "parity-scale-codec", version = "3.6.12", default-features = false, features = ["derive"] } | ||
| frame-support = { default-features = false, git = "https://github.com/subspace/polkadot-sdk", rev = "94a1a8143a89bbe9f938c1939ff68abc1519a305" } | ||
| frame-system = { default-features = false, git = "https://github.com/subspace/polkadot-sdk", rev = "94a1a8143a89bbe9f938c1939ff68abc1519a305" } | ||
| log = { version = "0.4.22", default-features = false } | ||
| scale-info = { version = "2.11.2", default-features = false, features = ["derive"] } | ||
| sp-core = { default-features = false, git = "https://github.com/subspace/polkadot-sdk", rev = "94a1a8143a89bbe9f938c1939ff68abc1519a305" } | ||
| sp-runtime = { default-features = false, git = "https://github.com/subspace/polkadot-sdk", rev = "94a1a8143a89bbe9f938c1939ff68abc1519a305" } | ||
|
|
||
|
|
||
| [features] | ||
| default = ["std"] | ||
| std = [ | ||
| "codec/std", | ||
| "frame-support/std", | ||
| "frame-system/std", | ||
| "log/std", | ||
| "scale-info/std", | ||
| "sp-core/std", | ||
| "sp-runtime/std", | ||
| ] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,202 @@ | ||
| //! Test pallet to check the overlay changes during the block execution and post block execution. | ||
| #![cfg_attr(not(feature = "std"), no_std)] | ||
|
|
||
| #[cfg(not(feature = "std"))] | ||
| extern crate alloc; | ||
|
|
||
| use frame_system::pallet_prelude::BlockNumberFor; | ||
| pub use pallet::*; | ||
| use sp_core::H256; | ||
|
|
||
| const LOG_TARGET: &str = "runtime::storage_overlay_checks"; | ||
|
|
||
| #[frame_support::pallet] | ||
| mod pallet { | ||
| use crate::StorageParams; | ||
| use frame_support::pallet_prelude::*; | ||
| use frame_system::pallet_prelude::BlockNumberFor; | ||
| use sp_core::H256; | ||
| use sp_runtime::traits::Zero; | ||
|
|
||
| #[pallet::config] | ||
| pub trait Config: frame_system::Config {} | ||
|
|
||
| #[pallet::storage] | ||
| pub(super) type Svvq<T> = StorageValue<_, H256, ValueQuery>; | ||
|
|
||
| #[pallet::storage] | ||
| pub(super) type Svoq<T> = StorageValue<_, H256, OptionQuery>; | ||
|
|
||
| #[pallet::storage] | ||
| pub(super) type Smvq<T> = StorageMap<_, Identity, H256, H256, ValueQuery>; | ||
|
|
||
| #[pallet::storage] | ||
| pub(super) type Smoq<T> = StorageMap<_, Identity, H256, H256, OptionQuery>; | ||
|
|
||
| #[pallet::storage] | ||
| pub(super) type Sdmvq<T> = | ||
| StorageDoubleMap<_, Identity, H256, Identity, H256, H256, ValueQuery>; | ||
|
|
||
| #[pallet::storage] | ||
| pub(super) type Sdmoq<T> = | ||
| StorageDoubleMap<_, Identity, H256, Identity, H256, H256, OptionQuery>; | ||
|
|
||
| #[pallet::storage] | ||
| pub(super) type Snmvq<T> = StorageNMap< | ||
| _, | ||
| ( | ||
| NMapKey<Identity, H256>, | ||
| NMapKey<Identity, H256>, | ||
| NMapKey<Identity, H256>, | ||
| ), | ||
| H256, | ||
| ValueQuery, | ||
| >; | ||
|
|
||
| #[pallet::storage] | ||
| pub(super) type Snmoq<T> = StorageNMap< | ||
| _, | ||
| ( | ||
| NMapKey<Identity, H256>, | ||
| NMapKey<Identity, H256>, | ||
| NMapKey<Identity, H256>, | ||
| ), | ||
| H256, | ||
| OptionQuery, | ||
| >; | ||
|
|
||
| #[pallet::pallet] | ||
| #[pallet::without_storage_info] | ||
| pub struct Pallet<T>(_); | ||
|
|
||
| #[pallet::hooks] | ||
| impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> { | ||
| fn on_initialize(n: BlockNumberFor<T>) -> Weight { | ||
| // even blocks, storages must be set already and are emptied | ||
| // odd block, storages must be empty and are set | ||
| let params = StorageParams::default(); | ||
| if n % BlockNumberFor::<T>::from(2u32) == Zero::zero() { | ||
| Pallet::<T>::check_storage_exists(n, params); | ||
| Pallet::<T>::clear_storage(n); | ||
| } else { | ||
| Pallet::<T>::check_storage_empty(n, params); | ||
| Pallet::<T>::set_storage(n, params); | ||
| } | ||
|
|
||
| Weight::zero() | ||
| } | ||
|
|
||
| fn on_finalize(n: BlockNumberFor<T>) { | ||
| // even blocks, storages must be emptied during initialize | ||
| // odd block, storages must be set during initialize | ||
| let params = StorageParams::default(); | ||
| if n % BlockNumberFor::<T>::from(2u32) == Zero::zero() { | ||
| Pallet::<T>::check_storage_empty(n, params); | ||
| } else { | ||
| Pallet::<T>::check_storage_exists(n, params); | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| #[derive(Clone, Copy)] | ||
| struct StorageParams { | ||
| key1: H256, | ||
| key2: H256, | ||
| key3: H256, | ||
| value: H256, | ||
| } | ||
|
|
||
| impl Default for StorageParams { | ||
| fn default() -> Self { | ||
| Self { | ||
| key1: H256::repeat_byte(1), | ||
| key2: H256::repeat_byte(2), | ||
| key3: H256::repeat_byte(3), | ||
| value: H256::repeat_byte(4), | ||
| } | ||
| } | ||
| } | ||
|
|
||
| impl<T: Config> Pallet<T> { | ||
| fn set_storage(n: BlockNumberFor<T>, params: StorageParams) { | ||
| log::debug!(target: LOG_TARGET, "Setting storages at: {:?}", n); | ||
| let StorageParams { | ||
| key1, | ||
| key2, | ||
| key3, | ||
| value, | ||
| } = params; | ||
|
|
||
| Svvq::<T>::set(value); | ||
| Svoq::<T>::set(Some(value)); | ||
| Smvq::<T>::set(key1, value); | ||
| Smoq::<T>::set(key1, Some(value)); | ||
| Sdmvq::<T>::set(key1, key2, value); | ||
| Sdmoq::<T>::set(key1, key2, Some(value)); | ||
| Snmvq::<T>::set((key1, key2, key3), value); | ||
| Snmoq::<T>::set((key1, key2, key3), Some(value)); | ||
| } | ||
|
|
||
| fn clear_storage(n: BlockNumberFor<T>) { | ||
| log::debug!(target: LOG_TARGET, "Clearing storages at: {:?}", n); | ||
| Svvq::<T>::kill(); | ||
| Svoq::<T>::kill(); | ||
| let _ = Smvq::<T>::clear(u32::MAX, None); | ||
| let _ = Smoq::<T>::clear(u32::MAX, None); | ||
| let _ = Sdmvq::<T>::clear(u32::MAX, None); | ||
| let _ = Sdmoq::<T>::clear(u32::MAX, None); | ||
| let _ = Snmvq::<T>::clear(u32::MAX, None); | ||
| let _ = Snmoq::<T>::clear(u32::MAX, None); | ||
| } | ||
|
|
||
| fn check_storage_exists(n: BlockNumberFor<T>, params: StorageParams) { | ||
| log::debug!(target: LOG_TARGET, "Checking storages exists at: {:?}", n); | ||
| let StorageParams { | ||
| key1, | ||
| key2, | ||
| key3, | ||
| value, | ||
| } = params; | ||
|
|
||
| assert_eq!(Svvq::<T>::get(), value); | ||
| assert_eq!(Svoq::<T>::get(), Some(value)); | ||
| assert_eq!(Smvq::<T>::get(key1), value); | ||
| assert_eq!(Smoq::<T>::get(key1), Some(value)); | ||
| assert_eq!(Sdmvq::<T>::get(key1, key2), value); | ||
| assert_eq!(Sdmoq::<T>::get(key1, key2), Some(value)); | ||
| assert_eq!(Snmvq::<T>::get((key1, key2, key3)), value); | ||
| assert_eq!(Snmoq::<T>::get((key1, key2, key3)), Some(value)); | ||
| } | ||
|
|
||
| fn check_storage_empty(n: BlockNumberFor<T>, params: StorageParams) { | ||
| log::debug!(target: LOG_TARGET, "Checking storages empty at: {:?}", n); | ||
| let StorageParams { | ||
| key1, | ||
| key2, | ||
| key3, | ||
| value: _, | ||
| } = params; | ||
|
|
||
| assert!(!Svvq::<T>::exists()); | ||
|
|
||
| assert!(!Svoq::<T>::exists()); | ||
| assert_eq!(Svoq::<T>::get(), None); | ||
|
|
||
| assert!(!Smvq::<T>::contains_key(key1)); | ||
|
|
||
| assert!(!Smoq::<T>::contains_key(key1)); | ||
| assert_eq!(Smoq::<T>::get(key1), None); | ||
|
|
||
| assert!(!Sdmvq::<T>::contains_key(key1, key2)); | ||
|
|
||
| assert!(!Sdmoq::<T>::contains_key(key1, key2)); | ||
| assert_eq!(Sdmoq::<T>::get(key1, key2), None); | ||
|
|
||
| assert!(!Snmvq::<T>::contains_key((key1, key2, key3))); | ||
|
|
||
| assert!(!Snmoq::<T>::contains_key((key1, key2, key3))); | ||
| assert_eq!(Snmoq::<T>::get((key1, key2, key3)), None); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters