mirror of https://github.com/rwf2/Rocket.git
45 lines
1.4 KiB
Rust
45 lines
1.4 KiB
Rust
|
use rocket;
|
||
|
|
use rocket::testing::MockRequest;
|
||
|
|
use rocket::http::Method::*;
|
||
|
|
use rocket::http::{Status, ContentType};
|
||
|
|
use rocket::Response;
|
||
|
|
|
||
|
|
#[derive(Serialize, Deserialize)]
|
||
|
|
struct Message {
|
||
|
|
id: usize,
|
||
|
|
contents: String
|
||
|
|
}
|
||
|
|
|
||
|
|
macro_rules! run_test {
|
||
|
|
($rocket: expr, $req:expr, $test_fn:expr) => ({
|
||
|
|
let mut req = $req;
|
||
|
|
$test_fn(req.dispatch_with($rocket));
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn msgpack_get() {
|
||
|
|
let rocket = rocket();
|
||
|
|
let req = MockRequest::new(Get, "/message/1").header(ContentType::MsgPack);
|
||
|
|
run_test!(&rocket, req, |mut response: Response| {
|
||
|
|
assert_eq!(response.status(), Status::Ok);
|
||
|
|
let body = response.body().unwrap().into_bytes().unwrap();
|
||
|
|
// Represents a message of `[1, "Hello, world!"]`
|
||
|
|
assert_eq!(&body, &[146, 1, 173, 72, 101, 108, 108, 111, 44, 32, 119, 111,
|
||
|
|
114, 108, 100, 33]);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn msgpack_post() {
|
||
|
|
let rocket = rocket();
|
||
|
|
let req = MockRequest::new(Post, "/message")
|
||
|
|
.header(ContentType::MsgPack)
|
||
|
|
// Represents a message of `[2, "Goodbye, world!"]`
|
||
|
|
.body(&[146, 2, 175, 71, 111, 111, 100, 98, 121, 101, 44, 32, 119, 111, 114, 108, 100, 33]);
|
||
|
|
run_test!(&rocket, req, |mut response: Response| {
|
||
|
|
assert_eq!(response.status(), Status::Ok);
|
||
|
|
assert_eq!(response.body().unwrap().into_string().unwrap(), "Goodbye, world!");
|
||
|
|
});
|
||
|
|
}
|