Mirrors/servo
1
0
Fork 0
mirror of https://github.com/servo/servo.git synced 2025-06-25 01:24:37 +01:00
Code Issues Projects Releases Packages Wiki Activity

Rewrite test_cookies_blocked.

Browse source
This commit is contained in:
Ms2ger 2016-11-21 11:40:41 +01:00
parent 2fa032109f
commit ce89fcf335
1 changed files with 27 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

48
tests/unit/net/http_loader.rs
View file

@ -1524,36 +1524,42 @@ fn test_content_blocked() {
#[test] #[test]
fn test_cookies_blocked() { fn test_cookies_blocked() {
let url_without_cookies = ServoUrl::parse("http://mozilla2.com").unwrap(); let handler = move |request: HyperRequest, response: HyperResponse| {
let mut http_state = HttpState::new(); assert_eq!(request.headers.get::<CookieHeader>(), None);
response.send(b"hi").unwrap();
};
let (mut server, url) = make_server(handler);
let blocked_content_list = "[{ \"trigger\": { \"url-filter\": \"https?://mozilla2.com\" }, \ let url_filter = url.as_str().replace("http://", "https?://");
\"action\": { \"type\": \"block-cookies\" } }]"; let blocked_content_list = format!("[{{ \
http_state.blocked_content = Arc::new(parse_list(blocked_content_list).ok()); \"trigger\": {{ \"url-filter\": \"{}\" }}, \
assert!(http_state.blocked_content.is_some()); \"action\": {{ \"type\": \"block-cookies\" }} \
}}]", url_filter);
let mut context = new_fetch_context(None);
context.state.blocked_content = Arc::new(Some(parse_list(&blocked_content_list).unwrap()));
{ {
let mut cookie_jar = http_state.cookie_jar.write().unwrap(); let mut cookie_jar = context.state.cookie_jar.write().unwrap();
let cookie = Cookie::new_wrapped( let cookie = Cookie::new_wrapped(
CookiePair::parse("mozillaIs=theBest;").unwrap(), CookiePair::new("mozillaIs".to_owned(), "theBest".to_owned()),
&url_without_cookies, &url,
CookieSource::HTTP CookieSource::HTTP
).unwrap(); ).unwrap();
cookie_jar.push(cookie, CookieSource::HTTP); cookie_jar.push(cookie, CookieSource::HTTP);
} }
let ui_provider = TestProvider::new(); let request = Request::from_init(RequestInit {
url: url.clone(),
method: Method::Get,
body: None,
destination: Destination::Document,
origin: url.clone(),
.. RequestInit::default()
});
let load_data = LoadData::new(LoadContext::Browsing, url_without_cookies, &HttpTest); let response = fetch(Rc::new(request), &mut None, &context);
let response = load( let _ = server.close();
&load_data, &ui_provider, &http_state,
None, &AssertMustNotIncludeHeadersRequestFactory { assert!(response.status.unwrap().is_success());
headers_not_expected: vec!["Cookie".to_owned()],
body: b"hi".to_vec(),
}, DEFAULT_USER_AGENT.into(), &CancellationListener::new(None), None);
match response {
Ok(_) => {},
_ => panic!("request should have succeeded without cookies"),
}
} }