From 4ef6561cc65e88c21b45590fa82511c4a1adaa3a Mon Sep 17 00:00:00 2001 From: Kitaiti Makoto Date: Sun, 24 Jan 2021 21:38:51 +0900 Subject: [PATCH] Extract login() from test function --- src/routes/blogs.rs | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/src/routes/blogs.rs b/src/routes/blogs.rs index 9b1e9e6f..287dbdb4 100644 --- a/src/routes/blogs.rs +++ b/src/routes/blogs.rs @@ -389,7 +389,7 @@ mod tests { }; use rocket::{ http::{Cookie, Cookies, SameSite}, - local::Client, + local::{Client, LocalRequest}, }; #[test] @@ -409,11 +409,7 @@ mod tests { assert!(!body.contains(&edit_link)); let request = client.get(&blog_path); - request.inner().guard::().unwrap().add_private( - Cookie::build(AUTH_COOKIE, user.id.to_string()) - .same_site(SameSite::Lax) - .finish(), - ); + login(&request, &user); let mut response = request.dispatch(); let body = response.body_string().unwrap(); assert!(body.contains(&edit_link)); @@ -499,4 +495,12 @@ mod tests { }) .unwrap() } + + fn login(request: &LocalRequest, user: &User) { + request.inner().guard::().unwrap().add_private( + Cookie::build(AUTH_COOKIE, user.id.to_string()) + .same_site(SameSite::Lax) + .finish(), + ); + } }