fix(STO-020): check entity stake for new account (#765)

alchemyplatform · Aug 26, 2024 · 376d552 · 376d552
1 parent f94f258
commit 376d552
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/crates/sim/src/simulation/simulator.rs b/crates/sim/src/simulation/simulator.rs
@@ -282,18 +282,23 @@ where
                             slot,
                         ) => {
                             let needs_stake_entity = needs_stake.and_then(|t| entity_infos.get(t));
-                            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 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(factory) = entity_infos.get(EntityType::Factory) {
-                                if factory.is_staked {
-                                    tracing::debug!("Associated storage accessed by staked entity during deploy, and factory is staked");
+
+                            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");
                                     continue;
                                 }
                             }
+
                             // [STO-022]
                             violations.push(SimulationViolation::AssociatedStorageDuringDeploy(
                                 needs_stake_entity.map(|ei| ei.entity),
@@ -1174,8 +1179,7 @@ mod tests {
             )]
         );
 
-        // staked causes no errors
-        context.entity_infos.factory.as_mut().unwrap().is_staked = true;
+        context.entity_infos.paymaster.as_mut().unwrap().is_staked = true;
         let res = simulator.gather_context_violations(&mut context);
         assert!(res.unwrap().is_empty());
     }