Struct MockGpuMonitor 

pub struct MockGpuMonitor { /* private fields */ }

A GpuMonitor backed by an in-memory Vec<GpuSnapshot>.

MockGpuMonitor is the workhorse of Concerto’s test suite: it lets tests declare the exact fleet they want to see (including edge cases like overheating GPUs, GPUs with ECC errors, or GPUs that disappear mid-run) and then drive the system under test through those scenarios.

Cloning is cheap — the monitor is backed by an Arc<RwLock<_>> so clones share the same underlying state. This lets a test hold a handle for mutation while also passing the monitor into the system under test.

The lock is a std::sync::RwLock (not tokio::sync::RwLock) so that GpuMonitor::gpu_count, which is synchronous, can read the length without entering the async runtime. Critical sections are always tiny (a clone, a field write, or a retain) so blocking here is harmless.

Implementations

impl MockGpuMonitor

pub fn new(snapshots: Vec<GpuSnapshot>) -> Self

Create a mock monitor from an explicit list of snapshots.

pub fn with_healthy_gpus(count: usize, memory_per_gpu_gb: u64) -> Self

Create a mock monitor reporting count healthy GPUs, each with memory_per_gpu_gb gigabytes of VRAM, zero utilisation, zero memory used, 40 degrees Celsius, and no ECC errors.

pub async fn set_memory_used(&self, gpu_id: GpuId, bytes: ByteSize)

Overwrite the memory_used field of the GPU with the given id.

No-op if the GPU is not present (e.g. it has been removed via MockGpuMonitor::remove_gpu).

pub async fn set_temperature(&self, gpu_id: GpuId, celsius: u32)

Overwrite the temperature_celsius field of the GPU with the given id.

pub async fn inject_ecc_error(&self, gpu_id: GpuId)

Increment the uncorrected ECC error count of the GPU with the given id by one.

pub async fn remove_gpu(&self, gpu_id: GpuId)

Remove a GPU from the monitor’s view, simulating a GPU that has dropped off the bus (driver crash, hardware fault, hot-unplug).

Trait Implementations

impl Clone for MockGpuMonitor

fn clone(&self) -> MockGpuMonitor

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for MockGpuMonitor

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for MockGpuMonitor

fn default() -> MockGpuMonitor

Returns the “default value” for a type.

impl GpuMonitor for MockGpuMonitor

fn gpu_count(&self) -> usize

The number of GPUs this monitor is tracking.

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

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

Take a snapshot of all GPUs currently visible to this monitor.