Rocket/examples/fairings/src/tests.rs

use super::rocket;
use rocket::local::Client;

#[test]
fn rewrite_get_put() {
    let client = Client::new(rocket()).unwrap();
    let mut response = client.get("/").dispatch();
    assert_eq!(response.body_string_wait(), Some("Hello, fairings!".into()));
}

#[test]
fn counts() {
    let client = Client::new(rocket()).unwrap();

    // Issue 1 GET request.
    client.get("/").dispatch();

    // Check the GET count, taking into account _this_ GET request.
    let mut response = client.get("/counts").dispatch();
    assert_eq!(response.body_string_wait(), Some("Get: 2\nPost: 0".into()));

    // Issue 1 more GET request and a POST.
    client.get("/").dispatch();
    client.post("/").dispatch();

    // Check the counts.
    let mut response = client.get("/counts").dispatch();
    assert_eq!(response.body_string_wait(), Some("Get: 4\nPost: 1".into()));
}

#[test]
fn token() {
    let client = Client::new(rocket()).unwrap();

    // Ensure the token is '123', which is what we have in `Rocket.toml`.
    let mut res = client.get("/token").dispatch();
    assert_eq!(res.body_string_wait(), Some("123".into()));
}