Index: third_party/WebKit/Source/platform/testing/weburl_loader_mock_factory_impl.h |
diff --git a/third_party/WebKit/Source/platform/testing/weburl_loader_mock_factory_impl.h b/third_party/WebKit/Source/platform/testing/weburl_loader_mock_factory_impl.h |
index 8fe9d11453917b5f6b0d718e4d9b16d7baece896..53b63536b2478f7498543fcfadfe01dfc6f8cadf 100644 |
--- a/third_party/WebKit/Source/platform/testing/weburl_loader_mock_factory_impl.h |
+++ b/third_party/WebKit/Source/platform/testing/weburl_loader_mock_factory_impl.h |
@@ -45,6 +45,10 @@ class WebURLLoaderMockFactoryImpl : public WebURLLoaderMockFactory { |
const WebURLResponse& response, |
const WebURLError& error) override; |
void UnregisterURL(const WebURL& url) override; |
+ void RegisterURLProtocol(const WebString& protocol, |
+ const WebURLResponse& response, |
+ const WebString& file_path) override; |
+ void UnregisterURLProtocol(const WebString& protocol) override; |
void UnregisterAllURLsAndClearMemoryCache() override; |
void ServeAsynchronousRequests() override; |
void SetLoaderDelegate(WebURLLoaderTestDelegate* delegate) override { |
@@ -84,6 +88,13 @@ class WebURLLoaderMockFactoryImpl : public WebURLLoaderMockFactory { |
// Checks if the loader is pending. Otherwise, it may have been deleted. |
bool IsPending(WeakPtr<WebURLLoaderMock> loader); |
+ // Looks up an URL in the mock URL table. |
+ // |
+ // If the URL is found, returns true and sets |error| and |response_info|. |
+ bool LookupURL(const WebURL& url, |
+ WebURLError* error, |
+ ResponseInfo* response_info); |
+ |
// Reads |m_filePath| and puts its content in |data|. |
// Returns true if it successfully read the file. |
static bool ReadFile(const base::FilePath& file_path, WebData* data); |
@@ -101,6 +112,11 @@ class WebURLLoaderMockFactoryImpl : public WebURLLoaderMockFactory { |
using URLToResponseMap = HashMap<KURL, ResponseInfo>; |
URLToResponseMap url_to_response_info_; |
+ // Table of the registered URL protocols and the responses that they should |
+ // receive. |
+ using ProtocolToResponseMap = HashMap<String, ResponseInfo>; |
+ ProtocolToResponseMap protocol_to_response_info_; |
+ |
TestingPlatformSupport* platform_; |
DISALLOW_COPY_AND_ASSIGN(WebURLLoaderMockFactoryImpl); |