96 lines
2.5 KiB
Rust
96 lines
2.5 KiB
Rust
use axum::http::{self, Request, StatusCode};
|
|
use echo::{
|
|
app,
|
|
messages::{Message, MessageBody},
|
|
};
|
|
use http_body_util::BodyExt;
|
|
use serde_json::{json, Value};
|
|
use tower::{Service, ServiceExt};
|
|
|
|
#[tokio::test]
|
|
async fn test_rw() {
|
|
let mut app = app().into_service();
|
|
|
|
let message = "test".to_string();
|
|
|
|
let response = {
|
|
let request: Message = {
|
|
let src = "c1".to_string();
|
|
let dest = "n1".to_string();
|
|
let body = {
|
|
let msg_id = 1;
|
|
let message = message.clone();
|
|
|
|
MessageBody::Broadcast { msg_id, message }
|
|
};
|
|
Message { src, dest, body }
|
|
};
|
|
|
|
let request = Request::builder()
|
|
.method(http::Method::POST)
|
|
.uri("/")
|
|
.header(http::header::CONTENT_TYPE, mime::APPLICATION_JSON.as_ref())
|
|
.body(serde_json::to_string(&request).unwrap())
|
|
.unwrap();
|
|
|
|
let response = ServiceExt::<Request<String>>::ready(&mut app)
|
|
.await
|
|
.unwrap()
|
|
.call(request)
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(response.status(), StatusCode::OK);
|
|
|
|
let request: Message = {
|
|
let src = "c1".to_string();
|
|
let dest = "n1".to_string();
|
|
let body = {
|
|
let msg_id = 2;
|
|
|
|
MessageBody::Read { msg_id }
|
|
};
|
|
Message { src, dest, body }
|
|
};
|
|
|
|
let request = Request::builder()
|
|
.method(http::Method::POST)
|
|
.uri("/")
|
|
.header(http::header::CONTENT_TYPE, mime::APPLICATION_JSON.as_ref())
|
|
.body(serde_json::to_string(&request).unwrap())
|
|
.unwrap();
|
|
|
|
let response = ServiceExt::<Request<String>>::ready(&mut app)
|
|
.await
|
|
.unwrap()
|
|
.call(request)
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(response.status(), StatusCode::OK);
|
|
response
|
|
};
|
|
|
|
{
|
|
let body: Value = {
|
|
let body = response.into_body().collect().await.unwrap().to_bytes();
|
|
serde_json::from_slice(&body).unwrap()
|
|
};
|
|
|
|
let expected = json!(
|
|
{
|
|
"src": "n1",
|
|
"dest": "c1",
|
|
"body": {
|
|
"type": "read_ok",
|
|
"msg_id": 2,
|
|
"in_reply_to": 2,
|
|
"message": vec![message]
|
|
}
|
|
}
|
|
);
|
|
|
|
assert_eq!(body, expected);
|
|
}
|
|
}
|