OLD | NEW |
| (Empty) |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #import "ios/web/public/test/response_providers/html_response_provider_impl.h" | |
6 | |
7 #import "ios/web/public/test/response_providers/response_provider.h" | |
8 #include "net/http/http_response_headers.h" | |
9 #include "net/http/http_status_code.h" | |
10 #include "url/gurl.h" | |
11 | |
12 namespace { | |
13 std::map<GURL, HtmlResponseProviderImpl::Response> BuildResponseMap( | |
14 const std::map<GURL, std::string>& responses_body, | |
15 scoped_refptr<net::HttpResponseHeaders> headers) { | |
16 std::map<GURL, HtmlResponseProviderImpl::Response> responses; | |
17 for (const std::pair<const GURL, std::string>& pair : responses_body) { | |
18 responses.insert(std::make_pair( | |
19 pair.first, HtmlResponseProviderImpl::Response(pair.second, headers))); | |
20 } | |
21 return responses; | |
22 } | |
23 | |
24 std::map<GURL, HtmlResponseProviderImpl::Response> BuildResponseMap( | |
25 const std::map<GURL, std::pair<std::string, std::string>>& responses, | |
26 const std::map<GURL, scoped_refptr<net::HttpResponseHeaders>>& headers) { | |
27 std::map<GURL, HtmlResponseProviderImpl::Response> response_map; | |
28 for (const auto& pair : responses) { | |
29 response_map.insert(std::make_pair( | |
30 pair.first, HtmlResponseProviderImpl::Response( | |
31 pair.second.second, headers.at(pair.first)))); | |
32 } | |
33 return response_map; | |
34 } | |
35 | |
36 } // namespace | |
37 | |
38 HtmlResponseProviderImpl::Response::Response( | |
39 const std::string& body, | |
40 const scoped_refptr<net::HttpResponseHeaders>& headers) | |
41 : body(body), headers(headers) {} | |
42 | |
43 HtmlResponseProviderImpl::Response::Response(const Response& response) | |
44 : body(response.body), headers(response.headers) {} | |
45 | |
46 HtmlResponseProviderImpl::Response::Response() : body(), headers() {} | |
47 | |
48 HtmlResponseProviderImpl::Response::~Response() {} | |
49 | |
50 HtmlResponseProviderImpl::HtmlResponseProviderImpl() : responses_() {} | |
51 | |
52 // static | |
53 HtmlResponseProviderImpl::Response | |
54 HtmlResponseProviderImpl::GetRedirectResponse( | |
55 const GURL& destination_url, | |
56 const net::HttpStatusCode& http_status) { | |
57 auto response_headers = web::ResponseProvider::GetRedirectResponseHeaders( | |
58 destination_url.spec(), http_status); | |
59 return HtmlResponseProviderImpl::Response("", response_headers); | |
60 } | |
61 | |
62 // static | |
63 HtmlResponseProviderImpl::Response HtmlResponseProviderImpl::GetSimpleResponse( | |
64 const std::string& body) { | |
65 auto response_headers = | |
66 web::ResponseProvider::GetResponseHeaders("text/html", net::HTTP_OK); | |
67 return HtmlResponseProviderImpl::Response(body, response_headers); | |
68 } | |
69 | |
70 HtmlResponseProviderImpl::HtmlResponseProviderImpl( | |
71 const std::map<GURL, std::string>& responses) | |
72 : responses_(BuildResponseMap( | |
73 responses, | |
74 web::ResponseProvider::GetDefaultResponseHeaders())) {} | |
75 | |
76 HtmlResponseProviderImpl::HtmlResponseProviderImpl( | |
77 const std::map<GURL, std::pair<std::string, std::string>>& responses) | |
78 : responses_(BuildResponseMap( | |
79 responses, | |
80 web::ResponseProvider::GetDefaultResponseHeaders(responses))) {} | |
81 | |
82 HtmlResponseProviderImpl::HtmlResponseProviderImpl( | |
83 const std::map<GURL, HtmlResponseProviderImpl::Response>& responses) | |
84 : responses_(responses) {} | |
85 | |
86 HtmlResponseProviderImpl::~HtmlResponseProviderImpl() {} | |
87 | |
88 bool HtmlResponseProviderImpl::CanHandleRequest( | |
89 const web::ResponseProvider::Request& request) { | |
90 return responses_.find(request.url) != responses_.end(); | |
91 } | |
92 | |
93 void HtmlResponseProviderImpl::GetResponseHeadersAndBody( | |
94 const web::ResponseProvider::Request& request, | |
95 scoped_refptr<net::HttpResponseHeaders>* headers, | |
96 std::string* response_body) { | |
97 const Response& response = responses_.at(request.url); | |
98 *headers = response.headers; | |
99 *response_body = response.body; | |
100 } | |
OLD | NEW |