fix the test

ComposableFi · Dec 7, 2023 · 5794d43 · 5794d43
1 parent f572af5
commit 5794d43
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/code/parachain/frame/liquid-staking/src/tests.rs b/code/parachain/frame/liquid-staking/src/tests.rs
@@ -705,7 +705,8 @@ fn test_on_initialize_work() {
 		LiquidStaking::on_finalize(System::block_number());
 		System::set_block_number(System::block_number() + 1);
 		LiquidStaking::on_initialize(System::block_number());
-		assert_eq!(EraStartBlock::<Test>::get(), total_era_blocknumbers);
+		let current_era: u64 = CurrentEra::<Test>::get() as u64;
+		assert_eq!(EraStartBlock::<Test>::get(), total_era_blocknumbers * current_era);
 		// ValidationDataProvider return relay_parent_number = 100
 		// total_era_blocknumbers  = 10 so the current era is 10
 		assert_eq!(CurrentEra::<Test>::get(), 10);