Adding a way to mock WebURLLoader in webkit_support....

webkit/support/weburl_loader_mock_factory.h

+// Copyright (c) 2010 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 WEBKIT_SUPPORT_WEBURL_LOADER_MOCK_FACTORY_H_
+#define WEBKIT_SUPPORT_WEBURL_LOADER_MOCK_FACTORY_H_
+
+#include <map>
+#include <vector>
+
+#include "base/file_path.h"
+#include "third_party/WebKit/WebKit/chromium/public/WebURL.h"
+#include "third_party/WebKit/WebKit/chromium/public/WebURLRequest.h"
+#include "third_party/WebKit/WebKit/chromium/public/WebURLResponse.h"
+
+namespace WebKit {
+class WebData;
+struct WebURLError;
+class WebURLLoader;
+}
+
+class WebURLLoaderMock;
+
+// A factory that creates WebURLLoaderMock to simulate resource loading in
+// tests.
+// You register files for specific URLs, the content of the file is then served
+// when these URLs are loaded.
+// In order to serve the asynchronous requests, you need to invoke
+// ServeAsynchronousRequest.
+class WebURLLoaderMockFactory {
+ public:
+  WebURLLoaderMockFactory() {}
+  virtual ~WebURLLoaderMockFactory() {}
+
+  // Called by TestWebKitClient to create a WebURLLoader.
+  // Non-mocked request are forwarded to |default_loader| which should not be
+  // NULL.
+  virtual WebKit::WebURLLoader* CreateURLLoader(
+      WebKit::WebURLLoader* default_loader);
+
+  // Registers a response and the contents to be served when the specified URL
+  // is loaded.
+  void RegisterURL(const WebKit::WebURL& url,
+                   const WebKit::WebURLResponse& response,
+                   const WebKit::WebString& filePath);
+
+  // Unregisters |url| so it will no longer be mocked.
+  void UnregisterURL(const WebKit::WebURL& url);
+
+  // Unregister all URLs so no URL will be mocked anymore.
+  void UnregisterAllURLs();
+
+  // Serves all the pending asynchronous requests.
+  void ServeAsynchronousRequests();
+
+  // Returns true if |url| was registered for being mocked.
+  bool IsMockedURL(const WebKit::WebURL& url);
+
+  // Called by the loader to load a resource.
+  void LoadSynchronously(const WebKit::WebURLRequest& request,
+                         WebKit::WebURLResponse* response,
+                         WebKit::WebURLError* error,
+                         WebKit::WebData* data);
+  void LoadAsynchronouly(const WebKit::WebURLRequest& request,
+                         WebURLLoaderMock* loader);
+
+  // Removes the loader from the list of pending loaders.
+  void CancelLoad(WebURLLoaderMock* loader);
+
+ private:
+  struct ResponseInfo {
+    WebKit::WebURLResponse response;
+    FilePath file_path;
+  };
+
+  // Loads the specified request and populates the response, error and data
+  // accordingly.
+  void LoadRequest(const WebKit::WebURLRequest& request,
+                   WebKit::WebURLResponse* response,
+                   WebKit::WebURLError* error,
+                   WebKit::WebData* data);
+
+  // Reads |m_filePath| and puts its content in |data|.
+  // Returns true if it successfully read the file.
+  static bool ReadFile(const FilePath& file_path, WebKit::WebData* data);
+
+  // The loaders that have not being served data yet.
+  typedef std::map<WebURLLoaderMock*, WebKit::WebURLRequest> LoaderToRequestMap;
+  LoaderToRequestMap pending_loaders_;
+
+  // Table of the registered URLs and the responses that they should receive.
+  typedef std::map<WebKit::WebURL, ResponseInfo> URLToResponseMap;
+  URLToResponseMap url_to_reponse_info_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebURLLoaderMockFactory);
+};
+
+#endif  // WEBKIT_SUPPORT_WEBURL_LOADER_MOCK_FACTORY_H_
+