From b3197bbd19f3b4e4094b04e6e63f86beae87df64 Mon Sep 17 00:00:00 2001 From: Andy Date: Wed, 28 Aug 2024 17:36:47 -0400 Subject: [PATCH] Revert "fix(STO-020): check entity stake for new account (#765)" This reverts commit 376d55246914dff85ca506f518f990a76f9b575e. --- crates/sim/src/simulation/simulator.rs | 22 +++++++++------------- 1 file changed, 9 insertions(+), 13 deletions(-) diff --git a/crates/sim/src/simulation/simulator.rs b/crates/sim/src/simulation/simulator.rs index 4b1dbe820..49f1a6de3 100644 --- a/crates/sim/src/simulation/simulator.rs +++ b/crates/sim/src/simulation/simulator.rs @@ -282,23 +282,18 @@ where slot, ) => { let needs_stake_entity = needs_stake.and_then(|t| entity_infos.get(t)); - - if needs_stake.is_none() { - if let Some(factory) = entity_infos.get(EntityType::Factory) { - if factory.is_staked { - tracing::debug!("Associated storage accessed by staked entity during deploy, and factory is staked"); - continue; - } + if let Some(needs_stake_entity) = needs_stake_entity { + if needs_stake_entity.is_staked { + tracing::debug!("Associated storage accessed by staked entity during deploy, and entity is staked"); + continue; } } - - if let Some(needs_stake_entity_info) = needs_stake_entity { - if needs_stake_entity_info.is_staked { - tracing::debug!("Associated storage accessed by staked entity during deploy, and entity is staked"); + if let Some(factory) = entity_infos.get(EntityType::Factory) { + if factory.is_staked { + tracing::debug!("Associated storage accessed by staked entity during deploy, and factory is staked"); continue; } } - // [STO-022] violations.push(SimulationViolation::AssociatedStorageDuringDeploy( needs_stake_entity.map(|ei| ei.entity), @@ -1179,7 +1174,8 @@ mod tests { )] ); - context.entity_infos.paymaster.as_mut().unwrap().is_staked = true; + // staked causes no errors + context.entity_infos.factory.as_mut().unwrap().is_staked = true; let res = simulator.gather_context_violations(&mut context); assert!(res.unwrap().is_empty()); }