diff options
Diffstat (limited to 'chrome/browser/extensions/autoupdate_interceptor.cc')
-rw-r--r-- | chrome/browser/extensions/autoupdate_interceptor.cc | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/chrome/browser/extensions/autoupdate_interceptor.cc b/chrome/browser/extensions/autoupdate_interceptor.cc new file mode 100644 index 0000000..b908133 --- /dev/null +++ b/chrome/browser/extensions/autoupdate_interceptor.cc @@ -0,0 +1,75 @@ +// Copyright (c) 2009 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/extensions/autoupdate_interceptor.h" + +#include "base/file_util.h" +#include "chrome/browser/chrome_thread.h" +#include "net/url_request/url_request_test_job.h" +#include "testing/gtest/include/gtest/gtest.h" + +// This is a specialized version of URLRequestTestJob that lets us specify +// response data and make sure the response code is 200, which the autoupdate +// code relies on. +class AutoUpdateTestRequestJob : public URLRequestTestJob { + public: + AutoUpdateTestRequestJob(URLRequest* request, + const std::string& response_data) : URLRequestTestJob( + request, URLRequestTestJob::test_headers(), response_data, true) {} + virtual int GetResponseCode() const { return 200; } +}; + + +AutoUpdateInterceptor::AutoUpdateInterceptor() { + URLRequest::RegisterRequestInterceptor(this); +} + +AutoUpdateInterceptor::~AutoUpdateInterceptor() { + URLRequest::UnregisterRequestInterceptor(this); +} + + +URLRequestJob* AutoUpdateInterceptor::MaybeIntercept(URLRequest* request) { + EXPECT_TRUE(ChromeThread::CurrentlyOn(ChromeThread::IO)); + if (request->url().scheme() != "http" || + request->url().host() != "localhost") { + return NULL; + } + + // Search for this request's url, ignoring any query parameters. + GURL url = request->url(); + if (url.has_query()) { + GURL::Replacements replacements; + replacements.ClearQuery(); + url = url.ReplaceComponents(replacements); + } + std::map<GURL, FilePath>::iterator i = responses_.find(url); + if (i == responses_.end()) { + return NULL; + } + std::string contents; + EXPECT_TRUE(file_util::ReadFileToString(i->second, &contents)); + + return new AutoUpdateTestRequestJob(request, contents); +} + + +void AutoUpdateInterceptor::SetResponse(const std::string url, + const FilePath& path) { + EXPECT_TRUE(ChromeThread::CurrentlyOn(ChromeThread::IO)); + GURL gurl(url); + EXPECT_EQ("http", gurl.scheme()); + EXPECT_EQ("localhost", gurl.host()); + EXPECT_TRUE(file_util::PathExists(path)); + responses_[gurl] = path; +} + + +void AutoUpdateInterceptor::SetResponseOnIOThread(const std::string url, + const FilePath& path) { + MessageLoop* io_loop = ChromeThread::GetMessageLoop(ChromeThread::IO); + io_loop->PostTask(FROM_HERE, + NewRunnableMethod(this, &AutoUpdateInterceptor::SetResponse, + url, path)); +} |