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

Side by Side Diff: chrome/browser/component_updater/test/url_request_post_interceptor.cc

Issue 300693005: Make URLRequestFilter use URLRequestInterceptors. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Fix merge Created 6 years, 6 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 unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « no previous file | chrome/browser/errorpage_browsertest.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright 2013 The Chromium Authors. All rights reserved. 1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "chrome/browser/component_updater/test/url_request_post_interceptor.h" 5 #include "chrome/browser/component_updater/test/url_request_post_interceptor.h"
6 #include "base/file_util.h" 6 #include "base/file_util.h"
7 #include "base/memory/scoped_ptr.h" 7 #include "base/memory/scoped_ptr.h"
8 #include "base/strings/stringprintf.h" 8 #include "base/strings/stringprintf.h"
9 #include "content/public/test/test_browser_thread.h" 9 #include "content/public/test/test_browser_thread.h"
10 #include "net/base/upload_bytes_element_reader.h" 10 #include "net/base/upload_bytes_element_reader.h"
11 #include "net/url_request/url_request.h" 11 #include "net/url_request/url_request.h"
12 #include "net/url_request/url_request_filter.h" 12 #include "net/url_request/url_request_filter.h"
13 #include "net/url_request/url_request_job_factory.h" 13 #include "net/url_request/url_request_interceptor.h"
14 #include "net/url_request/url_request_simple_job.h" 14 #include "net/url_request/url_request_simple_job.h"
15 #include "net/url_request/url_request_test_util.h" 15 #include "net/url_request/url_request_test_util.h"
16 16
17 using content::BrowserThread; 17 using content::BrowserThread;
18 18
19 namespace component_updater { 19 namespace component_updater {
20 20
21 // Returns a canned response. 21 // Returns a canned response.
22 class URLRequestMockJob : public net::URLRequestSimpleJob { 22 class URLRequestMockJob : public net::URLRequestSimpleJob {
23 public: 23 public:
(...skipping 90 matching lines...) Expand 10 before | Expand all | Expand 10 after
114 return s; 114 return s;
115 } 115 }
116 116
117 void URLRequestPostInterceptor::Reset() { 117 void URLRequestPostInterceptor::Reset() {
118 base::AutoLock auto_lock(interceptor_lock_); 118 base::AutoLock auto_lock(interceptor_lock_);
119 hit_count_ = 0; 119 hit_count_ = 0;
120 requests_.clear(); 120 requests_.clear();
121 ClearExpectations(); 121 ClearExpectations();
122 } 122 }
123 123
124 class URLRequestPostInterceptor::Delegate 124 class URLRequestPostInterceptor::Delegate : public net::URLRequestInterceptor {
125 : public net::URLRequestJobFactory::ProtocolHandler {
126 public: 125 public:
127 Delegate(const std::string& scheme, const std::string& hostname) 126 Delegate(const std::string& scheme, const std::string& hostname)
128 : scheme_(scheme), hostname_(hostname) {} 127 : scheme_(scheme), hostname_(hostname) {}
129 128
130 void Register() { 129 void Register() {
131 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 130 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
132 net::URLRequestFilter::GetInstance()->AddHostnameProtocolHandler( 131 net::URLRequestFilter::GetInstance()->AddHostnameInterceptor(
133 scheme_, 132 scheme_, hostname_, scoped_ptr<net::URLRequestInterceptor>(this));
134 hostname_,
135 scoped_ptr<net::URLRequestJobFactory::ProtocolHandler>(this));
136 } 133 }
137 134
138 void Unregister() { 135 void Unregister() {
139 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 136 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
140 for (InterceptorMap::iterator it = interceptors_.begin(); 137 for (InterceptorMap::iterator it = interceptors_.begin();
141 it != interceptors_.end(); 138 it != interceptors_.end();
142 ++it) 139 ++it)
143 delete (*it).second; 140 delete (*it).second;
144 net::URLRequestFilter::GetInstance()->RemoveHostnameHandler(scheme_, 141 net::URLRequestFilter::GetInstance()->RemoveHostnameHandler(scheme_,
145 hostname_); 142 hostname_);
146 } 143 }
147 144
148 void OnCreateInterceptor(URLRequestPostInterceptor* interceptor) { 145 void OnCreateInterceptor(URLRequestPostInterceptor* interceptor) {
149 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 146 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
150 CHECK(interceptors_.find(interceptor->GetUrl()) == interceptors_.end()); 147 CHECK(interceptors_.find(interceptor->GetUrl()) == interceptors_.end());
151 148
152 interceptors_.insert(std::make_pair(interceptor->GetUrl(), interceptor)); 149 interceptors_.insert(std::make_pair(interceptor->GetUrl(), interceptor));
153 } 150 }
154 151
155 private: 152 private:
156 virtual ~Delegate() {} 153 virtual ~Delegate() {}
157 154
158 virtual net::URLRequestJob* MaybeCreateJob( 155 virtual net::URLRequestJob* MaybeInterceptRequest(
159 net::URLRequest* request, 156 net::URLRequest* request,
160 net::NetworkDelegate* network_delegate) const OVERRIDE { 157 net::NetworkDelegate* network_delegate) const OVERRIDE {
161 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 158 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
162 159
163 // Only intercepts POST. 160 // Only intercepts POST.
164 if (!request->has_upload()) 161 if (!request->has_upload())
165 return NULL; 162 return NULL;
166 163
167 GURL url = request->url(); 164 GURL url = request->url();
168 if (url.has_query()) { 165 if (url.has_query()) {
(...skipping 84 matching lines...) Expand 10 before | Expand all | Expand 10 after
253 base::Unretained(interceptor))); 250 base::Unretained(interceptor)));
254 if (!res) { 251 if (!res) {
255 delete interceptor; 252 delete interceptor;
256 return NULL; 253 return NULL;
257 } 254 }
258 255
259 return interceptor; 256 return interceptor;
260 } 257 }
261 258
262 } // namespace component_updater 259 } // namespace component_updater
OLDNEW
« no previous file with comments | « no previous file | chrome/browser/errorpage_browsertest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698