| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef NET_PROXY_MOCK_PROXY_RESOLVER_H_ | |
| 6 #define NET_PROXY_MOCK_PROXY_RESOLVER_H_ | |
| 7 | |
| 8 #include <vector> | |
| 9 | |
| 10 #include "base/memory/scoped_ptr.h" | |
| 11 #include "net/base/net_errors.h" | |
| 12 #include "net/proxy/proxy_resolver.h" | |
| 13 #include "url/gurl.h" | |
| 14 | |
| 15 namespace base { | |
| 16 class MessageLoop; | |
| 17 } | |
| 18 | |
| 19 namespace net { | |
| 20 | |
| 21 // Asynchronous mock proxy resolver. All requests complete asynchronously, | |
| 22 // user must call Request::CompleteNow() on a pending request to signal it. | |
| 23 class MockAsyncProxyResolverBase : public ProxyResolver { | |
| 24 public: | |
| 25 class Request : public base::RefCounted<Request> { | |
| 26 public: | |
| 27 Request(MockAsyncProxyResolverBase* resolver, | |
| 28 const GURL& url, | |
| 29 ProxyInfo* results, | |
| 30 const net::CompletionCallback& callback); | |
| 31 | |
| 32 const GURL& url() const { return url_; } | |
| 33 ProxyInfo* results() const { return results_; } | |
| 34 const net::CompletionCallback& callback() const { return callback_; } | |
| 35 | |
| 36 void CompleteNow(int rv); | |
| 37 | |
| 38 private: | |
| 39 friend class base::RefCounted<Request>; | |
| 40 | |
| 41 virtual ~Request(); | |
| 42 | |
| 43 MockAsyncProxyResolverBase* resolver_; | |
| 44 const GURL url_; | |
| 45 ProxyInfo* results_; | |
| 46 net::CompletionCallback callback_; | |
| 47 base::MessageLoop* origin_loop_; | |
| 48 }; | |
| 49 | |
| 50 class SetPacScriptRequest { | |
| 51 public: | |
| 52 SetPacScriptRequest( | |
| 53 MockAsyncProxyResolverBase* resolver, | |
| 54 const scoped_refptr<ProxyResolverScriptData>& script_data, | |
| 55 const net::CompletionCallback& callback); | |
| 56 ~SetPacScriptRequest(); | |
| 57 | |
| 58 const ProxyResolverScriptData* script_data() const { | |
| 59 return script_data_.get(); | |
| 60 } | |
| 61 | |
| 62 void CompleteNow(int rv); | |
| 63 | |
| 64 private: | |
| 65 MockAsyncProxyResolverBase* resolver_; | |
| 66 const scoped_refptr<ProxyResolverScriptData> script_data_; | |
| 67 net::CompletionCallback callback_; | |
| 68 base::MessageLoop* origin_loop_; | |
| 69 }; | |
| 70 | |
| 71 typedef std::vector<scoped_refptr<Request> > RequestsList; | |
| 72 | |
| 73 ~MockAsyncProxyResolverBase() override; | |
| 74 | |
| 75 // ProxyResolver implementation. | |
| 76 int GetProxyForURL(const GURL& url, | |
| 77 ProxyInfo* results, | |
| 78 const net::CompletionCallback& callback, | |
| 79 RequestHandle* request_handle, | |
| 80 const BoundNetLog& /*net_log*/) override; | |
| 81 void CancelRequest(RequestHandle request_handle) override; | |
| 82 LoadState GetLoadState(RequestHandle request_handle) const override; | |
| 83 int SetPacScript(const scoped_refptr<ProxyResolverScriptData>& script_data, | |
| 84 const net::CompletionCallback& callback) override; | |
| 85 void CancelSetPacScript() override; | |
| 86 | |
| 87 const RequestsList& pending_requests() const { | |
| 88 return pending_requests_; | |
| 89 } | |
| 90 | |
| 91 const RequestsList& cancelled_requests() const { | |
| 92 return cancelled_requests_; | |
| 93 } | |
| 94 | |
| 95 SetPacScriptRequest* pending_set_pac_script_request() const; | |
| 96 | |
| 97 bool has_pending_set_pac_script_request() const { | |
| 98 return pending_set_pac_script_request_.get() != NULL; | |
| 99 } | |
| 100 | |
| 101 void RemovePendingRequest(Request* request); | |
| 102 | |
| 103 void RemovePendingSetPacScriptRequest(SetPacScriptRequest* request); | |
| 104 | |
| 105 protected: | |
| 106 explicit MockAsyncProxyResolverBase(bool expects_pac_bytes); | |
| 107 | |
| 108 private: | |
| 109 RequestsList pending_requests_; | |
| 110 RequestsList cancelled_requests_; | |
| 111 scoped_ptr<SetPacScriptRequest> pending_set_pac_script_request_; | |
| 112 }; | |
| 113 | |
| 114 class MockAsyncProxyResolver : public MockAsyncProxyResolverBase { | |
| 115 public: | |
| 116 MockAsyncProxyResolver() | |
| 117 : MockAsyncProxyResolverBase(false /*expects_pac_bytes*/) {} | |
| 118 }; | |
| 119 | |
| 120 class MockAsyncProxyResolverExpectsBytes : public MockAsyncProxyResolverBase { | |
| 121 public: | |
| 122 MockAsyncProxyResolverExpectsBytes() | |
| 123 : MockAsyncProxyResolverBase(true /*expects_pac_bytes*/) {} | |
| 124 }; | |
| 125 | |
| 126 } // namespace net | |
| 127 | |
| 128 #endif // NET_PROXY_MOCK_PROXY_RESOLVER_H_ | |
| OLD | NEW |