Rocket/core/lib/tests/head_handling.rs
Sergio Benitez 63e6845386 Revamp 'Response', 'Body' types.
This is a breaking change for many consumers of the 'Response' and all
consumers of the 'Body' API. The summary of breaking changes is:

  * 'Response::body()', 'Response::body_mut()' are infallible.
  * A 'Body' can represent an empty body in more cases.
  * 'ResponseBuilder' is now simply 'Builder'.
  * Direct body read methods on 'Response' were removed in favor of
    chaining through 'body_mut()': 'r.body_mut().to_string()'.
  * Notion of a 'chunked_body' was removed as it was inaccurate.
  * Maximum chunk size can be set on any body.
  * 'Response' no longer implements 'Responder'.

A few bugs were fixed in the process. Specifically, 'Body' will emit an
accurate size even for bodies that are partially read, and the size of
seek-determined bodies is emitted on HEAD request where it wasn't
before. Specifics on transport were clarified, and 'Body' docs greatly
improved as a result.
2021-04-28 02:21:33 -07:00

59 lines
1.6 KiB
Rust

#[macro_use] extern crate rocket;
use rocket::{http::Status, response::content};
#[get("/empty")]
fn empty() -> Status {
Status::NoContent
}
#[get("/")]
fn index() -> &'static str {
"Hello, world!"
}
#[head("/other")]
fn other() -> content::Json<&'static str> {
content::Json("{ 'hi': 'hello' }")
}
mod head_handling_tests {
use super::*;
use rocket::Route;
use rocket::local::blocking::Client;
use rocket::http::{Status, ContentType};
fn routes() -> Vec<Route> {
routes![index, empty, other]
}
#[test]
fn auto_head() {
let client = Client::debug_with(routes()).unwrap();
let response = client.head("/").dispatch();
let content_type: Vec<_> = response.headers().get("Content-Type").collect();
assert_eq!(content_type, vec![ContentType::Plain.to_string()]);
assert_eq!(response.status(), Status::Ok);
assert_eq!(response.body().preset_size(), Some(13));
assert!(response.into_bytes().unwrap().is_empty());
let response = client.head("/empty").dispatch();
assert_eq!(response.status(), Status::NoContent);
assert!(response.into_bytes().is_none());
}
#[test]
fn user_head() {
let client = Client::debug_with(routes()).unwrap();
let response = client.head("/other").dispatch();
let content_type: Vec<_> = response.headers().get("Content-Type").collect();
assert_eq!(content_type, vec![ContentType::JSON.to_string()]);
assert_eq!(response.status(), Status::Ok);
assert_eq!(response.body().preset_size(), Some(17));
assert!(response.into_bytes().unwrap().is_empty());
}
}