Index: ios/web/public/test/response_providers/html_response_provider_impl.mm |
diff --git a/ios/web/public/test/response_providers/html_response_provider_impl.mm b/ios/web/public/test/response_providers/html_response_provider_impl.mm |
index 97f000df161467ee863cfb721c3d0272482c77ae..0ae64fec99b410f7b020c22a95ddc8d9d5a85579 100644 |
--- a/ios/web/public/test/response_providers/html_response_provider_impl.mm |
+++ b/ios/web/public/test/response_providers/html_response_provider_impl.mm |
@@ -20,6 +20,19 @@ |
} |
return responses; |
} |
+ |
+std::map<GURL, HtmlResponseProviderImpl::Response> BuildResponseMap( |
+ const std::map<GURL, std::pair<std::string, std::string>>& responses, |
+ const std::map<GURL, scoped_refptr<net::HttpResponseHeaders>>& headers) { |
+ std::map<GURL, HtmlResponseProviderImpl::Response> response_map; |
+ for (const auto& pair : responses) { |
+ response_map.insert(std::make_pair( |
+ pair.first, HtmlResponseProviderImpl::Response( |
+ pair.second.second, headers.at(pair.first)))); |
+ } |
+ return response_map; |
+} |
+ |
} // namespace |
HtmlResponseProviderImpl::Response::Response( |
@@ -61,11 +74,10 @@ |
web::ResponseProvider::GetDefaultResponseHeaders())) {} |
HtmlResponseProviderImpl::HtmlResponseProviderImpl( |
- const std::map<GURL, std::string>& responses, |
- const std::string& cookie) |
+ const std::map<GURL, std::pair<std::string, std::string>>& responses) |
: responses_(BuildResponseMap( |
responses, |
- web::ResponseProvider::GetDefaultResponseHeaders(cookie))) {} |
+ web::ResponseProvider::GetDefaultResponseHeaders(responses))) {} |
HtmlResponseProviderImpl::HtmlResponseProviderImpl( |
const std::map<GURL, HtmlResponseProviderImpl::Response>& responses) |