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
content / test / fake_network.h [blame]
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CONTENT_TEST_FAKE_NETWORK_H_
#define CONTENT_TEST_FAKE_NETWORK_H_
#include "content/public/test/url_loader_interceptor.h"
namespace content {
// Mocks network activity. Meant to be used by URLLoaderInterceptor.
//
// 1. By default, it returns 200 OK with a simple body to any request:
// If request path ends in ".js", the body is
// "/*this body came from the network*/". Otherwise, the body is
// "this body came from the network".
// 2. Call SetDefaultResponse() to change the default response.
// 3. Call SetResponse() to set the response for a specific url.
//
// Example:
// FakeNetwork fake_network;
// URLLoaderInterceptor interceptor(base::BindRepeating(
// &FakeNetwork::HandleRequest, base::Unretained(&fake_network_)));
//
// // |fake_network| will now handle any network request.
//
// // Customize a response.
// fake_network_.SetResponse(...);
class FakeNetwork {
public:
FakeNetwork();
~FakeNetwork();
// Callback for URLLoaderInterceptor.
bool HandleRequest(URLLoaderInterceptor::RequestParams* params);
// Sets the default response.
void SetDefaultResponse(const std::string& headers,
const std::string& body,
bool network_accessed,
net::Error error_code);
// Sets the response for a specific url, to use instead of the default.
void SetResponse(const GURL& url,
const std::string& headers,
const std::string& body,
bool network_accessed,
net::Error error_code);
private:
struct ResponseInfo;
// Returns the ResponseInfo for the |url|, it follows the order:
// 1. Returns the matching entry in |response_info_map_| if exists.
// 2. Returns |user_defined_default_response_info_| if it's set.
// 3. Returns default response info (defined inside this method).
const ResponseInfo& FindResponseInfo(const GURL& url) const;
// This is user-defined default response info, it overrides the default
// response info.
std::unique_ptr<ResponseInfo> user_defined_default_response_info_;
// User-defined URL => ResponseInfo map.
base::flat_map<GURL, ResponseInfo> response_info_map_;
};
} // namespace content
#endif // CONTENT_TEST_FAKE_NETWORK_H_