From 2d2917b61fdf09e4009cd022f63f1d5608d588cc Mon Sep 17 00:00:00 2001 From: winkidney Date: Wed, 2 Jan 2019 11:41:50 +0800 Subject: [PATCH] Fix: Should assign a new cookie vector if it doesn't exist --- src/test.rs | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/src/test.rs b/src/test.rs index 7f5762258..66982f9ff 100644 --- a/src/test.rs +++ b/src/test.rs @@ -549,19 +549,19 @@ impl TestRequest { /// set cookie of this request pub fn cookie(mut self, cookie: Cookie<'static>) -> Self { - let mut should_insert = true; - match &mut self.cookies { - Some(old_cookies) => { - for old_cookie in old_cookies.iter() { - if old_cookie == &cookie { - should_insert = false - }; + if self.cookies.is_none() { + let mut should_insert = true; + let old_cookies = self.cookies.as_mut().unwrap(); + for old_cookie in old_cookies.iter() { + if old_cookie == &cookie { + should_insert = false }; - if should_insert { - old_cookies.push(cookie); - }; - } - None => {} + }; + if should_insert { + old_cookies.push(cookie); + }; + } else { + self.cookies = Some(vec![cookie]); }; self }