#![cfg_attr(not(test), allow(dead_code, unused_imports))]
use ruststream::memory::prelude::*;
use serde::{Deserialize, Serialize};
#[derive(Debug, PartialEq, Deserialize, Serialize, Outgoing)]
struct Order {
id: u64,
}
#[derive(Debug, PartialEq, Deserialize, Serialize, Outgoing)]
#[outgoing(name = "services.started")]
struct Started {
service: String,
}
#[subscriber("orders")]
async fn archive(order: &Order, Out(out): Out<impl Publisher>) -> HandlerOutcome {
if out
.message(order)
.to("orders.archive")
.publish()
.await
.is_err()
{
return HandlerOutcome::retry();
}
HandlerOutcome::ack()
}
fn app() -> impl App {
RustStream::new(AppInfo::new("archive", "0.1.0")).with_broker(MemoryBroker::new(), |b| {
b.include(archive).out(DefaultSlot, Publish).build();
b.after_startup(Publish, async move |publisher| {
publisher
.message(&Started {
service: "archive".into(),
})
.publish()
.await
});
})
}
fn main() {}
#[cfg(test)]
mod tests {
use std::error::Error;
use ruststream::memory::prelude::*;
use ruststream::testing::TestApp;
use super::{Order, Started, app};
#[tokio::test]
async fn the_service_announces_itself_once_started() -> Result<(), Box<dyn Error>> {
let tb = TestApp::start(app()).await?;
tb.broker::<MemoryBroker>()
.published::<Started>("services.started")
.assert_called_once()
.with(&Started {
service: "archive".into(),
});
Ok(())
}
#[tokio::test]
async fn an_order_is_archived() -> Result<(), Box<dyn Error>> {
let tb = TestApp::start(app()).await?;
let broker = tb.broker::<MemoryBroker>();
broker
.message(&Order { id: 6 })
.to("orders")
.publish()
.await?;
broker
.published::<Order>("orders.archive")
.assert_called_once()
.with(&Order { id: 6 });
Ok(())
}
}