blob: 8f78c911a52ea5f9c7e534dd0f374c415277a616 (
plain)
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
// Copyright (c) 2012 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 NET_PROXY_MOCK_PROXY_RESOLVER_H_
#define NET_PROXY_MOCK_PROXY_RESOLVER_H_
#include <vector>
#include "base/memory/scoped_ptr.h"
#include "googleurl/src/gurl.h"
#include "net/base/net_errors.h"
#include "net/proxy/proxy_resolver.h"
class MessageLoop;
namespace net {
// Asynchronous mock proxy resolver. All requests complete asynchronously,
// user must call Request::CompleteNow() on a pending request to signal it.
class MockAsyncProxyResolverBase : public ProxyResolver {
public:
class Request : public base::RefCounted<Request> {
public:
Request(MockAsyncProxyResolverBase* resolver,
const GURL& url,
ProxyInfo* results,
const net::CompletionCallback& callback);
const GURL& url() const { return url_; }
ProxyInfo* results() const { return results_; }
const net::CompletionCallback& callback() const { return callback_; }
void CompleteNow(int rv);
private:
friend class base::RefCounted<Request>;
virtual ~Request();
MockAsyncProxyResolverBase* resolver_;
const GURL url_;
ProxyInfo* results_;
net::CompletionCallback callback_;
MessageLoop* origin_loop_;
};
class SetPacScriptRequest {
public:
SetPacScriptRequest(
MockAsyncProxyResolverBase* resolver,
const scoped_refptr<ProxyResolverScriptData>& script_data,
const net::CompletionCallback& callback);
~SetPacScriptRequest();
const ProxyResolverScriptData* script_data() const { return script_data_; }
void CompleteNow(int rv);
private:
MockAsyncProxyResolverBase* resolver_;
const scoped_refptr<ProxyResolverScriptData> script_data_;
net::CompletionCallback callback_;
MessageLoop* origin_loop_;
};
typedef std::vector<scoped_refptr<Request> > RequestsList;
virtual ~MockAsyncProxyResolverBase();
// ProxyResolver implementation.
virtual int GetProxyForURL(const GURL& url,
ProxyInfo* results,
const net::CompletionCallback& callback,
RequestHandle* request_handle,
const BoundNetLog& /*net_log*/) OVERRIDE;
virtual void CancelRequest(RequestHandle request_handle) OVERRIDE;
virtual LoadState GetLoadState(RequestHandle request_handle) const OVERRIDE;
virtual int SetPacScript(
const scoped_refptr<ProxyResolverScriptData>& script_data,
const net::CompletionCallback& callback) OVERRIDE;
virtual void CancelSetPacScript() OVERRIDE;
const RequestsList& pending_requests() const {
return pending_requests_;
}
const RequestsList& cancelled_requests() const {
return cancelled_requests_;
}
SetPacScriptRequest* pending_set_pac_script_request() const;
bool has_pending_set_pac_script_request() const {
return pending_set_pac_script_request_.get() != NULL;
}
void RemovePendingRequest(Request* request);
void RemovePendingSetPacScriptRequest(SetPacScriptRequest* request);
protected:
explicit MockAsyncProxyResolverBase(bool expects_pac_bytes);
private:
RequestsList pending_requests_;
RequestsList cancelled_requests_;
scoped_ptr<SetPacScriptRequest> pending_set_pac_script_request_;
};
class MockAsyncProxyResolver : public MockAsyncProxyResolverBase {
public:
MockAsyncProxyResolver()
: MockAsyncProxyResolverBase(false /*expects_pac_bytes*/) {}
};
class MockAsyncProxyResolverExpectsBytes : public MockAsyncProxyResolverBase {
public:
MockAsyncProxyResolverExpectsBytes()
: MockAsyncProxyResolverBase(true /*expects_pac_bytes*/) {}
};
} // namespace net
#endif // NET_PROXY_MOCK_PROXY_RESOLVER_H_
|
