http-cache: re-enable and update test

This commit is contained in:
Gregory Terzian 2019-06-22 06:28:32 -07:00
parent 67494d4776
commit 689b7971b8
2 changed files with 27 additions and 25 deletions

View file

@ -2,48 +2,49 @@
* License, v. 2.0. If a copy of the MPL was not distributed with this * License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at https://mozilla.org/MPL/2.0/. */ * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
use hyper::header::{Expires, HttpDate}; use crossbeam_channel::unbounded;
use hyper::method::Method; use http::header::{HeaderValue, EXPIRES};
use hyper::status::StatusCode; use http::StatusCode;
use msg::constellation_msg::TEST_PIPELINE_ID; use msg::constellation_msg::TEST_PIPELINE_ID;
use net::http_cache::HttpCache; use net::http_cache::HttpCache;
use net_traits::request::{Destination, Request, RequestInit}; use net_traits::request::{Origin, Request};
use net_traits::response::{Response, ResponseBody}; use net_traits::response::{Response, ResponseBody};
use net_traits::{ResourceFetchTiming, ResourceTimingType};
use servo_url::ServoUrl; use servo_url::ServoUrl;
use std::sync::mpsc::channel;
#[test] #[test]
fn test_refreshing_resource_sets_done_chan_the_appropriate_value() { fn test_refreshing_resource_sets_done_chan_the_appropriate_value() {
let response_bodies = vec![ResponseBody::Receiving(vec![]), let response_bodies = vec![
ResponseBody::Receiving(vec![]),
ResponseBody::Empty, ResponseBody::Empty,
ResponseBody::Done(vec![])]; ResponseBody::Done(vec![]),
];
let url = ServoUrl::parse("https://servo.org").unwrap(); let url = ServoUrl::parse("https://servo.org").unwrap();
let request = Request::from_init(RequestInit { let request = Request::new(
url: url.clone(), url.clone(),
method: Method::Get, Some(Origin::Origin(url.clone().origin())),
destination: Destination::Document, Some(TEST_PIPELINE_ID),
origin: url.clone().origin(), );
pipeline_id: Some(TEST_PIPELINE_ID), let timing = ResourceFetchTiming::new(ResourceTimingType::Navigation);
.. RequestInit::default() let mut response = Response::new(url.clone(), timing);
});
let mut response = Response::new(url.clone());
// Expires header makes the response cacheable. // Expires header makes the response cacheable.
response.headers.set(Expires(HttpDate(time::now()))); response
response_bodies.iter().for_each(|body| { .headers
.insert(EXPIRES, HeaderValue::from_str("-10").unwrap());
let mut cache = HttpCache::new(); let mut cache = HttpCache::new();
*response.body.lock().unwrap() = body; response_bodies.iter().for_each(|body| {
*response.body.lock().unwrap() = body.clone();
// First, store the 'normal' response. // First, store the 'normal' response.
cache.store(&request, &response); cache.store(&request, &response);
// Second, mutate the response into a 304 response, and refresh the stored one. // Second, mutate the response into a 304 response, and refresh the stored one.
response.status = Some(StatusCode::NotModified); response.status = Some((StatusCode::NOT_MODIFIED, String::from("304")));
let mut done_chan = Some(channel()); let mut done_chan = Some(unbounded());
let refreshed_response = cache.refresh(&request, response, &mut done_chan); let refreshed_response = cache.refresh(&request, response.clone(), &mut done_chan);
// Ensure a resource was found, and refreshed. // Ensure a resource was found, and refreshed.
assert!(refreshed_response.is_some()); assert!(refreshed_response.is_some());
match body { match body {
ResponseBody::Receiving(_) => assert!(done_chan.is_some()), ResponseBody::Receiving(_) => assert!(done_chan.is_some()),
ResponseBody::Empty | ResponseBody::Done(_) => assert!(done_chan.is_none()) ResponseBody::Empty | ResponseBody::Done(_) => assert!(done_chan.is_none()),
} }
}) })
} }

View file

@ -14,6 +14,7 @@ mod fetch;
mod file_loader; mod file_loader;
mod filemanager_thread; mod filemanager_thread;
mod hsts; mod hsts;
mod http_cache;
mod http_loader; mod http_loader;
mod mime_classifier; mod mime_classifier;
mod resource_thread; mod resource_thread;