Rocket/core/lib/tests/many-cookie-jars-at-once.rs
Sergio Benitez 52320020bc Use thread-safe 'CookieJar's.
The user-facing changes effected by this commit are:

  * The 'http::Cookies<'_>' guard is now '&http::CookieJar<'_>'.
  * The "one-at-a-time" jar restriction is no longer imposed.
  * 'CookieJar' retrieval methods return 'http::CookieCrumb'.
  * The 'private-cookies' feature is now called 'secrets'.
  * Docs flag private cookie methods with feature cfg.
  * Local, async request dispatching is never serialized.
  * 'Client::cookies()' returns the tracked 'CookieJar'.
  * 'LocalResponse::cookies()' returns a 'CookieJar'.
  * 'Response::cookies()' returns an 'impl Iterator'.
  * A path of '/' is set by default on all cookies.
  * 'SameSite=strict' is set by default on all cookies.
  * 'LocalRequest::cookies()' accepts any 'Cookie' iterator.
  * The 'Debug' impl for 'Request' prints the cookie jar.

Resolves #1332.
2020-08-16 02:19:45 -07:00

50 lines
1.5 KiB
Rust

#[macro_use] extern crate rocket;
use rocket::http::{Cookie, CookieJar};
#[post("/")]
fn multi_add(jar_a: &CookieJar<'_>, jar_b: &CookieJar<'_>) {
jar_a.add(Cookie::new("a", "v1"));
jar_b.add(Cookie::new("b", "v2"));
}
#[get("/")]
fn multi_get(jar_a: &CookieJar<'_>, jar_b: &CookieJar<'_>, jar_c: &CookieJar<'_>) -> String {
let (a, a2, a3) = (jar_a.get("a"), jar_b.get("a"), jar_c.get("a"));
let (b, b2, b3) = (jar_a.get("b"), jar_b.get("b"), jar_c.get("b"));
assert_eq!(a, a2); assert_eq!(a2, a3);
assert_eq!(b, b2); assert_eq!(b2, b3);
format!("{}{}", a.unwrap().value(), b.unwrap().value())
}
#[cfg(test)]
mod many_cookie_jars_tests {
use super::*;
use rocket::local::blocking::Client;
fn rocket() -> rocket::Rocket {
rocket::ignite().mount("/", routes![multi_add, multi_get])
}
#[test]
fn test_mutli_add() {
let client = Client::new(rocket()).unwrap();
let response = client.post("/").dispatch();
let cookies = response.cookies();
assert_eq!(cookies.iter().count(), 2);
assert_eq!(cookies.get("a").unwrap().value(), "v1");
assert_eq!(cookies.get("b").unwrap().value(), "v2");
}
#[test]
fn test_mutli_get() {
let client = Client::new(rocket()).unwrap();
let response = client.get("/")
.cookie(Cookie::new("a", "a_val"))
.cookie(Cookie::new("b", "hi!"))
.dispatch();
assert_eq!(response.into_string().unwrap(), "a_valhi!");
}
}