fix: 🐛 test

bifrost-io · Dec 17, 2024 · 193c8a1 · 193c8a1
1 parent 232e77a
commit 193c8a1
Show file tree

Hide file tree

Showing 8 changed files with 16 additions and 2 deletions.
diff --git a/pallets/buy-back/src/mock.rs b/pallets/buy-back/src/mock.rs
@@ -491,6 +491,7 @@ impl bb_bnc::Config for Runtime {
 	type FarmingInfo = ();
 	type FourYears = MaxBlock;
 	type OneYear = OneYear;
+	type BlockNumberProvider = System;
 }
 
 pub struct ExtBuilder {

diff --git a/pallets/buy-back/src/tests.rs b/pallets/buy-back/src/tests.rs
@@ -243,8 +243,11 @@ fn on_initialize_no_burn_should_work() {
 				Some(buyback_account.clone()),
 			);
 			assert_ok!(BuyBack::charge(RuntimeOrigin::signed(ALICE), VKSM, 1000));
+			System::set_block_number(1);
 			BuyBack::on_initialize(1);
+			System::set_block_number(2);
 			BuyBack::on_initialize(2);
+			System::set_block_number(3);
 			BuyBack::on_initialize(3);
 			assert_eq!(Currencies::free_balance(VKSM, &buyback_account), 9000);
 			assert_eq!(
@@ -301,10 +304,10 @@ fn on_initialize_with_burn_should_work() {
 				Some(buyback_account.clone()),
 			);
 			assert_ok!(BuyBack::charge(RuntimeOrigin::signed(ALICE), VKSM, 1000));
-			BuyBack::on_initialize(<frame_system::Pallet<Runtime>>::block_number() + 1);
 			System::set_block_number(System::block_number() + 1);
-			BuyBack::on_initialize(<frame_system::Pallet<Runtime>>::block_number() + 1);
+			BuyBack::on_initialize(0);
 			System::set_block_number(System::block_number() + 1);
+			BuyBack::on_initialize(0);
 			assert_eq!(Currencies::free_balance(VKSM, &buyback_account), 9000);
 			assert_eq!(
 				Currencies::free_balance(VKSM, &zenlink_pair_account_id),
@@ -360,6 +363,7 @@ fn on_initialize_with_bias_should_work() {
 				Some(buyback_account.clone()),
 			);
 			assert_ok!(BuyBack::charge(RuntimeOrigin::signed(ALICE), VKSM, 1000));
+			System::set_block_number(1);
 			BuyBack::on_initialize(1);
 			let path = vec![
 				AssetId::try_convert_from(VKSM, PARAID).unwrap(),
@@ -374,7 +378,9 @@ fn on_initialize_with_bias_should_work() {
 				<frame_system::Pallet<Runtime>>::block_number()
 					+ BlockNumberFor::<Runtime>::from(1u32)
 			));
+			System::set_block_number(2);
 			BuyBack::on_initialize(2);
+			System::set_block_number(3);
 			BuyBack::on_initialize(3);
 			assert_eq!(Currencies::free_balance(VKSM, &buyback_account), 9000);
 			assert_eq!(

diff --git a/pallets/clouds-convert/src/mock.rs b/pallets/clouds-convert/src/mock.rs
@@ -205,6 +205,7 @@ impl bb_bnc::Config for Runtime {
 	type FarmingInfo = ();
 	type FourYears = MaxBlock;
 	type OneYear = OneYear;
+	type BlockNumberProvider = System;
 }
 
 pub struct ExtBuilder {

diff --git a/pallets/farming/src/mock.rs b/pallets/farming/src/mock.rs
@@ -183,6 +183,7 @@ impl bb_bnc::Config for Runtime {
 	type FarmingInfo = Farming;
 	type FourYears = MaxBlock;
 	type OneYear = OneYear;
+	type BlockNumberProvider = System;
 }
 
 ord_parameter_types! {

diff --git a/pallets/leverage-staking/src/mock.rs b/pallets/leverage-staking/src/mock.rs
@@ -252,6 +252,7 @@ impl bifrost_stable_asset::Config for Test {
 	type WeightInfo = ();
 	type ListingOrigin = EnsureSignedBy<One, u128>;
 	type EnsurePoolAssetId = EnsurePoolAssetId;
+	type BlockNumberProvider = System;
 }
 
 impl bifrost_stable_pool::Config for Test {
@@ -454,6 +455,7 @@ impl lend_market::Config for Test {
 	type RewardAssetId = RewardAssetId;
 	type LiquidationFreeAssetId = LiquidationFreeAssetId;
 	type MaxLengthLimit = MaxLengthLimit;
+	type BlockNumberProvider = System;
 }
 
 impl pallet_prices::Config for Test {

diff --git a/pallets/slp/src/mocks/mock_kusama.rs b/pallets/slp/src/mocks/mock_kusama.rs
@@ -109,6 +109,7 @@ impl bifrost_stable_asset::Config for Runtime {
 	type WeightInfo = ();
 	type ListingOrigin = EnsureSignedBy<One, AccountId>;
 	type EnsurePoolAssetId = EnsurePoolAssetId;
+	type BlockNumberProvider = System;
 }
 
 parameter_types! {

diff --git a/pallets/stable-pool/src/mock.rs b/pallets/stable-pool/src/mock.rs
@@ -240,6 +240,7 @@ impl bifrost_stable_asset::Config for Test {
 	type WeightInfo = ();
 	type ListingOrigin = EnsureSignedBy<One, u128>;
 	type EnsurePoolAssetId = EnsurePoolAssetId;
+	type BlockNumberProvider = System;
 }
 
 impl bifrost_stable_pool::Config for Test {

diff --git a/pallets/system-staking/src/mock.rs b/pallets/system-staking/src/mock.rs
@@ -298,6 +298,7 @@ impl bifrost_farming::Config for Runtime {
 	type BlockNumberToBalance = ConvertInto;
 	type WhitelistMaximumLimit = WhitelistMaximumLimit;
 	type GaugeRewardIssuer = FarmingGaugeRewardIssuerPalletId;
+	type BlockNumberProvider = System;
 }
 
 parameter_types! {