sim-scaffold

name	sim-scaffold
description	Scaffold simulation-testable modules with sans-IO pattern, proptest state machine tests, and fault injection points — prevents retrofitting costs by making code DST-ready from the start
user-invocable	true

Scaffold Simulation-Testable Module

Generate boilerplate for new modules that are deterministic simulation testing (DST) ready from the start. Prevents the costly retrofitting that FoundationDB avoided by building simulation infrastructure before writing database code.

Evidence Base

Source	Pattern Used
Firezone (sans-IO blog)	Pure state machine core, effects as return values
sled (simulation.html)	receive(msg, at) -> [(msg, destination)] pattern
Stateright	Actor trait: on_msg(state, msg) -> Vec<Action>
FoundationDB (SIGMOD 2021)	Simulation-first architecture, SimContext
proptest	State machine testing with prop_state_machine!

When to Use

• Creating a new module in gossip-coordination/
• Adding a new gossip protocol component
• Building a new pipeline stage that touches coordination
• Any new code that manages distributed state, leases, or shard lifecycle

When NOT to Use

• Adding helper functions to existing modules (use /sim-review instead)
• Detection engine rules or regex patterns
• Pure data types with no state transitions

---

Procedure

Step 1: Determine Module Type

Ask the user which type of module they are building:

Type	Description	Template
A: Coordination module	Manages distributed state (shards, leases, epochs)	State machine + InMemory backend
B: Gossip protocol	Message-passing protocol between nodes	Sans-IO state machine
C: Pipeline component	Processing stage with checkpointing	Effect-based with trait boundaries

Step 2: Gather Invariants

Ask the user:

1. What are the key state transitions? (e.g., Open -> Acquired -> Checkpointing -> Done)
2. What invariants must hold? (e.g., "only one worker holds a lease at a time")
3. What are the error conditions? (e.g., "lease expired", "fence rejected")
4. What external dependencies exist? (e.g., persistence, network, timers)

Step 3: Generate Scaffold

Type A: Coordination Module

Generate this file structure under the appropriate crate:

src/<module>/
  mod.rs            — Public API, module docs, re-exports
  state.rs          — State machine types
  logic.rs          — Pure state transition functions
  sim.rs            — Simulation harness (behind test-support feature)
  tests/
    proptest_sm.rs  — proptest state machine tests
    invariants.rs   — Invariant assertion functions

Prerequisite: LogicalTime must be defined in gossip_contracts::identity before the scaffold compiles. If the identity contracts module is still a doc stub, define LogicalTime there first (or use a temporary type alias).

state.rs template:

//! State machine types for <module>.
//!
//! All types are plain data — no I/O, no time reads, no randomness.

use gossip_contracts::identity::LogicalTime;

/// The states this module can be in.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum State {
    // TODO: Fill in states from Step 2
}

/// Inputs that drive state transitions.
#[derive(Debug, Clone)]
pub enum Input {
    // TODO: Fill in inputs from Step 2
}

/// Side effects produced by state transitions.
///
/// Effects are returned as data — the caller is responsible for executing them.
/// This keeps the state machine pure and deterministically testable.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum Effect {
    // TODO: Fill in effects from Step 2
}

logic.rs template:

//! Pure state transition logic for <module>.
//!
//! Every function in this module is a pure function:
//! `fn(current_state, input, now) -> (new_state, Vec<Effect>)`
//!
//! No I/O, no clock reads, no randomness without explicit seed.

use super::state::{Effect, Input, State};
use gossip_contracts::identity::LogicalTime;

/// Apply an input to the current state, producing a new state and effects.
///
/// # Invariants
///
/// TODO: Document invariants from Step 2
pub fn transition(
    state: &State,
    input: &Input,
    now: LogicalTime,
) -> (State, Vec<Effect>) {
    match (state, input) {
        // TODO: Implement transitions
        _ => (state.clone(), vec![]),
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    // TODO: Unit tests for each transition
}

sim.rs template:

//! Simulation harness for <module>.
//!
//! Connects the pure state machine to a deterministic execution environment
//! with seeded PRNG, simulated clock, and fault injection.

#![cfg(feature = "test-support")]

use std::collections::BinaryHeap;

use rand::rngs::StdRng;
use rand::SeedableRng;

use gossip_contracts::identity::LogicalTime;

use super::logic;
use super::state::{Effect, Input, State};

/// Deterministic execution context for simulation.
pub struct SimContext {
    /// Seeded PRNG for reproducible randomness.
    pub rng: StdRng,
    /// Simulated logical clock.
    pub clock: LogicalTime,
    /// Seed used to create this context (for reproduction).
    pub seed: u64,
}

impl SimContext {
    /// Create a new simulation context with the given seed.
    pub fn new(seed: u64) -> Self {
        Self {
            rng: StdRng::seed_from_u64(seed),
            clock: LogicalTime::default(),
            seed,
        }
    }

    /// Advance the simulated clock by `ticks`.
    pub fn advance(&mut self, ticks: u64) {
        self.clock = .clock.(ticks);
    }
}



   {
    
     effect_drop_rate: ,
    
     expire_lease_after: <>,
    
     inject_pause: ,
}


   {
     ctx: SimContext,
     states: <State>,
     effects_log: <(, Effect)>,
     fault_config: FaultConfig,
    op_count: ,
}

  {
      (seed: , initial_states: <State>)   {
         {
            ctx: SimContext::(seed),
            states: initial_states,
            effects_log: ::(),
            fault_config: FaultConfig::(),
            op_count: ,
        }
    }

    
      (& , instance: , input: &Input)  <Effect> {
         (new_state, effects) =
            logic::(&.states[instance], input, .ctx.clock);
        .states[instance] = new_state;
        .op_count += ;

        
           &effects {
            .effects_log
                .((instance, effect.()));
        }

        effects
    }

    
    
    
      (&)  <> {
          = ::();

        
        
        
        
        

        violations
    }
}

tests/proptest_sm.rs template:

//! proptest state machine tests for <module>.
//!
//! Generates random sequences of valid inputs and verifies that
//! invariants hold after every transition.

#![cfg(all(test, feature = "test-support"))]

use proptest::prelude::*;

use super::sim::{FaultConfig, SimHarness};
use super::state::Input;

/// Strategy for generating valid inputs.
fn arb_input() -> impl Strategy<Value = Input> {
    // TODO: Generate valid inputs based on module types
    prop_oneof![
        // Just(Input::VariantA),
        // Just(Input::VariantB { ... }),
    ]
}

/// Strategy for generating input sequences.
fn arb_input_sequence(max_len: usize) -> impl Strategy<Value = Vec<Input>> {
    prop::collection::vec(arb_input(), 1..=max_len)
}

proptest! {
    /// Invariants hold after every transition under normal conditions.
    #[test]
    fn invariants_hold_sunny_day(
        seed in any::<u64>(),
        inputs in arb_input_sequence(50),
    ) {
        let mut harness = SimHarness::new(seed, vec![]);

           &inputs {
            harness.(, input);

              = harness.();
            prop_assert!(
                violations.(),
                ,
                input,
                violations,
            );
        }
    }

    
    
     (
        seed  any::<>(),
        inputs  (),
        drop_rate  ..,
    ) {
          = SimHarness::(seed, []);
        harness.fault_config = FaultConfig {
            effect_drop_rate: drop_rate,
            ..::()
        };

           &inputs {
            harness.(, input);

              = harness.();
            prop_assert!(
                violations.(),
                ,
                drop_rate,
                input,
                violations,
            );
        }
    }
}

tests/invariants.rs template:

//! Invariant assertion functions for <module>.
//!
//! Each function checks one invariant and returns Ok(()) or
//! Err(description) if violated.

#![cfg(all(test, feature = "test-support"))]

use super::state::State;

/// Check: TODO describe invariant
pub fn check_invariant_name(states: &[State]) -> Result<(), String> {
    // TODO: Implement invariant check
    Ok(())
}

---

Type B: Gossip Protocol Module

Generate this file structure:

src/<module>/
  mod.rs              — Public API, module docs
  protocol.rs         — Sans-IO state machine
  messages.rs         — Protocol message types
  sim.rs              — SimNetwork connecting N protocol instances
  tests/
    proptest_convergence.rs  — Convergence property tests
    proptest_partitions.rs   — Partition tolerance tests

protocol.rs template (sans-IO pattern):

//! Sans-IO gossip protocol state machine.
//!
//! This module contains NO I/O. The protocol is driven by four methods:
//!
//! - `handle_input(msg, now)` — process an incoming message
//! - `poll_transmit()` — dequeue the next outbound message
//! - `poll_timeout()` — query when the next timer fires
//! - `handle_timeout(now)` — process a timer expiry
//!
//! The caller (runtime or simulation harness) is responsible for actually
//! sending messages and managing real/simulated time.
//!
//! # Evidence
//!
//! This pattern is used by:
//! - sled: `receive(msg, at) -> [(msg, destination)]`
//! - Firezone: sans-IO connlib architecture
//! - Stateright: `Actor::on_msg(state, msg) -> Vec<Action>`

use std::collections::VecDeque;

use gossip_contracts::identity::LogicalTime;

use super::messages::{GossipMessage, NodeId, Transmit};

/// Configuration for the gossip protocol.
#[derive(Debug, Clone)]
pub struct ProtocolConfig {
    /// This node's identity.
    pub node_id: NodeId,
    /// Gossip interval (in logical time ticks).
    pub gossip_interval: u64,
    /// Failure detection timeout (in logical time ticks).
    pub failure_timeout: u64,
    /// Fan-out: number of peers to gossip to per round.
    pub fanout: usize,
}

/// Sans-IO gossip protocol state machine.
pub   {
    config: ProtocolConfig,
    outbox: VecDeque<Transmit>,
    next_gossip: <LogicalTime>,
    
}

  {
    
      (config: ProtocolConfig, now: LogicalTime)   {
         {
            next_gossip: (now.(config.gossip_interval)),
            outbox: VecDeque::(),
            config,
        }
    }

    
    
    
    
      (& , msg: &GossipMessage, now: LogicalTime) {
         msg {
            
            _ => {}
        }
    }

    
    
    
    
      (& )  <Transmit> {
        .outbox.()
    }

    
    
    
      (&)  <LogicalTime> {
        .next_gossip
    }

    
    
    
    
      (& , now: LogicalTime) {
         .next_gossip.(|t| now >= t) {
            .(now);
            .next_gossip = (now.(.config.gossip_interval));
        }
    }

     (& , _now: LogicalTime) {
        
    }
}

sim.rs template (SimNetwork):

//! Simulation network connecting N gossip protocol instances.
//!
//! Messages are delivered via an in-process priority queue ordered by
//! logical time, following sled's discrete-event simulation pattern.

#![cfg(feature = "test-support")]

use std::cmp::Reverse;
use std::collections::BinaryHeap;

use rand::rngs::StdRng;
use rand::SeedableRng;

use gossip_contracts::identity::LogicalTime;

use super::messages::{NodeId, Transmit};
use super::protocol::{GossipProtocol, ProtocolConfig};

/// A scheduled event in the simulation.
#[derive(Debug)]
struct ScheduledEvent {
    time: LogicalTime,
    kind: EventKind,
}

#[derive(Debug)]
enum EventKind {
    Deliver(Transmit),
    Timeout(NodeId),
}

impl PartialEq for ScheduledEvent {
    fn eq(&self, other: &Self) -> bool {
        self.time == other.time
    }
}

impl Eq for ScheduledEvent {}

impl PartialOrd for ScheduledEvent {
    fn partial_cmp(&self, other: &Self)  <std::cmp::Ordering> {
        (.(other))
    }
}

    {
     (&, other: &)  std::cmp::Ordering {
        
        (.time).(&(other.time))
    }
}



   {
    
     drop_rate: ,
    
     max_delay: ,
    
     partitions: <(NodeId, NodeId)>,
}


   {
     nodes: <GossipProtocol>,
     rng: StdRng,
     clock: LogicalTime,
    event_queue: BinaryHeap<ScheduledEvent>,
     faults: NetworkFaults,
     seed: ,
}

  {
    
      (seed: , n: , config_fn:  ()  ProtocolConfig)   {
          = LogicalTime::();
          = (..n)
            .(|i| GossipProtocol::((i), clock))
            .collect::<<_>>();

          =  {
            nodes,
            rng: StdRng::(seed),
            clock,
            event_queue: BinaryHeap::(),
            faults: NetworkFaults::(),
            seed,
        };

        
         (i, node)  net.nodes.().() {
              (t) = node.() {
                net.event_queue.(ScheduledEvent {
                    time: t,
                    kind: EventKind::((i  )),
                });
            }
        }

        net
    }

    
    
    
      (& , max_ticks: )   {
          = .clock.(max_ticks);
          = ;

          (event) = .event_queue.() {
             event.time > deadline {
                .event_queue.(event);
                ;
            }

            .clock = event.time;
            events_processed += ;

             event.kind {
                EventKind::(transmit) => {
                      = transmit.destination.  ;
                     dst < .nodes.() {
                        .nodes[dst].(&transmit.message, .clock);
                        .(dst);
                    }
                }
                EventKind::(node_id) => {
                      = node_id.  ;
                     idx < .nodes.() {
                        .nodes[idx].(.clock);
                        .(idx);

                        
                          (t) = .nodes[idx].() {
                            .event_queue.(ScheduledEvent {
                                time: t,
                                kind: EventKind::(node_id),
                            });
                        }
                    }
                }
            }
        }

        events_processed
    }

     (& , node_idx: ) {
          (transmit) = .nodes[node_idx].() {
            
            
            .event_queue.(ScheduledEvent {
                time: .clock.(), 
                kind: EventKind::(transmit),
            });
        }
    }

    
      (&)   {
        
        
    }
}

---

Type C: Pipeline Component

Generate this file structure:

src/<component>/
  mod.rs          — Public API
  state.rs        — Processing state types
  logic.rs        — Pure processing logic
  traits.rs       — I/O trait boundaries
  sim.rs          — Simulation with mock I/O
  tests/
    proptest_processing.rs

Use the same SimContext pattern as Type A, but the trait boundaries focus on I/O operations (read source data, write checkpoints, emit findings) rather than distributed coordination.

---

Step 4: Verify Scaffold Compiles

After generating the scaffold:

cargo check --all-features

If compilation fails, fix the generated code. Common issues:

• Missing imports (add use statements)
• Type mismatches with existing crate types
• Feature gate misalignment

Step 5: Generate Initial Tests

Run the generated proptest to verify the scaffold works:

cargo test --features test-support -- <module>::tests

The initial tests should pass (they test the empty/default state machine).

Step 6: Report

SIM-SCAFFOLD REPORT
════════════════════
Module:          <name>
Type:            {Coordination | Gossip Protocol | Pipeline Component}
Files created:   {count}
Pattern:         {sans-IO | state machine + effects | trait boundary}

Invariants to implement:
  1. {invariant from Step 2}
  2. {invariant from Step 2}

Next steps:
  1. Implement state transitions in logic.rs
  2. Add proptest strategies for Input variants
  3. Implement invariant checks in sim.rs
  4. Run /sim-review on completed implementation

---

Quick Reference: Module Types

	Type A: Coordination	Type B: Gossip Protocol	Type C: Pipeline
Manages	Distributed state (shards, leases, epochs)	Message-passing between nodes	Processing stages with checkpoints
Core pattern	State machine + InMemory backend	Sans-IO (handle_input/poll_transmit/poll_timeout/handle_timeout)	Effect-based with trait boundaries
sim-review boundary	STRICT	STRICT	MODERATE
Time model	LogicalTime parameter on every op	LogicalTime parameter on every method	LogicalTime at checkpoint boundaries
Key test	Proptest state machine (invariants after every transition)	Proptest convergence + partition tolerance	Proptest processing correctness
Sim harness	SimHarness (single-node state machine)	SimNetwork (N-node discrete-event sim)	SimContext + mock I/O traits
Fault injection	Lease expiry, concurrent access, clock advance	Message loss/reorder, partitions, delays	I/O failure, corrupt input

---

Related Skills

• /sim-review — Review existing code for DST compatibility
• /sim-run — Execute simulation tests
• /test-strategy — Choose between unit/property/fuzz/simulation testing
• /dist-sys-auditor — Validate distributed systems design decisions