Index: ios/web/public/test/response_providers/html_response_provider.h |
diff --git a/ios/web/public/test/response_providers/html_response_provider.h b/ios/web/public/test/response_providers/html_response_provider.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..26f0020b728b115a38e52944dd3a8f18e5e791ef |
--- /dev/null |
+++ b/ios/web/public/test/response_providers/html_response_provider.h |
@@ -0,0 +1,48 @@ |
+// Copyright 2016 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#ifndef IOS_WEB_PUBLIC_TEST_RESPONSE_PROVIDERS_HTML_RESPONSE_PROVIDER_H_ |
+#define IOS_WEB_PUBLIC_TEST_RESPONSE_PROVIDERS_HTML_RESPONSE_PROVIDER_H_ |
+ |
+#include <map> |
+#include <string> |
+ |
+#include "base/memory/ref_counted.h" |
+#include "ios/web/public/test/response_providers/data_response_provider.h" |
+#include "ios/web/public/test/response_providers/html_response_provider_impl.h" |
+#include "ios/web/public/test/response_providers/response_provider.h" |
+#include "url/gurl.h" |
+ |
+namespace net { |
+class HttpResponseHeaders; |
+} |
+ |
+// A web::DataResponseProvider that maps URLs to requests. |
+class HtmlResponseProvider : public web::DataResponseProvider { |
+ public: |
+ // Constructs an HtmlResponseProvider that does not respond to any request. |
+ HtmlResponseProvider(); |
+ // Constructs an HtmlResponseProvider that generates a simple string response |
+ // to a URL based on the mapping present in |responses|. |
+ explicit HtmlResponseProvider(const std::map<GURL, std::string>& responses); |
+ // Constructs an HtmlResponseProvider that generates a response to a URL based |
+ // on the mapping present in |responses|. |
+ explicit HtmlResponseProvider( |
+ const std::map<GURL, HtmlResponseProviderImpl::Response>& responses); |
+ |
+ ~HtmlResponseProvider() override; |
+ |
+ // web::ResponseProvider implementation. |
+ bool CanHandleRequest(const Request& request) override; |
+ // web::DataResponseProvider implementation. |
+ void GetResponseHeadersAndBody( |
+ const Request& request, |
+ scoped_refptr<net::HttpResponseHeaders>* headers, |
+ std::string* response_body) override; |
+ |
+ private: |
+ std::unique_ptr<HtmlResponseProviderImpl> response_provider_impl_; |
+}; |
+ |
+#endif // IOS_WEB_PUBLIC_TEST_RESPONSE_PROVIDERS_HTML_RESPONSE_PROVIDER_H_ |