use automato_sync::statemachine; use serde::{Deserialize, Serialize}; use serde_json; use std::marker::PhantomData; #[derive(Serialize, Deserialize)] pub struct JobData {} #[derive(Serialize, Deserialize)] pub struct QueuedData {} #[derive(Serialize, Deserialize)] pub struct ProcessingData {} #[derive(Serialize, Deserialize)] pub struct CompletedData {} statemachine! { Job: JobData { init Queued: QueuedData { start => Processing }, Processing: ProcessingData { complete => Completed, queue => Queued }, Completed: CompletedData {} } } struct Log {} impl Observer<()> for Log { type Error = (); } #[test] fn init() { let _job = Job::init( &mut (), Log {}, Some("foo".to_string()), JobData {}, QueuedData {}, ) .unwrap(); } #[test] fn init_without_id() { let result = Job::init(&mut (), Log {}, None, JobData {}, QueuedData {}); let err = result.err().unwrap(); match err { InitError::EmptyId => {} _ => panic!("expected InitErr::EmptyId"), }; } #[test] fn init_with_deferred_id() { struct DeferredIdInitLog {} impl Observer<()> for DeferredIdInitLog { type Error = (); fn on_init( &mut self, _ctx: &mut (), _id: Option, _to: State, _data: Option, _state_data: Option, ) -> Result, Self::Error> { Ok(Some("foo".to_string())) } } let _job = Job::init( &mut (), DeferredIdInitLog {}, None, JobData {}, QueuedData {}, ) .unwrap(); } #[test] fn on_init() { struct InitLog { initiated_to_state: Option, } impl Observer<()> for &mut InitLog { type Error = (); fn on_init( &mut self, _ctx: &mut (), id: Option, to: State, _data: Option, _state_data: Option, ) -> Result, Self::Error> { self.initiated_to_state = Some(to); Ok(id) } } let mut init_log = InitLog { initiated_to_state: None, }; let _job = Job::init( &mut (), &mut init_log, Some("foo".to_string()), JobData {}, QueuedData {}, ) .unwrap(); match init_log.initiated_to_state { Some(state) => assert_eq!("Queued", state.to_string()), None => panic!("expected some initiated_to_state value"), }; } #[test] fn read_id() { let job = Job::init( &mut (), Log {}, Some("foo".to_string()), JobData {}, QueuedData {}, ) .unwrap(); let id = job.id(); assert_eq!(id, "foo"); } #[test] fn read_data() { let job = Job::init( &mut (), Log {}, Some("foo".to_string()), JobData {}, QueuedData {}, ) .unwrap(); let _job_data = job.data(); let _job_state_data = job.state.data(); } #[test] fn transition() { let job = Job::init( &mut (), Log {}, Some("foo".to_string()), JobData {}, QueuedData {}, ) .unwrap(); let _job = job.start(&mut (), ProcessingData {}).unwrap(); } #[test] fn on_transition() { struct TransitionLog { from: Option, to: Option, } impl Observer<()> for &mut TransitionLog { type Error = (); fn on_transition( &mut self, _ctx: &mut (), _id: &str, from: State, to: State, _data: Option, _state_data: Option, ) -> Result<(), Self::Error> { self.from = Some(from); self.to = Some(to); Ok(()) } } let mut transition_log = TransitionLog { from: None, to: None, }; let job = Job::init( &mut (), &mut transition_log, Some("foo".to_string()), JobData {}, QueuedData {}, ) .unwrap(); let _job = job.start(&mut (), ProcessingData {}).unwrap(); match transition_log.from { Some(state) => assert_eq!("Queued", state.to_string()), None => panic!("expected some from value"), }; match transition_log.to { Some(state) => assert_eq!("Processing", state.to_string()), None => panic!("expected some to value"), }; }