Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(261)

Unified Diff: net/proxy/mock_proxy_resolver.cc

Issue 1102463002: Add a MockAsyncProxyResolverFactory and update some tests to use it. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@proxy-factory-refactor
Patch Set: Created 5 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: net/proxy/mock_proxy_resolver.cc
diff --git a/net/proxy/mock_proxy_resolver.cc b/net/proxy/mock_proxy_resolver.cc
index 2ff09007eb116822b819dfa78806771987a5b7bd..95591b30522a40d82da6272d9da8200712e44b66 100644
--- a/net/proxy/mock_proxy_resolver.cc
+++ b/net/proxy/mock_proxy_resolver.cc
@@ -112,6 +112,90 @@ void MockAsyncProxyResolverBase::RemovePendingSetPacScriptRequest(
MockAsyncProxyResolverBase::MockAsyncProxyResolverBase(bool expects_pac_bytes)
: ProxyResolver(expects_pac_bytes) {}
+MockAsyncProxyResolverFactory::Request::Request(
+ MockAsyncProxyResolverFactory* factory,
+ const scoped_refptr<ProxyResolverScriptData>& script_data,
+ scoped_ptr<ProxyResolver>* resolver,
+ const CompletionCallback& callback)
+ : factory_(factory),
+ script_data_(script_data),
+ resolver_(resolver),
+ callback_(callback) {
+}
+
+MockAsyncProxyResolverFactory::Request::~Request() {
+}
+
+void MockAsyncProxyResolverFactory::Request::CompleteNow(
+ int rv,
+ scoped_ptr<ProxyResolver> resolver) {
+ CompletionCallback callback = callback_;
eroman 2015/04/22 16:30:38 How about instead: scoped_refptr<Request> keep_al
Sam McNally 2015/04/23 03:04:43 Done.
+ *resolver_ = resolver.Pass();
+
+ // Will delete |this|.
+ factory_->RemovePendingRequest(this);
+
+ callback.Run(rv);
+}
+
+void MockAsyncProxyResolverFactory::Request::CompleteNowWithForwarder(
+ int rv,
+ ProxyResolver* resolver) {
+ DCHECK(resolver);
+ CompleteNow(rv, make_scoped_ptr(new ForwardingProxyResolver(resolver)));
+}
+
+class MockAsyncProxyResolverFactory::Job
+ : public ProxyResolverFactory::Request {
+ public:
+ Job(MockAsyncProxyResolverFactory* factory,
+ const scoped_refptr<MockAsyncProxyResolverFactory::Request>& request)
+ : factory_(factory), request_(request) {}
+ ~Job() override {
+ if (std::find(factory_->pending_requests_.begin(),
eroman 2015/04/22 16:30:38 The assumption here is that the ProxyResolverFacto
Sam McNally 2015/04/23 03:04:43 Done.
+ factory_->pending_requests_.end(),
+ request_.get()) == factory_->pending_requests_.end()) {
+ return;
+ }
+ factory_->cancelled_requests_.push_back(request_);
+ factory_->RemovePendingRequest(request_.get());
+ }
+
+ private:
+ MockAsyncProxyResolverFactory* factory_;
+ scoped_refptr<MockAsyncProxyResolverFactory::Request> request_;
+};
+
+MockAsyncProxyResolverFactory::MockAsyncProxyResolverFactory(
+ bool resolvers_expect_pac_bytes)
+ : ProxyResolverFactory(resolvers_expect_pac_bytes) {
+}
+
+int MockAsyncProxyResolverFactory::CreateProxyResolver(
+ const scoped_refptr<ProxyResolverScriptData>& pac_script,
+ scoped_ptr<ProxyResolver>* resolver,
+ const net::CompletionCallback& callback,
+ scoped_ptr<ProxyResolverFactory::Request>* request_handle) {
+ scoped_refptr<Request> request =
+ new Request(this, pac_script, resolver, callback);
+ pending_requests_.push_back(request);
+
+ request_handle->reset(new Job(this, request));
eroman 2015/04/22 16:30:38 see comment above. |this| is redundant with |reque
Sam McNally 2015/04/23 03:04:43 Done.
+
+ // Test code completes the request by calling request->CompleteNow().
+ return ERR_IO_PENDING;
+}
+
+void MockAsyncProxyResolverFactory::RemovePendingRequest(Request* request) {
+ RequestsList::iterator it =
+ std::find(pending_requests_.begin(), pending_requests_.end(), request);
+ DCHECK(it != pending_requests_.end());
+ pending_requests_.erase(it);
+}
+
+MockAsyncProxyResolverFactory::~MockAsyncProxyResolverFactory() {
eroman 2015/04/22 16:30:38 See earlier comment about cancelling outstanding r
Sam McNally 2015/04/23 03:04:43 Done.
+}
+
ForwardingProxyResolver::ForwardingProxyResolver(ProxyResolver* impl)
: ProxyResolver(impl->expects_pac_bytes()), impl_(impl) {
}
@@ -143,14 +227,16 @@ int ForwardingProxyResolver::SetPacScript(
}
ForwardingProxyResolverFactory::ForwardingProxyResolverFactory(
- ProxyResolver* resolver)
- : LegacyProxyResolverFactory(resolver->expects_pac_bytes()),
- resolver_(resolver) {
+ ProxyResolverFactory* impl)
+ : ProxyResolverFactory(impl->expects_pac_bytes()), impl_(impl) {
}
-scoped_ptr<ProxyResolver>
-ForwardingProxyResolverFactory::CreateProxyResolver() {
- return make_scoped_ptr(new ForwardingProxyResolver(resolver_));
+int ForwardingProxyResolverFactory::CreateProxyResolver(
+ const scoped_refptr<ProxyResolverScriptData>& pac_script,
+ scoped_ptr<ProxyResolver>* resolver,
+ const net::CompletionCallback& callback,
+ scoped_ptr<Request>* request) {
+ return impl_->CreateProxyResolver(pac_script, resolver, callback, request);
}
} // namespace net

Powered by Google App Engine
This is Rietveld 408576698