Rewrite test_check_default_headers_loaded_in_every_request using fetch.

This commit is contained in:
Ms2ger 2016-11-07 17:06:45 +01:00
parent d777016b3a
commit b403e0004b

View file

@ -7,6 +7,7 @@ use cookie_rs::Cookie as CookiePair;
use devtools_traits::{ChromeToDevtoolsControlMsg, DevtoolsControlMsg, NetworkEvent}; use devtools_traits::{ChromeToDevtoolsControlMsg, DevtoolsControlMsg, NetworkEvent};
use devtools_traits::HttpRequest as DevtoolsHttpRequest; use devtools_traits::HttpRequest as DevtoolsHttpRequest;
use devtools_traits::HttpResponse as DevtoolsHttpResponse; use devtools_traits::HttpResponse as DevtoolsHttpResponse;
use fetch_sync;
use flate2::Compression; use flate2::Compression;
use flate2::write::{DeflateEncoder, GzEncoder}; use flate2::write::{DeflateEncoder, GzEncoder};
use hyper::LanguageTag; use hyper::LanguageTag;
@ -17,7 +18,9 @@ use hyper::header::{StrictTransportSecurity, UserAgent};
use hyper::http::RawStatus; use hyper::http::RawStatus;
use hyper::method::Method; use hyper::method::Method;
use hyper::mime::{Mime, SubLevel, TopLevel}; use hyper::mime::{Mime, SubLevel, TopLevel};
use hyper::server::{Request as HyperRequest, Response as HyperResponse};
use hyper::status::StatusCode; use hyper::status::StatusCode;
use make_server;
use msg::constellation_msg::{PipelineId, TEST_PIPELINE_ID}; use msg::constellation_msg::{PipelineId, TEST_PIPELINE_ID};
use net::cookie::Cookie; use net::cookie::Cookie;
use net::cookie_storage::CookieStorage; use net::cookie_storage::CookieStorage;
@ -27,9 +30,10 @@ use net::test::{HttpRequest, HttpRequestFactory, HttpState, LoadError, UIProvide
use net::test::{HttpResponse, LoadErrorType}; use net::test::{HttpResponse, LoadErrorType};
use net_traits::{CookieSource, IncludeSubdomains, LoadContext, LoadData}; use net_traits::{CookieSource, IncludeSubdomains, LoadContext, LoadData};
use net_traits::{CustomResponse, LoadOrigin, Metadata, ReferrerPolicy}; use net_traits::{CustomResponse, LoadOrigin, Metadata, ReferrerPolicy};
use net_traits::request::{Request, RequestInit, Destination};
use std::borrow::Cow; use std::borrow::Cow;
use std::io::{self, Cursor, Read, Write}; use std::io::{self, Cursor, Read, Write};
use std::sync::{Arc, RwLock, mpsc}; use std::sync::{Arc, Mutex, RwLock, mpsc};
use std::sync::mpsc::Receiver; use std::sync::mpsc::Receiver;
use std::thread; use std::thread;
use url::Url; use url::Url;
@ -229,20 +233,6 @@ impl HttpRequest for MockRequest {
} }
} }
struct AssertMustHaveHeadersRequestFactory {
expected_headers: Headers,
body: Vec<u8>
}
impl HttpRequestFactory for AssertMustHaveHeadersRequestFactory {
type R = MockRequest;
fn create(&self, _: Url, _: Method, headers: Headers) -> Result<MockRequest, LoadError> {
assert_eq!(headers, self.expected_headers);
Ok(MockRequest::new(ResponseType::Text(self.body.clone())))
}
}
struct AssertAuthHeaderRequestFactory { struct AssertAuthHeaderRequestFactory {
expected_headers: Headers, expected_headers: Headers,
body: Vec<u8> body: Vec<u8>
@ -373,14 +363,12 @@ pub fn expect_devtools_http_response(devtools_port: &Receiver<DevtoolsControlMsg
#[test] #[test]
fn test_check_default_headers_loaded_in_every_request() { fn test_check_default_headers_loaded_in_every_request() {
let url = Url::parse("http://mozilla.com").unwrap(); let expected_headers = Arc::new(Mutex::new(None));
let expected_headers_clone = expected_headers.clone();
let http_state = HttpState::new(); let handler = move |request: HyperRequest, _: HyperResponse| {
let ui_provider = TestProvider::new(); assert_eq!(request.headers, expected_headers_clone.lock().unwrap().take().unwrap());
};
let mut load_data = LoadData::new(LoadContext::Browsing, url.clone(), &HttpTest); let (mut server, url) = make_server(handler);
load_data.data = None;
load_data.method = Method::Get;
let mut headers = Headers::new(); let mut headers = Headers::new();
@ -388,7 +376,12 @@ fn test_check_default_headers_loaded_in_every_request() {
qitem(Encoding::Deflate), qitem(Encoding::Deflate),
qitem(Encoding::EncodingExt("br".to_owned()))])); qitem(Encoding::EncodingExt("br".to_owned()))]));
headers.set(Host { hostname: "mozilla.com".to_owned() , port: None }); let hostname = match url.host_str() {
Some(hostname) => hostname.to_owned(),
_ => panic!()
};
headers.set(Host { hostname: hostname, port: url.port() });
let accept = Accept(vec![ let accept = Accept(vec![
qitem(Mime(TopLevel::Text, SubLevel::Html, vec![])), qitem(Mime(TopLevel::Text, SubLevel::Html, vec![])),
@ -408,25 +401,37 @@ fn test_check_default_headers_loaded_in_every_request() {
QualityItem::new(en, Quality(500)), QualityItem::new(en, Quality(500)),
])); ]));
headers.set(UserAgent(DEFAULT_USER_AGENT.to_owned())); headers.set(UserAgent(::DEFAULT_USER_AGENT.to_owned()));
*expected_headers.lock().unwrap() = Some(headers.clone());
// Testing for method.GET // Testing for method.GET
let _ = load(&load_data, &ui_provider, &http_state, None, let request = Request::from_init(RequestInit {
&AssertMustHaveHeadersRequestFactory { url: url.clone(),
expected_headers: headers.clone(), method: Method::Get,
body: <[_]>::to_vec(&[]) destination: Destination::Document,
}, DEFAULT_USER_AGENT.into(), &CancellationListener::new(None), None); origin: url.clone(),
pipeline_id: Some(TEST_PIPELINE_ID),
.. RequestInit::default()
});
let response = fetch_sync(request, None);
assert!(response.status.unwrap().is_success());
// Testing for method.POST // Testing for method.POST
load_data.method = Method::Post;
headers.set(ContentLength(0 as u64)); headers.set(ContentLength(0 as u64));
*expected_headers.lock().unwrap() = Some(headers.clone());
let request = Request::from_init(RequestInit {
url: url.clone(),
method: Method::Post,
destination: Destination::Document,
origin: url.clone(),
pipeline_id: Some(TEST_PIPELINE_ID),
.. RequestInit::default()
});
let response = fetch_sync(request, None);
assert!(response.status.unwrap().is_success());
let _ = load(&load_data, &ui_provider, &http_state, None, let _ = server.close();
&AssertMustHaveHeadersRequestFactory {
expected_headers: headers,
body: <[_]>::to_vec(&[])
}, DEFAULT_USER_AGENT.into(), &CancellationListener::new(None), None);
} }
#[test] #[test]