Struct MockAllocWrapper

pub struct MockAllocWrapper {
    pub alloc: MockAlloc,
    pub block_next_after: usize,
    /* private fields */
}

Test wrapper around MockAlloc to allow us to block next() calls since mockall doesn’t support returning futures.

Fields

alloc: MockAlloc

block_next_after: usize

Implementations

impl MockAllocWrapper

pub fn new(alloc: MockAlloc) -> Self

pub fn new_block_next(alloc: MockAlloc, count: usize) -> Self

pub fn notify_tx(&self) -> Sender<()>

Trait Implementations

impl Alloc for MockAllocWrapper

fn next<'life0, 'async_trait>( &'life0 mut self, ) -> Pin<Box<dyn Future<Output = Option<ProcState>> + Send + 'async_trait>>

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

Return the next proc event. None indicates that there are no more events, and that the alloc is stopped.

fn extent(&self) -> &Extent

The shape of the alloc.

fn world_id(&self) -> &WorldId

The world id of this alloc, uniquely identifying the alloc. Note: This will be removed in favor of a different naming scheme, once we exise “worlds” from hyperactor core.

fn transport(&self) -> ChannelTransport

The channel transport used the procs in this alloc.

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

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

Stop this alloc, shutting down all of its procs. A clean shutdown should result in Stop events from all allocs, followed by the end of the event stream.

fn shape(&self) -> Shape

The shape of the alloc. (Deprecated.)

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

where Self: Send + 'async_trait, 'life0: 'async_trait,

Stop this alloc and wait for all procs to stop. Call will block until all ProcState events have been drained.