#![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 = "receipts")]
struct Receipt {
order_id: u64,
}
#[subscriber("orders", reply)]
async fn issue_receipt(order: &Order) -> Result<Receipt, HandlerOutcome> {
if order.id == 0 {
return Err(HandlerOutcome::drop());
}
Ok(Receipt { order_id: order.id })
}
#[derive(Debug, PartialEq, Deserialize, Serialize, Outgoing)]
struct Quote {
order_id: u64,
price: u64,
}
#[subscriber("quote-requests", reply("quotes"))]
async fn quote(order: &Order) -> Quote {
Quote {
order_id: order.id,
price: order.id * 10,
}
}
fn app() -> impl App {
RustStream::new(AppInfo::new("billing", "0.1.0")).with_broker(MemoryBroker::new(), |b| {
b.include(issue_receipt);
b.include(quote).out_reply(Publish);
})
}
fn main() {}
#[cfg(test)]
mod tests {
use std::error::Error;
use ruststream::memory::prelude::*;
use ruststream::testing::TestApp;
use super::{Order, Quote, Receipt, app};
#[tokio::test]
async fn an_order_gets_its_receipt() -> Result<(), Box<dyn Error>> {
let tb = TestApp::start(app()).await?;
let broker = tb.broker::<MemoryBroker>();
broker
.message(&Order { id: 7 })
.to("orders")
.publish()
.await?;
broker
.published::<Receipt>("receipts")
.assert_called_once()
.with(&Receipt { order_id: 7 });
broker
.subscriber("orders")
.assert_called_once()
.settled(HandlerOutcome::ack());
Ok(())
}
#[tokio::test]
async fn an_order_without_an_id_gets_no_receipt() -> Result<(), Box<dyn Error>> {
let tb = TestApp::start(app()).await?;
let broker = tb.broker::<MemoryBroker>();
broker
.message(&Order { id: 0 })
.to("orders")
.publish()
.await?;
broker.published::<Receipt>("receipts").assert_not_called();
broker
.subscriber("orders")
.assert_called_once()
.settled(HandlerOutcome::drop());
Ok(())
}
#[tokio::test]
async fn a_quote_goes_where_the_mount_site_says() -> Result<(), Box<dyn Error>> {
let tb = TestApp::start(app()).await?;
let broker = tb.broker::<MemoryBroker>();
broker
.message(&Order { id: 4 })
.to("quote-requests")
.publish()
.await?;
broker
.published::<Quote>("quotes")
.assert_called_once()
.with(&Quote {
order_id: 4,
price: 40,
});
Ok(())
}
}