43 lines
1.0 KiB
Rust
43 lines
1.0 KiB
Rust
use std::convert::TryFrom;
|
|
|
|
use http::header::{Entry, CONTENT_TYPE};
|
|
use ruma_api::ruma_api;
|
|
|
|
ruma_api! {
|
|
metadata: {
|
|
description: "Does something.",
|
|
method: GET,
|
|
name: "no_fields",
|
|
path: "/_matrix/my/endpoint",
|
|
rate_limited: false,
|
|
authentication: None,
|
|
}
|
|
|
|
request: {
|
|
#[ruma_api(header = LOCATION)]
|
|
pub location: Option<String>,
|
|
}
|
|
|
|
response: {
|
|
#[ruma_api(header = CONTENT_TYPE)]
|
|
pub stuff: String,
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn content_type_override() {
|
|
let res = Response { stuff: "magic".into() };
|
|
let mut http_res = http::Response::<Vec<u8>>::try_from(res).unwrap();
|
|
|
|
// Test that we correctly replaced the default content type,
|
|
// not adding another content-type header.
|
|
assert_eq!(
|
|
match http_res.headers_mut().entry(CONTENT_TYPE) {
|
|
Entry::Occupied(occ) => occ.iter().count(),
|
|
_ => 0,
|
|
},
|
|
1
|
|
);
|
|
assert_eq!(http_res.headers().get("content-type").unwrap(), "magic");
|
|
}
|