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
use std::ops::{Deref, DerefMut}; pub struct ShouldResult { pass: bool, message: String, } impl ShouldResult { pub fn new(pass: bool, message: String) -> Self { Self { pass, message } } pub fn or(self, other: Self) -> Self { Self::new( self.pass || other.pass, format!("({} || {})", self.message, other.message), ) } pub fn assert(&self) { assert!(self.pass, "{}", self.message) } } impl From<(bool, String)> for ShouldResult { fn from((pass, message): (bool, String)) -> Self { Self::new(pass, message) } } pub struct ResultsContainer { inner: Vec<ShouldResult>, } impl Default for ResultsContainer { fn default() -> Self { Self { inner: Vec::new() } } } impl Drop for ResultsContainer { fn drop(&mut self) { if std::thread::panicking() { return; } for result in self.inner.iter() { result.assert() } } } impl Deref for ResultsContainer { type Target = Vec<ShouldResult>; fn deref(&self) -> &Self::Target { &self.inner } } impl DerefMut for ResultsContainer { fn deref_mut(&mut self) -> &mut Self::Target { &mut self.inner } }