From bf1cf336fed3287213d8b07201f56b3a7329916e Mon Sep 17 00:00:00 2001 From: GregTheDev Date: Fri, 25 Jul 2025 17:42:00 +0700 Subject: [PATCH] chore: make consistent --- crates/cheatcodes/src/credible.rs | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/crates/cheatcodes/src/credible.rs b/crates/cheatcodes/src/credible.rs index 9efb876335d09..0d85d93fcf7e4 100644 --- a/crates/cheatcodes/src/credible.rs +++ b/crates/cheatcodes/src/credible.rs @@ -129,14 +129,18 @@ pub fn execute_assertion( AssertionState::new_active(assertion.create_data.clone().into(), &config) .expect("Failed to create assertion state"); + let mut trigger_types_to_remove = Vec::new(); // Filter triggers for one fn selector - for fn_selectors in assertion_state.trigger_recorder.triggers.values_mut() { + for (trigger_type, fn_selectors) in assertion_state.trigger_recorder.triggers.iter_mut() { if fn_selectors.contains(&assertion.fn_selector) { *fn_selectors = HashSet::from_iter([assertion.fn_selector]); } else { - *fn_selectors = HashSet::new(); + trigger_types_to_remove.push(trigger_type.clone()); } } + for trigger_type in trigger_types_to_remove { + assertion_state.trigger_recorder.triggers.remove(&trigger_type); + } store.insert(assertion.adopter, assertion_state).expect("Failed to store assertions"); let tx_env = TxEnv {