Struct MockSource 

pub struct MockSource { /* private fields */ }

Mock source that generates synthetic data for testing and development.

This source runs an internal tokio task that generates data at configurable intervals. It supports different data types (Counter, SensorReading, Generic) to simulate various real-world scenarios.

Event Generation Behavior

• Counter: Always emits INSERT events with sequential IDs
• SensorReading: Emits INSERT for first reading per sensor, UPDATE thereafter
• Generic: Always emits INSERT events with sequential IDs

Thread Safety

This type is Send + Sync and can be safely shared across threads. Internal state is protected by RwLock.

Implementations

impl MockSource

pub fn new(id: impl Into<String>, config: MockSourceConfig) -> Result<Self>

Creates a new MockSource with the given ID and configuration.

The source is created in a stopped state. Call start() to begin generating events, or add it to DrasiLib which will start it automatically (unless auto_start is disabled via the builder).

Arguments

• id - Unique identifier for this source instance. Must be unique within a DrasiLib instance.
• config - Configuration specifying data type and generation interval.

Returns

A new MockSource instance, or an error if validation fails.

Errors

Returns anyhow::Error if:

• config.interval_ms is 0 (would cause spin loop)
• config.data_type is SensorReading with sensor_count of 0

Example

use drasi_source_mock::{MockSource, MockSourceConfig, DataType};

let config = MockSourceConfig {
    data_type: DataType::sensor_reading(10),
    interval_ms: 1000,
};

let source = MockSource::new("my-mock-source", config)?;

pub fn with_dispatch( id: impl Into<String>, config: MockSourceConfig, dispatch_mode: Option<DispatchMode>, dispatch_buffer_capacity: Option<usize>, ) -> Result<Self>

Creates a new MockSource with custom dispatch settings.

This is a lower-level constructor for advanced use cases where you need control over event dispatching. For most cases, prefer MockSource::builder().

Arguments

• id - Unique identifier for this source instance.
• config - Configuration specifying data type and generation interval.
• dispatch_mode - Optional dispatch mode (Channel or Broadcast).
• dispatch_buffer_capacity - Optional buffer size for dispatch channels.

Errors

Returns anyhow::Error if:

• config.interval_ms is 0
• config.data_type is SensorReading with sensor_count of 0

impl MockSource

pub async fn inject_event(&self, change: SourceChange) -> Result<()>

Injects a custom event into the source for testing purposes.

This allows tests to send specific SourceChange events (INSERT, UPDATE, DELETE) without waiting for automatic generation. Useful for deterministic testing scenarios.

The source does not need to be started to inject events.

Arguments

• change - The SourceChange to inject (e.g., SourceChange::Insert { element })

Errors

Returns anyhow::Error if dispatching fails (e.g., all receivers have been dropped).

pub fn test_subscribe(&self) -> Box<dyn ChangeReceiver<SourceEventWrapper>>

Creates a test subscription to receive events from this source.

This bypasses DrasiLib’s subscription mechanism and directly subscribes to the source’s event dispatcher. Useful for unit testing the source in isolation.

Returns

A boxed receiver that yields SourceEventWrapper for each event generated or injected.

Example

let source = MockSource::new("test", config)?;
let mut rx = source.test_subscribe();

source.start().await?;

// Receive events
while let Some(event) = rx.recv().await {
    println!("Received: {:?}", event);
}

impl MockSource

pub fn builder(id: impl Into<String>) -> MockSourceBuilder

Create a builder for MockSource with the given ID.

This is the recommended way to construct a MockSource.

Arguments

• id - Unique identifier for the source instance

Example

let source = MockSource::builder("my-source")
    .with_data_type(DataType::sensor_reading(10))
    .with_interval_ms(1000)
    .build()?;

Trait Implementations

impl Source for MockSource

fn id(&self) -> &str

Get the source’s unique identifier

fn type_name(&self) -> &str

Get the source type name (e.g., “postgres”, “http”, “mock”)

fn properties(&self) -> HashMap<String, Value>

Get the source’s configuration properties for inspection

fn auto_start(&self) -> bool

Whether this source should auto-start when DrasiLib starts

fn start<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Start the source

fn stop<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Stop the source

fn status<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = ComponentStatus> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Get the current status of the source

fn subscribe<'life0, 'async_trait>( &'life0 self, settings: SourceSubscriptionSettings, ) -> Pin<Box<dyn Future<Output = Result<SubscriptionResponse>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Subscribe to this source for change events

fn as_any(&self) -> &dyn Any

Downcast helper for testing - allows access to concrete types

fn initialize<'life0, 'async_trait>( &'life0 self, context: SourceRuntimeContext, ) -> Pin<Box<dyn Future<Output = ()> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Initialize the source with runtime context.

fn set_bootstrap_provider<'life0, 'async_trait>( &'life0 self, provider: Box<dyn BootstrapProvider + 'static>, ) -> Pin<Box<dyn Future<Output = ()> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Set the bootstrap provider for this source

fn dispatch_mode(&self) -> DispatchMode

Get the dispatch mode for this source (Channel or Broadcast)

fn supports_replay(&self) -> bool

Whether this source supports positional replay via resume_from.

fn deprovision<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, Self: 'async_trait,

Permanently clean up internal state when the source is being removed.