tor_rtmock/
net_runtime.rs

1//! Declare MockNetRuntime.
2
3// TODO(nickm): This is mostly copy-paste from MockSleepRuntime.  If possible,
4// we should make it so that more code is more shared.
5
6use crate::util::impl_runtime_prelude::*;
7
8use crate::net::MockNetProvider;
9
10/// A wrapper Runtime that overrides the SleepProvider trait for the
11/// underlying runtime.
12#[derive(Clone, Debug, Deftly)]
13#[derive_deftly(SomeMockRuntime)]
14pub struct MockNetRuntime<R: Runtime> {
15    /// The underlying runtime. Most calls get delegated here.
16    #[deftly(mock(task, sleep))]
17    runtime: R,
18    /// A MockNetProvider.  Network-related calls get delegated here.
19    #[deftly(mock(net))]
20    net: MockNetProvider,
21}
22
23impl<R: Runtime> MockNetRuntime<R> {
24    /// Create a new runtime that wraps `runtime`, but overrides
25    /// its view of the network with a [`MockNetProvider`], `net`.
26    pub fn new(runtime: R, net: MockNetProvider) -> Self {
27        MockNetRuntime { runtime, net }
28    }
29
30    /// Return a reference to the underlying runtime.
31    pub fn inner(&self) -> &R {
32        &self.runtime
33    }
34
35    /// Return a reference to the [`MockNetProvider`]
36    pub fn mock_net(&self) -> &MockNetProvider {
37        &self.net
38    }
39}
tor_rtmock/net_runtime.rs

tor_rtmock/
net_runtime.rs
