mirror of
https://github.com/rwf2/Rocket.git
synced 2025-01-05 17:22:36 +00:00
5d9035ddc1
In brief, this commit: * Updates to the latest upstream 'cookie', fixing a memory leak. * Make changes to 'CookieJar' observable only through 'pending()'. * Deprecates 'Client::new()' in favor of 'Client::tracked()'. * Makes 'dispatch()' on tracked 'Client's synchronize on cookies. * Makes 'Client::untracked()' actually untracked. This commit updates to the latest 'cookie' which removes support for 'Sync' cookie jars. Instead of relying on 'cookie', this commit implements an op-log based 'CookieJar' which internally keeps track of changes. The API is such that changes are only observable through specialized '_pending()' methods.
50 lines
1.5 KiB
Rust
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::tracked(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::tracked(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!");
|
|
}
|
|
}
|