| Index: components/component_updater/test/url_request_post_interceptor.cc
|
| diff --git a/components/component_updater/test/url_request_post_interceptor.cc b/components/component_updater/test/url_request_post_interceptor.cc
|
| deleted file mode 100644
|
| index f2dc839ccd0492ea03b3528108b79d01f0e7e0ae..0000000000000000000000000000000000000000
|
| --- a/components/component_updater/test/url_request_post_interceptor.cc
|
| +++ /dev/null
|
| @@ -1,306 +0,0 @@
|
| -// Copyright 2013 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 "components/component_updater/test/url_request_post_interceptor.h"
|
| -
|
| -#include "base/files/file_util.h"
|
| -#include "base/macros.h"
|
| -#include "base/memory/scoped_ptr.h"
|
| -#include "base/strings/stringprintf.h"
|
| -#include "components/component_updater/test/test_configurator.h"
|
| -#include "net/base/upload_bytes_element_reader.h"
|
| -#include "net/base/upload_data_stream.h"
|
| -#include "net/url_request/url_request.h"
|
| -#include "net/url_request/url_request_filter.h"
|
| -#include "net/url_request/url_request_interceptor.h"
|
| -#include "net/url_request/url_request_simple_job.h"
|
| -#include "net/url_request/url_request_test_util.h"
|
| -
|
| -namespace component_updater {
|
| -
|
| -// Returns a canned response.
|
| -class URLRequestMockJob : public net::URLRequestSimpleJob {
|
| - public:
|
| - URLRequestMockJob(net::URLRequest* request,
|
| - net::NetworkDelegate* network_delegate,
|
| - int response_code,
|
| - const std::string& response_body)
|
| - : net::URLRequestSimpleJob(request, network_delegate),
|
| - response_code_(response_code),
|
| - response_body_(response_body) {}
|
| -
|
| - protected:
|
| - int GetResponseCode() const override { return response_code_; }
|
| -
|
| - int GetData(std::string* mime_type,
|
| - std::string* charset,
|
| - std::string* data,
|
| - const net::CompletionCallback& callback) const override {
|
| - mime_type->assign("text/plain");
|
| - charset->assign("US-ASCII");
|
| - data->assign(response_body_);
|
| - return net::OK;
|
| - }
|
| -
|
| - private:
|
| - ~URLRequestMockJob() override {}
|
| -
|
| - int response_code_;
|
| - std::string response_body_;
|
| - DISALLOW_COPY_AND_ASSIGN(URLRequestMockJob);
|
| -};
|
| -
|
| -URLRequestPostInterceptor::URLRequestPostInterceptor(
|
| - const GURL& url,
|
| - const scoped_refptr<base::SequencedTaskRunner>& io_task_runner)
|
| - : url_(url), io_task_runner_(io_task_runner), hit_count_(0) {
|
| -}
|
| -
|
| -URLRequestPostInterceptor::~URLRequestPostInterceptor() {
|
| - DCHECK(io_task_runner_->RunsTasksOnCurrentThread());
|
| - ClearExpectations();
|
| -}
|
| -
|
| -void URLRequestPostInterceptor::ClearExpectations() {
|
| - while (!expectations_.empty()) {
|
| - Expectation expectation(expectations_.front());
|
| - delete expectation.first;
|
| - expectations_.pop();
|
| - }
|
| -}
|
| -
|
| -GURL URLRequestPostInterceptor::GetUrl() const {
|
| - return url_;
|
| -}
|
| -
|
| -bool URLRequestPostInterceptor::ExpectRequest(
|
| - class RequestMatcher* request_matcher) {
|
| - expectations_.push(std::make_pair(request_matcher,
|
| - ExpectationResponse(kResponseCode200, "")));
|
| - return true;
|
| -}
|
| -
|
| -bool URLRequestPostInterceptor::ExpectRequest(
|
| - class RequestMatcher* request_matcher,
|
| - int response_code) {
|
| - expectations_.push(
|
| - std::make_pair(request_matcher, ExpectationResponse(response_code, "")));
|
| - return true;
|
| -}
|
| -
|
| -bool URLRequestPostInterceptor::ExpectRequest(
|
| - class RequestMatcher* request_matcher,
|
| - const base::FilePath& filepath) {
|
| - std::string response;
|
| - if (filepath.empty() || !base::ReadFileToString(filepath, &response))
|
| - return false;
|
| -
|
| - expectations_.push(std::make_pair(
|
| - request_matcher, ExpectationResponse(kResponseCode200, response)));
|
| - return true;
|
| -}
|
| -
|
| -int URLRequestPostInterceptor::GetHitCount() const {
|
| - base::AutoLock auto_lock(interceptor_lock_);
|
| - return hit_count_;
|
| -}
|
| -
|
| -int URLRequestPostInterceptor::GetCount() const {
|
| - base::AutoLock auto_lock(interceptor_lock_);
|
| - return static_cast<int>(requests_.size());
|
| -}
|
| -
|
| -std::vector<std::string> URLRequestPostInterceptor::GetRequests() const {
|
| - base::AutoLock auto_lock(interceptor_lock_);
|
| - return requests_;
|
| -}
|
| -
|
| -std::string URLRequestPostInterceptor::GetRequestsAsString() const {
|
| - std::vector<std::string> requests(GetRequests());
|
| -
|
| - std::string s = "Requests are:";
|
| -
|
| - int i = 0;
|
| - for (std::vector<std::string>::const_iterator it = requests.begin();
|
| - it != requests.end();
|
| - ++it) {
|
| - s.append(base::StringPrintf("\n (%d): %s", ++i, it->c_str()));
|
| - }
|
| -
|
| - return s;
|
| -}
|
| -
|
| -void URLRequestPostInterceptor::Reset() {
|
| - base::AutoLock auto_lock(interceptor_lock_);
|
| - hit_count_ = 0;
|
| - requests_.clear();
|
| - ClearExpectations();
|
| -}
|
| -
|
| -class URLRequestPostInterceptor::Delegate : public net::URLRequestInterceptor {
|
| - public:
|
| - Delegate(const std::string& scheme,
|
| - const std::string& hostname,
|
| - const scoped_refptr<base::SequencedTaskRunner>& io_task_runner)
|
| - : scheme_(scheme), hostname_(hostname), io_task_runner_(io_task_runner) {}
|
| -
|
| - void Register() {
|
| - DCHECK(io_task_runner_->RunsTasksOnCurrentThread());
|
| - net::URLRequestFilter::GetInstance()->AddHostnameInterceptor(
|
| - scheme_, hostname_, scoped_ptr<net::URLRequestInterceptor>(this));
|
| - }
|
| -
|
| - void Unregister() {
|
| - DCHECK(io_task_runner_->RunsTasksOnCurrentThread());
|
| - for (InterceptorMap::iterator it = interceptors_.begin();
|
| - it != interceptors_.end();
|
| - ++it)
|
| - delete (*it).second;
|
| - net::URLRequestFilter::GetInstance()->RemoveHostnameHandler(scheme_,
|
| - hostname_);
|
| - }
|
| -
|
| - void OnCreateInterceptor(URLRequestPostInterceptor* interceptor) {
|
| - DCHECK(io_task_runner_->RunsTasksOnCurrentThread());
|
| - DCHECK(interceptors_.find(interceptor->GetUrl()) == interceptors_.end());
|
| -
|
| - interceptors_.insert(std::make_pair(interceptor->GetUrl(), interceptor));
|
| - }
|
| -
|
| - private:
|
| - ~Delegate() override {}
|
| -
|
| - net::URLRequestJob* MaybeInterceptRequest(
|
| - net::URLRequest* request,
|
| - net::NetworkDelegate* network_delegate) const override {
|
| - DCHECK(io_task_runner_->RunsTasksOnCurrentThread());
|
| -
|
| - // Only intercepts POST.
|
| - if (!request->has_upload())
|
| - return NULL;
|
| -
|
| - GURL url = request->url();
|
| - if (url.has_query()) {
|
| - GURL::Replacements replacements;
|
| - replacements.ClearQuery();
|
| - url = url.ReplaceComponents(replacements);
|
| - }
|
| -
|
| - InterceptorMap::const_iterator it(interceptors_.find(url));
|
| - if (it == interceptors_.end())
|
| - return NULL;
|
| -
|
| - // There is an interceptor hooked up for this url. Read the request body,
|
| - // check the existing expectations, and handle the matching case by
|
| - // popping the expectation off the queue, counting the match, and
|
| - // returning a mock object to serve the canned response.
|
| - URLRequestPostInterceptor* interceptor(it->second);
|
| -
|
| - const net::UploadDataStream* stream = request->get_upload();
|
| - const net::UploadBytesElementReader* reader =
|
| - (*stream->GetElementReaders())[0]->AsBytesReader();
|
| - const int size = reader->length();
|
| - scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(size));
|
| - const std::string request_body(reader->bytes());
|
| -
|
| - {
|
| - base::AutoLock auto_lock(interceptor->interceptor_lock_);
|
| - interceptor->requests_.push_back(request_body);
|
| - if (interceptor->expectations_.empty())
|
| - return NULL;
|
| - const URLRequestPostInterceptor::Expectation& expectation(
|
| - interceptor->expectations_.front());
|
| - if (expectation.first->Match(request_body)) {
|
| - const int response_code(expectation.second.response_code);
|
| - const std::string response_body(expectation.second.response_body);
|
| - delete expectation.first;
|
| - interceptor->expectations_.pop();
|
| - ++interceptor->hit_count_;
|
| -
|
| - return new URLRequestMockJob(
|
| - request, network_delegate, response_code, response_body);
|
| - }
|
| - }
|
| -
|
| - return NULL;
|
| - }
|
| -
|
| - typedef std::map<GURL, URLRequestPostInterceptor*> InterceptorMap;
|
| - InterceptorMap interceptors_;
|
| -
|
| - const std::string scheme_;
|
| - const std::string hostname_;
|
| - scoped_refptr<base::SequencedTaskRunner> io_task_runner_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(Delegate);
|
| -};
|
| -
|
| -URLRequestPostInterceptorFactory::URLRequestPostInterceptorFactory(
|
| - const std::string& scheme,
|
| - const std::string& hostname,
|
| - const scoped_refptr<base::SequencedTaskRunner>& io_task_runner)
|
| - : scheme_(scheme),
|
| - hostname_(hostname),
|
| - io_task_runner_(io_task_runner),
|
| - delegate_(new URLRequestPostInterceptor::Delegate(scheme,
|
| - hostname,
|
| - io_task_runner)) {
|
| - io_task_runner_->PostTask(
|
| - FROM_HERE,
|
| - base::Bind(&URLRequestPostInterceptor::Delegate::Register,
|
| - base::Unretained(delegate_)));
|
| -}
|
| -
|
| -URLRequestPostInterceptorFactory::~URLRequestPostInterceptorFactory() {
|
| - io_task_runner_->PostTask(
|
| - FROM_HERE,
|
| - base::Bind(&URLRequestPostInterceptor::Delegate::Unregister,
|
| - base::Unretained(delegate_)));
|
| -}
|
| -
|
| -URLRequestPostInterceptor* URLRequestPostInterceptorFactory::CreateInterceptor(
|
| - const base::FilePath& filepath) {
|
| - const GURL base_url(
|
| - base::StringPrintf("%s://%s", scheme_.c_str(), hostname_.c_str()));
|
| - GURL absolute_url(base_url.Resolve(filepath.MaybeAsASCII()));
|
| - URLRequestPostInterceptor* interceptor(
|
| - new URLRequestPostInterceptor(absolute_url, io_task_runner_));
|
| - bool res = io_task_runner_->PostTask(
|
| - FROM_HERE,
|
| - base::Bind(&URLRequestPostInterceptor::Delegate::OnCreateInterceptor,
|
| - base::Unretained(delegate_),
|
| - base::Unretained(interceptor)));
|
| - if (!res) {
|
| - delete interceptor;
|
| - return NULL;
|
| - }
|
| -
|
| - return interceptor;
|
| -}
|
| -
|
| -bool PartialMatch::Match(const std::string& actual) const {
|
| - return actual.find(expected_) != std::string::npos;
|
| -}
|
| -
|
| -InterceptorFactory::InterceptorFactory(
|
| - const scoped_refptr<base::SequencedTaskRunner>& io_task_runner)
|
| - : URLRequestPostInterceptorFactory(POST_INTERCEPT_SCHEME,
|
| - POST_INTERCEPT_HOSTNAME,
|
| - io_task_runner) {
|
| -}
|
| -
|
| -InterceptorFactory::~InterceptorFactory() {
|
| -}
|
| -
|
| -URLRequestPostInterceptor* InterceptorFactory::CreateInterceptor() {
|
| - return CreateInterceptorForPath(POST_INTERCEPT_PATH);
|
| -}
|
| -
|
| -URLRequestPostInterceptor* InterceptorFactory::CreateInterceptorForPath(
|
| - const char* url_path) {
|
| - return URLRequestPostInterceptorFactory::CreateInterceptor(
|
| - base::FilePath::FromUTF8Unsafe(url_path));
|
| -}
|
| -
|
| -} // namespace component_updater
|
|
|