git.delta.rocks / unique-network / refs/commits / 94b124f78550

3636
37use sp_runtime::{37use sp_runtime::{
38	Perbill,38	Perbill,
39	traits::{Zero},39	traits::{BlockNumberProvider, Zero},
40};40};
41use sp_std::convert::TryInto;41use sp_std::convert::TryInto;
4242
57	type TreasuryAccountId: Get<Self::AccountId>;57	type TreasuryAccountId: Get<Self::AccountId>;
58	type InflationBlockInterval: Get<Self::BlockNumber>;58	type InflationBlockInterval: Get<Self::BlockNumber>;
59
60	// The block number provider
61	type BlockNumberProvider: BlockNumberProvider<BlockNumber = Self::BlockNumber>;
59}62}
6063
61decl_storage! {64decl_storage! {
75	{78	{
76		const InflationBlockInterval: T::BlockNumber = T::InflationBlockInterval::get();79		const InflationBlockInterval: T::BlockNumber = T::InflationBlockInterval::get();
7780
78		fn on_initialize(now: T::BlockNumber) -> Weight81		fn on_initialize() -> Weight
79		{82		{
80			let mut consumed_weight = 0;83			let mut consumed_weight = 0;
81			let mut add_weight = |reads, writes, weight| {84			let mut add_weight = |reads, writes, weight| {
84			};87			};
8588
86			let block_interval: u32 = T::InflationBlockInterval::get().try_into().unwrap_or(0);89			let block_interval: u32 = T::InflationBlockInterval::get().try_into().unwrap_or(0);
8790			let _now = T::BlockNumberProvider::current_block_number();
88			// TODO: Rewrite inflation to use block timestamp instead of block number
89			// let _now = <timestamp::Module<T>>::get();
9091
91			// Recalculate inflation on the first block of the year (or if it is not initialized yet)92			// Recalculate inflation on the first block of the year (or if it is not initialized yet)
92			if (now % T::BlockNumber::from(YEAR)).is_zero() || <BlockInflation<T>>::get().is_zero() {93			if (_now % T::BlockNumber::from(YEAR)).is_zero() || <BlockInflation<T>>::get().is_zero() {
93				let current_year: u32 = (now / T::BlockNumber::from(YEAR)).try_into().unwrap_or(0);94				let current_year: u32 = (_now / T::BlockNumber::from(YEAR)).try_into().unwrap_or(0);
9495
95				let one_percent = Perbill::from_percent(1);96				let one_percent = Perbill::from_percent(1);
9697
117			}118			}
118119
119			// Apply inflation every InflationBlockInterval blocks and in the 1st block to initialize Treasury account120			// Apply inflation every InflationBlockInterval blocks and in the 1st block to initialize Treasury account
120			else if (now % T::BlockNumber::from(block_interval)).is_zero() {121			else if (_now % T::BlockNumber::from(block_interval)).is_zero() {
121				T::Currency::deposit_into_existing(&T::TreasuryAccountId::get(), <BlockInflation<T>>::get()).ok();122				T::Currency::deposit_into_existing(&T::TreasuryAccountId::get(), <BlockInflation<T>>::get()).ok();
122123
123				add_weight(3, 2, 12_900_000);124				add_weight(3, 2, 12_900_000);

--- a/pallets/inflation/src/tests.rs
+++ b/pallets/inflation/src/tests.rs
@@ -11,7 +11,7 @@
 };
 use sp_core::H256;
 use sp_runtime::{
-	traits::{BlakeTwo256, IdentityLookup},
+	traits::{BlakeTwo256, BlockNumberProvider, IdentityLookup},
 	testing::Header,
 };
 
@@ -85,12 +85,22 @@
 parameter_types! {
 	pub TreasuryAccountId: u64 = 1234;
 	pub const InflationBlockInterval: u32 = 100; // every time per how many blocks inflation is applied
+	pub static MockBlockNumberProvider: u64 = 0;
 }
 
+impl BlockNumberProvider for MockBlockNumberProvider {
+	type BlockNumber = u64;
+
+	fn current_block_number() -> Self::BlockNumber {
+		Self::get()
+	}
+}
+
 impl pallet_inflation::Config for Test {
 	type Currency = Balances;
 	type TreasuryAccountId = TreasuryAccountId;
 	type InflationBlockInterval = InflationBlockInterval;
+	type BlockNumberProvider = MockBlockNumberProvider;
 }
 
 pub fn new_test_ext() -> sp_io::TestExternalities {
@@ -110,7 +120,8 @@
 
 		// BlockInflation should be set after 1st block and
 		// first inflation deposit should be equal to BlockInflation
-		Inflation::on_initialize(1);
+		MockBlockNumberProvider::set(1);
+		Inflation::on_initialize(0);
 
 		// Expected 100-block inflation for year 1 is 100 * 100_000_000 / YEAR = 3803
 		assert_eq!(Inflation::block_inflation(), 3803);
@@ -128,20 +139,23 @@
 		let initial_issuance: u64 = 1_000_000_000;
 		let _ = <Balances as Currency<_>>::deposit_creating(&1234, initial_issuance);
 		assert_eq!(Balances::free_balance(1234), initial_issuance);
-		Inflation::on_initialize(1);
+		MockBlockNumberProvider::set(1);
+		Inflation::on_initialize(0);
 
 		// Next inflation deposit happens when block is multiple of InflationBlockInterval
 		let mut block: u32 = 2;
 		let balance_before: u64 = Balances::free_balance(1234);
 		while block % InflationBlockInterval::get() != 0 {
-			Inflation::on_initialize(block as u64);
+			MockBlockNumberProvider::set(block as u64);
+			Inflation::on_initialize(0);
 			block += 1;
 		}
 		let balance_just_before: u64 = Balances::free_balance(1234);
 		assert_eq!(balance_before, balance_just_before);
 
 		// The block with inflation
-		Inflation::on_initialize(block as u64);
+		MockBlockNumberProvider::set(block as u64);
+		Inflation::on_initialize(0);
 		let balance_after: u64 = Balances::free_balance(1234);
 		assert_eq!(
 			balance_after - balance_just_before,
@@ -157,19 +171,22 @@
 		let initial_issuance: u64 = 1_000_000_000;
 		let _ = <Balances as Currency<_>>::deposit_creating(&1234, initial_issuance);
 		assert_eq!(Balances::free_balance(1234), initial_issuance);
-		Inflation::on_initialize(1);
+		MockBlockNumberProvider::set(1);
+		Inflation::on_initialize(0);
 
 		// Go through all the block inflations for year 1,
 		// total issuance will be updated accordingly
 		for block in (100..YEAR).step_by(100) {
-			Inflation::on_initialize(block);
+			MockBlockNumberProvider::set(block);
+			Inflation::on_initialize(0);
 		}
 		assert_eq!(
 			initial_issuance + (3803 * (YEAR / 100)),
 			<Balances as Currency<_>>::total_issuance()
 		);
 
-		Inflation::on_initialize(YEAR);
+		MockBlockNumberProvider::set(YEAR);
+		Inflation::on_initialize(0);
 		let block_inflation_year_1 = Inflation::block_inflation();
 		// Expected 100-block inflation for year 2: 100 * 9.33% * initial issuance * 110% / YEAR = 3904
 		assert_eq!(block_inflation_year_1, 3904);
@@ -181,13 +198,16 @@
 	new_test_ext().execute_with(|| {
 		// Total issuance = 1_000_000_000
 		let initial_issuance: u64 = 1_000_000_000;
+
 		let _ = <Balances as Currency<_>>::deposit_creating(&1234, initial_issuance);
 		assert_eq!(Balances::free_balance(1234), initial_issuance);
-		Inflation::on_initialize(1);
+		MockBlockNumberProvider::set(1);
+		Inflation::on_initialize(0);
 
 		for year in 1..=9 {
 			let block_inflation_year_before = Inflation::block_inflation();
-			Inflation::on_initialize(YEAR * year);
+			MockBlockNumberProvider::set(YEAR * year);
+			Inflation::on_initialize(0);
 			let block_inflation_year_after = Inflation::block_inflation();
 
 			// SBP M2 review: this is actually not true (not for the first few years)
@@ -204,11 +224,13 @@
 		let initial_issuance: u64 = 1_000_000_000;
 		let _ = <Balances as Currency<_>>::deposit_creating(&1234, initial_issuance);
 		assert_eq!(Balances::free_balance(1234), initial_issuance);
-		Inflation::on_initialize(YEAR * 9);
+		MockBlockNumberProvider::set(YEAR * 9);
+		Inflation::on_initialize(0);
 
 		for year in 10..=20 {
 			let block_inflation_year_before = Inflation::block_inflation();
-			Inflation::on_initialize(YEAR * year);
+			MockBlockNumberProvider::set(YEAR * year);
+			Inflation::on_initialize(0);
 			let block_inflation_year_after = Inflation::block_inflation();
 
 			// Assert that next year inflation is equal to previous year inflation
@@ -233,22 +255,26 @@
 
 		for year in 0..=10 {
 			// Year first block
-			Inflation::on_initialize(year * YEAR);
+			MockBlockNumberProvider::set(YEAR * year);
+			Inflation::on_initialize(0);
 			let mut actual_payout = Inflation::block_inflation();
 			assert_eq!(actual_payout, payout_by_year[year as usize]);
 
 			// Year second block
-			Inflation::on_initialize(year * YEAR + 1);
+			MockBlockNumberProvider::set(YEAR * year + 1);
+			Inflation::on_initialize(0);
 			actual_payout = Inflation::block_inflation();
 			assert_eq!(actual_payout, payout_by_year[year as usize]);
 
 			// Year middle block
-			Inflation::on_initialize(year * YEAR + YEAR / 2);
+			MockBlockNumberProvider::set(year * YEAR + YEAR / 2);
+			Inflation::on_initialize(0);
 			actual_payout = Inflation::block_inflation();
 			assert_eq!(actual_payout, payout_by_year[year as usize]);
 
 			// Year last block
-			Inflation::on_initialize((year + 1) * YEAR - 1);
+			MockBlockNumberProvider::set((year + 1) * YEAR - 1);
+			Inflation::on_initialize(0);
 			actual_payout = Inflation::block_inflation();
 			assert_eq!(actual_payout, payout_by_year[year as usize]);
 		}

--- a/runtime/src/lib.rs
+++ b/runtime/src/lib.rs
@@ -768,6 +768,7 @@
 	type Currency = Balances;
 	type TreasuryAccountId = TreasuryAccountId;
 	type InflationBlockInterval = InflationBlockInterval;
+	type BlockNumberProvider = RelayChainBlockNumberProvider<Runtime>;
 }
 
 // parameter_types! {