tlongwell-block
22 lines
702 B
Rust
22 lines
702 B
Rust
//! Test-only utilities for the scheduler
|
|
#![cfg(test)]
|
|
|
|
use once_cell::sync::Lazy;
|
|
use std::sync::Arc;
|
|
use tokio::sync::Mutex;
|
|
|
|
use goose::providers::base::Provider as GooseProvider;
|
|
|
|
static TEST_PROVIDER: Lazy<Mutex<Option<Arc<dyn GooseProvider>>>> = Lazy::new(|| Mutex::new(None));
|
|
|
|
/// Register a default provider for scheduler job executions when running under tests.
|
|
/// The provider will be used by [`Scheduler`] when no provider_override is supplied.
|
|
pub async fn set_test_provider(p: Arc<dyn GooseProvider>) {
|
|
let mut guard = TEST_PROVIDER.lock().await;
|
|
*guard = Some(p);
|
|
}
|
|
|
|
pub async fn get_test_provider() -> Option<Arc<dyn GooseProvider>> {
|
|
TEST_PROVIDER.lock().await.clone()
|
|
}
|