Chromium Code Reviews| Index: chrome/browser/component_updater/test/url_request_post_interceptor.cc |
| diff --git a/chrome/browser/component_updater/test/url_request_post_interceptor.cc b/chrome/browser/component_updater/test/url_request_post_interceptor.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..f5917a276d3f4f9a2e84d42fa8d31d3df2bccbeb |
| --- /dev/null |
| +++ b/chrome/browser/component_updater/test/url_request_post_interceptor.cc |
| @@ -0,0 +1,68 @@ |
| +// 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. |
| + |
| +#include "chrome/browser/component_updater/test/url_request_post_interceptor.h" |
| + |
| +#include "content/public/test/test_browser_thread.h" |
| +#include "net/url_request/url_request.h" |
| +#include "net/url_request/url_request_filter.h" |
| +#include "net/url_request/url_request_test_util.h" |
| + |
| +using content::BrowserThread; |
| + |
| +URLRequestPostInterceptor::URLRequestPostInterceptor( |
| + RequestCounter* counter) : delegate_(new Delegate(counter)) { |
| + BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, |
| + base::Bind(&Delegate::Register, |
| + base::Unretained(delegate_))); |
| +} |
| + |
| +URLRequestPostInterceptor::~URLRequestPostInterceptor() { |
| + BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, |
| + base::Bind(&Delegate::Unregister, |
| + base::Unretained(delegate_))); |
| +} |
| + |
| +URLRequestPostInterceptor::Delegate::Delegate( |
| + RequestCounter* counter) : counter_(counter) { |
| +} |
| + |
| +void URLRequestPostInterceptor::Delegate::Register() { |
| + net::URLRequestFilter::GetInstance()->AddHostnameProtocolHandler( |
| + "http", "localhost2", |
| + scoped_ptr<net::URLRequestJobFactory::ProtocolHandler>(this)); |
| +} |
| + |
| +void URLRequestPostInterceptor::Delegate::Unregister() { |
| + net::URLRequestFilter::GetInstance()-> |
| + RemoveHostnameHandler("http", "localhost2"); |
| +} |
| + |
| +net::URLRequestJob* URLRequestPostInterceptor::Delegate::MaybeCreateJob( |
| + net::URLRequest* request, |
| + net::NetworkDelegate* network_delegate) const { |
| + CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| + if (request->has_upload()) { |
| + counter_->Trial(request); |
| + return new URLRequestPingMockJob(request, network_delegate); |
| + } |
| + return NULL; |
| +} |
| + |
|
cpu_(ooo_6.6-7.5)
2013/06/20 22:34:25
pingMockJob and related code, I think is not neede
Sorin Jianu
2013/06/20 23:51:59
Done.
|
| +URLRequestPingMockJob::URLRequestPingMockJob( |
| + net::URLRequest* request, |
| + net::NetworkDelegate* network_delegate) |
| + : net::URLRequestSimpleJob(request, network_delegate) { |
| +} |
| + |
| +int URLRequestPingMockJob::GetData( |
| + std::string* mime_type, |
| + std::string* charset, |
| + std::string* data, |
| + const net::CompletionCallback& callback) const { |
| + mime_type->assign("text/plain"); |
| + charset->assign("US-ASCII"); |
| + data->assign(""); // There is no reason to have a response body. |
| + return net::OK; |
| +} |