ruwuma/crates/ruma-api/tests/conversions.rs
2021-05-06 11:07:31 +00:00

148 lines
3.9 KiB
Rust

use ruma_api::{
ruma_api, IncomingRequest as _, OutgoingRequest as _, OutgoingRequestAppserviceExt,
SendAccessToken,
};
use ruma_identifiers::{user_id, UserId};
ruma_api! {
metadata: {
description: "Does something.",
method: POST,
name: "my_endpoint",
path: "/_matrix/foo/:bar/:user",
rate_limited: false,
authentication: None,
}
request: {
pub hello: String,
#[ruma_api(header = CONTENT_TYPE)]
pub world: String,
#[ruma_api(query)]
pub q1: String,
#[ruma_api(query)]
pub q2: u32,
#[ruma_api(path)]
pub bar: String,
#[ruma_api(path)]
pub user: UserId,
}
response: {
pub hello: String,
#[ruma_api(header = CONTENT_TYPE)]
pub world: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub optional_flag: Option<bool>,
}
}
#[test]
fn request_serde() {
let req = Request {
hello: "hi".to_owned(),
world: "test".to_owned(),
q1: "query_param_special_chars %/&@!".to_owned(),
q2: 55,
bar: "barVal".to_owned(),
user: user_id!("@bazme:ruma.io"),
};
let http_req = req
.clone()
.try_into_http_request::<Vec<u8>>("https://homeserver.tld", SendAccessToken::None)
.unwrap();
let req2 = Request::try_from_http_request(http_req).unwrap();
assert_eq!(req.hello, req2.hello);
assert_eq!(req.world, req2.world);
assert_eq!(req.q1, req2.q1);
assert_eq!(req.q2, req2.q2);
assert_eq!(req.bar, req2.bar);
assert_eq!(req.user, req2.user);
}
#[test]
fn request_with_user_id_serde() {
let req = Request {
hello: "hi".to_owned(),
world: "test".to_owned(),
q1: "query_param_special_chars %/&@!".to_owned(),
q2: 55,
bar: "barVal".to_owned(),
user: user_id!("@bazme:ruma.io"),
};
let user_id = user_id!("@_virtual_:ruma.io");
let http_req = req
.try_into_http_request_with_user_id::<Vec<u8>>(
"https://homeserver.tld",
SendAccessToken::None,
user_id,
)
.unwrap();
let query = http_req.uri().query().unwrap();
assert_eq!(
query,
"q1=query_param_special_chars+%25%2F%26%40%21&q2=55&user_id=%40_virtual_%3Aruma.io"
);
}
mod without_query {
use super::{ruma_api, user_id, OutgoingRequestAppserviceExt, SendAccessToken, UserId};
ruma_api! {
metadata: {
description: "Does something without query.",
method: POST,
name: "my_endpoint",
path: "/_matrix/foo/:bar/:user",
rate_limited: false,
authentication: None,
}
request: {
pub hello: String,
#[ruma_api(header = CONTENT_TYPE)]
pub world: String,
#[ruma_api(path)]
pub bar: String,
#[ruma_api(path)]
pub user: UserId,
}
response: {
pub hello: String,
#[ruma_api(header = CONTENT_TYPE)]
pub world: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub optional_flag: Option<bool>,
}
}
#[test]
fn request_without_query_with_user_id_serde() {
let req = Request {
hello: "hi".to_owned(),
world: "test".to_owned(),
bar: "barVal".to_owned(),
user: user_id!("@bazme:ruma.io"),
};
let user_id = user_id!("@_virtual_:ruma.io");
let http_req = req
.try_into_http_request_with_user_id::<Vec<u8>>(
"https://homeserver.tld",
SendAccessToken::None,
user_id,
)
.unwrap();
let query = http_req.uri().query().unwrap();
assert_eq!(query, "user_id=%40_virtual_%3Aruma.io");
}
}