Mirrors/servo
1
0
Fork 0
mirror of https://github.com/servo/servo.git synced 2025-08-04 21:20:23 +01:00
Code Issues Projects Releases Packages Wiki Activity

Rewrite test_load_should_decode_the_response_as_deflate_when_response_headers_have_content_encoding_deflate.

Browse source
This commit is contained in:
Ms2ger 2016-11-08 17:25:34 +01:00
parent 90dd7c922a
commit 4fe105b8e1
1 changed files with 23 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -31,6 +31,7 @@ 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 net_traits::request::{Request, RequestInit, Destination};
use net_traits::response::ResponseBody;
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, Mutex, RwLock, mpsc}; use std::sync::{Arc, Mutex, RwLock, mpsc};
@ -656,37 +657,31 @@ fn test_load_when_redirecting_from_a_post_should_rewrite_next_request_as_get() {
#[test] #[test]
fn test_load_should_decode_the_response_as_deflate_when_response_headers_have_content_encoding_deflate() { fn test_load_should_decode_the_response_as_deflate_when_response_headers_have_content_encoding_deflate() {
struct Factory; let handler = move |_: HyperRequest, mut response: HyperResponse| {
response.headers_mut().set(ContentEncoding(vec![Encoding::Deflate]));
impl HttpRequestFactory for Factory {
type R = MockRequest;
fn create(&self, _: Url, _: Method, _: Headers) -> Result<MockRequest, LoadError> {
let mut e = DeflateEncoder::new(Vec::new(), Compression::Default); let mut e = DeflateEncoder::new(Vec::new(), Compression::Default);
e.write(b"Yay!").unwrap(); e.write(b"Yay!").unwrap();
let encoded_content = e.finish().unwrap(); let encoded_content = e.finish().unwrap();
response.send(&encoded_content).unwrap();
};
let (mut server, url) = make_server(handler);
let mut headers = Headers::new(); let request = Request::from_init(RequestInit {
headers.set(ContentEncoding(vec![Encoding::Deflate])); url: url.clone(),
Ok(MockRequest::new(ResponseType::WithHeaders(encoded_content, headers))) method: Method::Get,
} body: None,
} destination: Destination::Document,
origin: url.clone(),
pipeline_id: Some(TEST_PIPELINE_ID),
.. RequestInit::default()
});
let response = fetch_sync(request, None);
let url = Url::parse("http://mozilla.com").unwrap(); let _ = server.close();
let load_data = LoadData::new(LoadContext::Browsing, url.clone(), &HttpTest);
let http_state = HttpState::new(); assert!(response.status.unwrap().is_success());
let ui_provider = TestProvider::new(); assert_eq!(*response.body.lock().unwrap(),
ResponseBody::Done(b"Yay!".to_vec()));
let mut response = load(
&load_data, &ui_provider, &http_state, None,
&Factory,
DEFAULT_USER_AGENT.into(),
&CancellationListener::new(None),
None)
.unwrap();
assert_eq!(read_response(&mut response), "Yay!");
} }
#[test] #[test]