xmr-btc-swap/xmr-btc/tests/harness/transport.rs

65 lines
1.6 KiB
Rust
Raw Normal View History

use anyhow::{anyhow, Result};
use async_trait::async_trait;
use tokio::{
stream::StreamExt,
sync::mpsc::{Receiver, Sender},
};
2020-10-08 02:00:31 -04:00
use xmr_btc::{
alice, bob,
transport::{Receive, Send},
};
#[derive(Debug)]
pub struct Transport<SendMsg, RecvMsg> {
pub sender: Sender<SendMsg>,
pub receiver: Receiver<RecvMsg>,
}
#[async_trait]
2020-10-08 02:00:31 -04:00
impl Send<alice::Message> for Transport<alice::Message, bob::Message> {
async fn send_message(&mut self, message: alice::Message) -> Result<()> {
let _ = self
.sender
.send(message)
.await
.map_err(|_| anyhow!("failed to send message"))?;
Ok(())
}
2020-10-08 02:00:31 -04:00
}
2020-10-08 02:00:31 -04:00
#[async_trait]
impl Receive<bob::Message> for Transport<alice::Message, bob::Message> {
async fn receive_message(&mut self) -> Result<bob::Message> {
let message = self
.receiver
.next()
.await
.ok_or_else(|| anyhow!("failed to receive message"))?;
Ok(message)
}
}
#[async_trait]
2020-10-08 02:00:31 -04:00
impl Send<bob::Message> for Transport<bob::Message, alice::Message> {
async fn send_message(&mut self, message: bob::Message) -> Result<()> {
let _ = self
.sender
.send(message)
.await
.map_err(|_| anyhow!("failed to send message"))?;
Ok(())
}
2020-10-08 02:00:31 -04:00
}
2020-10-08 02:00:31 -04:00
#[async_trait]
impl Receive<alice::Message> for Transport<bob::Message, alice::Message> {
async fn receive_message(&mut self) -> Result<alice::Message> {
let message = self
.receiver
.next()
.await
.ok_or_else(|| anyhow!("failed to receive message"))?;
Ok(message)
}
}