memfault-ssf 1.26.0

Supporting crate for the Memfault memfaultd embedded Linux agent
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

//
// Copyright (c) Memfault, Inc.
// See License.txt for details
use std::sync::mpsc::{Receiver, Sender};

use crate::{Message, MsgMailbox};

/// The ServiceMock allows you to mock a service processing messages of a specific type.
pub struct ServiceMock<M: Message> {
    pub mbox: MsgMailbox<M>,
    receiver: Receiver<(M, Sender<M::Reply>)>,
}

impl<M: Message> ServiceMock<M> {
    pub fn new() -> Self {
        let (mbox, receiver) = MsgMailbox::mock();
        Self { mbox, receiver }
    }

    pub fn new_bounded(channel_size: usize) -> Self {
        let (mbox, receiver) = MsgMailbox::bounded_mock(channel_size);
        Self { mbox, receiver }
    }

    pub fn take_messages(&mut self) -> Vec<M> {
        self.receiver
            .try_iter()
            .map(|(message, _)| message)
            .collect()
    }
}

impl<M: Message> Default for ServiceMock<M> {
    fn default() -> Self {
        Self::new()
    }
}