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
38
39
40
41
42
43

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

use crate::{Envelope, Mailbox, Service};

/// The ServiceJig allows you to create a mailbox for a service and control when
/// messages will be processed. It does not use thread and is specifically well
/// suited for unit tests.
pub struct ServiceJig<S: Service> {
    pub mailbox: Mailbox<S>,
    service: S,
    receiver: Receiver<Envelope<S>>,
}

impl<S: Service> ServiceJig<S> {
    pub fn prepare(service: S) -> Self {
        let (mailbox, receiver) = Mailbox::create();

        ServiceJig {
            service,
            receiver,
            mailbox,
        }
    }

    /// Process all waiting messages
    pub fn process_all(&mut self) {
        let iter = self.receiver.try_iter();
        for mut m in iter {
            let _x = m.deliver_to(&mut self.service);
        }
    }

    pub fn get_service(&self) -> &S {
        &self.service
    }

    pub fn get_service_mut(&mut self) -> &mut S {
        &mut self.service
    }
}