Fix compute budget bump (#21238)

This commit is contained in:
Jack May
2021-11-11 14:09:28 -08:00
committed by GitHub
parent 4071761daf
commit 131c0d78c4
7 changed files with 153 additions and 55 deletions

View File

@ -373,6 +373,18 @@ impl FeatureSet {
inactive: HashSet::new(),
}
}
/// Activate a feature
pub fn activate(&mut self, feature_id: &Pubkey, slot: u64) {
self.inactive.remove(feature_id);
self.active.insert(*feature_id, slot);
}
/// Deactivate a feature
pub fn deactivate(&mut self, feature_id: &Pubkey) {
self.active.remove(feature_id);
self.inactive.insert(*feature_id);
}
}
#[cfg(test)]
@ -433,4 +445,16 @@ mod test {
.collect()
);
}
#[test]
fn test_feature_set_activate_deactivate() {
let mut feature_set = FeatureSet::default();
let feature = Pubkey::new_unique();
assert!(!feature_set.is_active(&feature));
feature_set.activate(&feature, 0);
assert!(feature_set.is_active(&feature));
feature_set.deactivate(&feature);
assert!(!feature_set.is_active(&feature));
}
}