diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs index bfc05fad91..775b5447eb 100644 --- a/runtime/src/bank.rs +++ b/runtime/src/bank.rs @@ -1492,7 +1492,8 @@ impl Bank { #[allow(dead_code)] fn is_partitioned_rewards_feature_enabled(&self) -> bool { - false // Will be feature later. It is convenient to have a constant fn at the moment. + self.feature_set + .is_active(&feature_set::enable_partitioned_epoch_reward::id()) } #[allow(dead_code)] diff --git a/runtime/src/bank/tests.rs b/runtime/src/bank/tests.rs index ed6940f7b5..2632c8e55f 100644 --- a/runtime/src/bank/tests.rs +++ b/runtime/src/bank/tests.rs @@ -12604,6 +12604,17 @@ fn test_rewards_point_calculation_empty() { assert!(point_value.is_none()); } +/// Test partitioned_reward feature enable/disable +#[test] +fn test_is_partitioned_reward_enable() { + let (genesis_config, _mint_keypair) = create_genesis_config(1_000_000 * LAMPORTS_PER_SOL); + + let mut bank = Bank::new_for_tests(&genesis_config); + assert!(!bank.is_partitioned_rewards_feature_enabled()); + bank.activate_feature(&feature_set::enable_partitioned_epoch_reward::id()); + assert!(bank.is_partitioned_rewards_feature_enabled()); +} + #[test] fn test_deactivate_epoch_reward_status() { let (genesis_config, _mint_keypair) = create_genesis_config(1_000_000 * LAMPORTS_PER_SOL);