Struct TestDaemonContext 

pub struct TestDaemonContext {
    pub daemon_name: String,
    pub instance_id: Uuid,
    pub shutdown_tx: Sender<bool>,
    /* private fields */
}

Test context for daemon functions.

Provides an isolated testing environment for daemons with shutdown signal control, HTTP mocking, and optional database access.

Example

let ctx = TestDaemonContext::builder("heartbeat_daemon")
    .with_pool(db_pool)
    .build();

// Test shutdown handling
assert!(!ctx.is_shutdown_requested());
ctx.request_shutdown();
assert!(ctx.is_shutdown_requested());

Fields

daemon_name: String

Daemon name.

instance_id: Uuid

Unique instance ID.

shutdown_tx: Sender<bool>

Shutdown signal sender (for triggering shutdown in tests).

Implementations

impl TestDaemonContext

pub fn builder(daemon_name: impl Into<String>) -> TestDaemonContextBuilder

Create a new builder.

pub fn db(&self) -> Option<&PgPool>

Get the database pool (if available).

pub fn http(&self) -> &MockHttp

Get the mock HTTP client.

pub fn is_shutdown_requested(&self) -> bool

Check if shutdown has been requested.

pub fn request_shutdown(&self)

Request shutdown (trigger from test).

pub async fn shutdown_signal(&self)

Wait for shutdown signal.

Use this in a tokio::select! to handle graceful shutdown:

tokio::select! {
    _ = tokio::time::sleep(Duration::from_secs(60)) => {}
    _ = ctx.shutdown_signal() => break,
}

pub async fn heartbeat(&self) -> Result<()>

Simulate heartbeat (no-op in tests).

pub fn env_mock(&self) -> &MockEnvProvider

Get the mock env provider for verification.

Trait Implementations

impl EnvAccess for TestDaemonContext

fn env_provider(&self) -> &dyn EnvProvider

Get the environment provider.

fn env(&self, key: &str) -> Option<String>

Get an environment variable.

fn env_or(&self, key: &str, default: &str) -> String

Get an environment variable with a default value.

fn env_require(&self, key: &str) -> Result<String>

Get a required environment variable.

fn env_parse<T: FromStr>(&self, key: &str) -> Result<T>

where T::Err: Display,

Get an environment variable and parse it to the specified type.

fn env_parse_or<T: FromStr>(&self, key: &str, default: T) -> Result<T>

where T::Err: Display,

Get an environment variable and parse it, with a default.

fn env_contains(&self, key: &str) -> bool

Check if an environment variable is set.