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 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
// Copyright 2019, The Tari Project // // Redistribution and use in source and binary forms, with or without modification, are permitted provided that the // following conditions are met: // // 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following // disclaimer. // // 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the // following disclaimer in the documentation and/or other materials provided with the distribution. // // 3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote // products derived from this software without specific prior written permission. // // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, // INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE // DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, // WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE // USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. use futures::FutureExt; use std::{future::Future, pin::Pin}; use tokio::{runtime, runtime::Runtime, task, task::JoinError}; pub fn create_runtime() -> Runtime { tokio::runtime::Builder::new() .threaded_scheduler() .enable_io() .enable_time() .max_threads(8) .core_threads(4) .build() .expect("Could not create runtime") } /// Create a runtime and report if it panics. If there are tasks still running after the panic, this /// will carry on running forever. // #[deprecated(note = "use tokio_macros::test instead")] pub fn test_async<F>(f: F) where F: FnOnce(&mut TestRuntime) { let mut rt = TestRuntime::from(create_runtime()); f(&mut rt); let handles = rt.handles.drain(..).collect::<Vec<_>>(); for h in handles { rt.block_on(h).unwrap(); } } type BoxedJoinFuture = Pin<Box<dyn Future<Output = Result<(), JoinError>>>>; pub struct TestRuntime { inner: Runtime, handles: Vec<BoxedJoinFuture>, } impl TestRuntime { pub fn block_on<F: Future>(&mut self, future: F) -> F::Output { self.inner.block_on(future) } pub fn spawn<F>(&mut self, future: F) where F: Future + Send + 'static, F::Output: Send + 'static, { let handle = self.inner.spawn(future); self.handles.push( handle .map(|result| match result { Ok(_) => Ok(()), Err(err) => Err(err), }) .boxed(), ); } pub fn spawn_unchecked<F>(&mut self, future: F) -> task::JoinHandle<F::Output> where F: Future + Send + 'static, F::Output: Send + 'static, { self.inner.spawn(future) } pub fn handle(&self) -> &runtime::Handle { self.inner.handle() } } impl From<Runtime> for TestRuntime { fn from(rt: Runtime) -> Self { Self { inner: rt, handles: Vec::new(), } } }