use crate::reporter::event::Message;
use crate::Event;
#[derive(Clone, Debug, Eq, PartialEq, serde::Serialize)]
#[serde(rename_all = "snake_case")]
pub struct Progress {
current: u64,
search_space_size: u64,
iteration: u64,
}
impl From<Progress> for Event {
fn from(it: Progress) -> Self {
Message::Progress(it).into()
}
}
impl Progress {
pub fn new(current: u64, search_space_size: u64, iteration: u64) -> Self {
Self {
current,
search_space_size,
iteration,
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::reporter::event::Message;
use crate::reporter::TestReporterWrapper;
use storyteller::EventReporter;
#[test]
fn reported_event() {
let reporter = TestReporterWrapper::default();
let event = Progress::new(10, 100, 30);
reporter.get().report_event(event.clone()).unwrap();
assert_eq!(
reporter.wait_for_events(),
vec![Event::unscoped(Message::Progress(event)),]
);
}
}