1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
// 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 "base/file_path.h"
#include "third_party/WebKit/Source/WebKit/chromium/public/WebURL.h"
#include "third_party/WebKit/Source/WebKit/chromium/public/WebURLRequest.h"
#include "third_party/WebKit/Source/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;
// 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_
|
