mirror of
https://gitlab.com/msrd0/gotham-restful.git
synced 2025-02-23 04:52:28 +00:00
49 lines
949 B
Rust
49 lines
949 B
Rust
mod custom_request_body
|
|
{
|
|
|
|
|
|
use gotham::{
|
|
hyper::header::CONTENT_TYPE,
|
|
router::builder::*,
|
|
test::TestServer
|
|
};
|
|
use gotham_restful::*;
|
|
use mime::TEXT_PLAIN;
|
|
|
|
const RESPONSE : &[u8] = b"This is the only valid response.";
|
|
|
|
#[derive(Resource)]
|
|
#[resource(create)]
|
|
struct FooResource;
|
|
|
|
#[derive(FromBody, RequestBody)]
|
|
#[supported_types(TEXT_PLAIN)]
|
|
struct Foo {
|
|
content: Vec<u8>,
|
|
content_type: Mime
|
|
}
|
|
|
|
#[create(FooResource)]
|
|
fn create(body : Foo) -> Raw<Vec<u8>> {
|
|
Raw::new(body.content, body.content_type)
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn test()
|
|
{
|
|
let server = TestServer::new(build_simple_router(|router| {
|
|
router.resource::<FooResource>("foo");
|
|
})).unwrap();
|
|
|
|
let res = server.client()
|
|
.post("http://localhost/foo", RESPONSE, TEXT_PLAIN)
|
|
.perform().unwrap();
|
|
assert_eq!(res.headers().get(CONTENT_TYPE).unwrap().to_str().unwrap(), "text/plain");
|
|
let res = res.read_body().unwrap();
|
|
let body : &[u8] = res.as_ref();
|
|
assert_eq!(body, RESPONSE);
|
|
}
|
|
|
|
|
|
}
|