diff options
Diffstat (limited to 'src/test/fixtures/message.rs')
| -rw-r--r-- | src/test/fixtures/message.rs | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/src/test/fixtures/message.rs b/src/test/fixtures/message.rs index 3aebdd9..8cb50c1 100644 --- a/src/test/fixtures/message.rs +++ b/src/test/fixtures/message.rs @@ -8,7 +8,7 @@ use crate::{ clock::RequestedAt, event::Event, login::Login, - message::{self, Body, Message}, + message::{self, event, Body, Message}, }; pub async fn send(app: &App, channel: &Channel, sender: &Login, sent_at: &RequestedAt) -> Message { @@ -31,6 +31,20 @@ pub fn events(event: Event) -> future::Ready<Option<message::Event>> { }) } +pub fn sent(event: message::Event) -> future::Ready<Option<event::Sent>> { + future::ready(match event { + message::Event::Sent(event) => Some(event), + message::Event::Deleted(_) => None, + }) +} + +pub fn deleted(event: message::Event) -> future::Ready<Option<event::Deleted>> { + future::ready(match event { + message::Event::Deleted(event) => Some(event), + message::Event::Sent(_) => None, + }) +} + pub fn fictitious() -> message::Id { message::Id::generate() } |
