| Index: net/proxy/dhcp_proxy_script_adapter_fetcher_win_unittest.cc
|
| ===================================================================
|
| --- net/proxy/dhcp_proxy_script_adapter_fetcher_win_unittest.cc (revision 85647)
|
| +++ net/proxy/dhcp_proxy_script_adapter_fetcher_win_unittest.cc (working copy)
|
| @@ -1,300 +0,0 @@
|
| -// Copyright (c) 2011 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 "net/proxy/dhcp_proxy_script_adapter_fetcher_win.h"
|
| -
|
| -#include "base/perftimer.h"
|
| -#include "base/synchronization/waitable_event.h"
|
| -#include "base/timer.h"
|
| -#include "net/base/net_errors.h"
|
| -#include "net/base/test_completion_callback.h"
|
| -#include "net/proxy/mock_proxy_script_fetcher.h"
|
| -#include "net/proxy/proxy_script_fetcher_impl.h"
|
| -#include "net/test/test_server.h"
|
| -#include "net/url_request/url_request_test_util.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -namespace net {
|
| -
|
| -namespace {
|
| -
|
| -const char* const kPacUrl = "http://pacserver/script.pac";
|
| -
|
| -// In net/proxy/dhcp_proxy_script_fetcher_win_unittest.cc there are a few
|
| -// tests that exercise DhcpProxyScriptAdapterFetcher end-to-end along with
|
| -// DhcpProxyScriptFetcherWin, i.e. it tests the end-to-end usage of Win32
|
| -// APIs and the network. In this file we test only by stubbing out
|
| -// functionality.
|
| -
|
| -// Version of DhcpProxyScriptAdapterFetcher that mocks out dependencies
|
| -// to allow unit testing.
|
| -class MockDhcpProxyScriptAdapterFetcher
|
| - : public DhcpProxyScriptAdapterFetcher {
|
| - public:
|
| - explicit MockDhcpProxyScriptAdapterFetcher(URLRequestContext* context)
|
| - : DhcpProxyScriptAdapterFetcher(context),
|
| - dhcp_delay_ms_(1),
|
| - timeout_ms_(100),
|
| - configured_url_(kPacUrl),
|
| - fetcher_delay_ms_(1),
|
| - fetcher_result_(OK),
|
| - pac_script_("bingo") {
|
| - }
|
| -
|
| - void Cancel() {
|
| - DhcpProxyScriptAdapterFetcher::Cancel();
|
| - fetcher_ = NULL;
|
| - }
|
| -
|
| - virtual ProxyScriptFetcher* ImplCreateScriptFetcher() OVERRIDE {
|
| - // We don't maintain ownership of the fetcher, it is transferred to
|
| - // the caller.
|
| - fetcher_ = new MockProxyScriptFetcher();
|
| - if (fetcher_delay_ms_ != -1) {
|
| - fetcher_timer_.Start(
|
| - base::TimeDelta::FromMilliseconds(fetcher_delay_ms_),
|
| - this, &MockDhcpProxyScriptAdapterFetcher::OnFetcherTimer);
|
| - }
|
| - return fetcher_;
|
| - }
|
| -
|
| - class DelayingWorkerThread : public WorkerThread {
|
| - public:
|
| - explicit DelayingWorkerThread(
|
| - const base::WeakPtr<DhcpProxyScriptAdapterFetcher>& owner)
|
| - : WorkerThread(owner),
|
| - test_finished_event_(true, false) {
|
| - }
|
| -
|
| - std::string ImplGetPacURLFromDhcp(
|
| - const std::string& adapter_name) OVERRIDE {
|
| - PerfTimer timer;
|
| - test_finished_event_.TimedWait(dhcp_delay_);
|
| - return configured_url_;
|
| - }
|
| -
|
| - base::WaitableEvent test_finished_event_;
|
| - TimeDelta dhcp_delay_;
|
| - std::string configured_url_;
|
| - };
|
| -
|
| - virtual WorkerThread* ImplCreateWorkerThread(
|
| - const base::WeakPtr<DhcpProxyScriptAdapterFetcher>& owner) OVERRIDE {
|
| - worker_thread_ = new DelayingWorkerThread(owner);
|
| - worker_thread_->dhcp_delay_ = TimeDelta::FromMilliseconds(dhcp_delay_ms_);
|
| - worker_thread_->configured_url_ = configured_url_;
|
| - return worker_thread_;
|
| - }
|
| -
|
| - // Use a shorter timeout so tests can finish more quickly.
|
| - virtual base::TimeDelta ImplGetTimeout() const OVERRIDE {
|
| - return base::TimeDelta::FromMilliseconds(timeout_ms_);
|
| - }
|
| -
|
| - void OnFetcherTimer() {
|
| - // Note that there is an assumption by this mock implementation that
|
| - // DhcpProxyScriptAdapterFetcher::Fetch will call ImplCreateScriptFetcher
|
| - // and call Fetch on the fetcher before the message loop is re-entered.
|
| - // This holds true today, but if you hit this DCHECK the problem can
|
| - // possibly be resolved by having a separate subclass of
|
| - // MockProxyScriptFetcher that adds the delay internally (instead of
|
| - // the simple approach currently used in ImplCreateScriptFetcher above).
|
| - DCHECK(fetcher_ && fetcher_->has_pending_request());
|
| - fetcher_->NotifyFetchCompletion(fetcher_result_, pac_script_);
|
| - fetcher_ = NULL;
|
| - }
|
| -
|
| - bool IsWaitingForFetcher() const {
|
| - return state_ == STATE_WAIT_URL;
|
| - }
|
| -
|
| - bool WasCancelled() const {
|
| - return state_ == STATE_CANCEL;
|
| - }
|
| -
|
| - void FinishTest() {
|
| - DCHECK(worker_thread_);
|
| - worker_thread_->test_finished_event_.Signal();
|
| - }
|
| -
|
| - int dhcp_delay_ms_;
|
| - int timeout_ms_;
|
| - std::string configured_url_;
|
| - int fetcher_delay_ms_;
|
| - int fetcher_result_;
|
| - std::string pac_script_;
|
| - MockProxyScriptFetcher* fetcher_;
|
| - base::OneShotTimer<MockDhcpProxyScriptAdapterFetcher> fetcher_timer_;
|
| - scoped_refptr<DelayingWorkerThread> worker_thread_;
|
| -};
|
| -
|
| -class FetcherClient {
|
| - public:
|
| - FetcherClient()
|
| - : url_request_context_(new TestURLRequestContext()),
|
| - fetcher_(
|
| - new MockDhcpProxyScriptAdapterFetcher(url_request_context_.get())) {
|
| - }
|
| -
|
| - void WaitForResult(int expected_error) {
|
| - EXPECT_EQ(expected_error, callback_.WaitForResult());
|
| - }
|
| -
|
| - void RunTest() {
|
| - fetcher_->Fetch("adapter name", &callback_);
|
| - }
|
| -
|
| - void FinishTestAllowCleanup() {
|
| - fetcher_->FinishTest();
|
| - MessageLoop::current()->RunAllPending();
|
| - }
|
| -
|
| - TestCompletionCallback callback_;
|
| - scoped_refptr<URLRequestContext> url_request_context_;
|
| - scoped_ptr<MockDhcpProxyScriptAdapterFetcher> fetcher_;
|
| - string16 pac_text_;
|
| -};
|
| -
|
| -TEST(DhcpProxyScriptAdapterFetcher, NormalCaseURLNotInDhcp) {
|
| - FetcherClient client;
|
| - client.fetcher_->configured_url_ = "";
|
| - client.RunTest();
|
| - client.WaitForResult(ERR_PAC_NOT_IN_DHCP);
|
| - ASSERT_TRUE(client.fetcher_->DidFinish());
|
| - EXPECT_EQ(ERR_PAC_NOT_IN_DHCP, client.fetcher_->GetResult());
|
| - EXPECT_EQ(string16(L""), client.fetcher_->GetPacScript());
|
| -}
|
| -
|
| -TEST(DhcpProxyScriptAdapterFetcher, NormalCaseURLInDhcp) {
|
| - FetcherClient client;
|
| - client.RunTest();
|
| - client.WaitForResult(OK);
|
| - ASSERT_TRUE(client.fetcher_->DidFinish());
|
| - EXPECT_EQ(OK, client.fetcher_->GetResult());
|
| - EXPECT_EQ(string16(L"bingo"), client.fetcher_->GetPacScript());
|
| - EXPECT_EQ(GURL(kPacUrl), client.fetcher_->GetPacURL());
|
| -}
|
| -
|
| -TEST(DhcpProxyScriptAdapterFetcher, TimeoutDuringDhcp) {
|
| - // Does a Fetch() with a long enough delay on accessing DHCP that the
|
| - // fetcher should time out. This is to test a case manual testing found,
|
| - // where under certain circumstances (e.g. adapter enabled for DHCP and
|
| - // needs to retrieve its configuration from DHCP, but no DHCP server
|
| - // present on the network) accessing DHCP can take on the order of tens
|
| - // of seconds.
|
| - FetcherClient client;
|
| - client.fetcher_->dhcp_delay_ms_ = 20 * 1000;
|
| - client.fetcher_->timeout_ms_ = 25;
|
| -
|
| - PerfTimer timer;
|
| - client.RunTest();
|
| - client.WaitForResult(ERR_TIMED_OUT);
|
| -
|
| - // The timeout should occur within about 25 ms, way before the 20s set as
|
| - // the API delay above.
|
| - ASSERT_GT(base::TimeDelta::FromMilliseconds(35), timer.Elapsed());
|
| - ASSERT_TRUE(client.fetcher_->DidFinish());
|
| - EXPECT_EQ(ERR_TIMED_OUT, client.fetcher_->GetResult());
|
| - EXPECT_EQ(string16(L""), client.fetcher_->GetPacScript());
|
| - EXPECT_EQ(GURL(), client.fetcher_->GetPacURL());
|
| - client.FinishTestAllowCleanup();
|
| -}
|
| -
|
| -TEST(DhcpProxyScriptAdapterFetcher, CancelWhileDhcp) {
|
| - FetcherClient client;
|
| - client.fetcher_->dhcp_delay_ms_ = 10;
|
| - client.RunTest();
|
| - client.fetcher_->Cancel();
|
| - MessageLoop::current()->RunAllPending();
|
| - ASSERT_FALSE(client.fetcher_->DidFinish());
|
| - ASSERT_TRUE(client.fetcher_->WasCancelled());
|
| - EXPECT_EQ(ERR_ABORTED, client.fetcher_->GetResult());
|
| - EXPECT_EQ(string16(L""), client.fetcher_->GetPacScript());
|
| - EXPECT_EQ(GURL(), client.fetcher_->GetPacURL());
|
| - client.FinishTestAllowCleanup();
|
| -}
|
| -
|
| -TEST(DhcpProxyScriptAdapterFetcher, CancelWhileFetcher) {
|
| - FetcherClient client;
|
| - // This causes the mock fetcher not to pretend the
|
| - // fetcher finishes after a timeout.
|
| - client.fetcher_->fetcher_delay_ms_ = -1;
|
| - client.RunTest();
|
| - int max_loops = 4;
|
| - while (!client.fetcher_->IsWaitingForFetcher() && max_loops--) {
|
| - base::PlatformThread::Sleep(10);
|
| - MessageLoop::current()->RunAllPending();
|
| - }
|
| - client.fetcher_->Cancel();
|
| - MessageLoop::current()->RunAllPending();
|
| - ASSERT_FALSE(client.fetcher_->DidFinish());
|
| - ASSERT_TRUE(client.fetcher_->WasCancelled());
|
| - EXPECT_EQ(ERR_ABORTED, client.fetcher_->GetResult());
|
| - EXPECT_EQ(string16(L""), client.fetcher_->GetPacScript());
|
| - // GetPacURL() still returns the URL fetched in this case.
|
| - EXPECT_EQ(GURL(kPacUrl), client.fetcher_->GetPacURL());
|
| - client.FinishTestAllowCleanup();
|
| -}
|
| -
|
| -TEST(DhcpProxyScriptAdapterFetcher, CancelAtCompletion) {
|
| - FetcherClient client;
|
| - client.RunTest();
|
| - client.WaitForResult(OK);
|
| - client.fetcher_->Cancel();
|
| - // Canceling after you're done should have no effect, so these
|
| - // are identical expectations to the NormalCaseURLInDhcp test.
|
| - ASSERT_TRUE(client.fetcher_->DidFinish());
|
| - EXPECT_EQ(OK, client.fetcher_->GetResult());
|
| - EXPECT_EQ(string16(L"bingo"), client.fetcher_->GetPacScript());
|
| - EXPECT_EQ(GURL(kPacUrl), client.fetcher_->GetPacURL());
|
| - client.FinishTestAllowCleanup();
|
| -}
|
| -
|
| -// Does a real fetch on a mock DHCP configuration.
|
| -class MockDhcpRealFetchProxyScriptAdapterFetcher
|
| - : public MockDhcpProxyScriptAdapterFetcher {
|
| - public:
|
| - explicit MockDhcpRealFetchProxyScriptAdapterFetcher(
|
| - URLRequestContext* context)
|
| - : MockDhcpProxyScriptAdapterFetcher(context),
|
| - url_request_context_(context) {
|
| - }
|
| -
|
| - // Returns a real proxy script fetcher.
|
| - ProxyScriptFetcher* ImplCreateScriptFetcher() OVERRIDE {
|
| - ProxyScriptFetcher* fetcher =
|
| - new ProxyScriptFetcherImpl(url_request_context_);
|
| - return fetcher;
|
| - }
|
| -
|
| - URLRequestContext* url_request_context_;
|
| -};
|
| -
|
| -TEST(DhcpProxyScriptAdapterFetcher, MockDhcpRealFetch) {
|
| - TestServer test_server(
|
| - TestServer::TYPE_HTTP,
|
| - FilePath(FILE_PATH_LITERAL("net/data/proxy_script_fetcher_unittest")));
|
| - ASSERT_TRUE(test_server.Start());
|
| -
|
| - GURL configured_url = test_server.GetURL("files/downloadable.pac");
|
| -
|
| - FetcherClient client;
|
| - scoped_refptr<URLRequestContext> url_request_context(
|
| - new TestURLRequestContext());
|
| - client.fetcher_.reset(
|
| - new MockDhcpRealFetchProxyScriptAdapterFetcher(
|
| - url_request_context.get()));
|
| - client.fetcher_->configured_url_ = configured_url.spec();
|
| - client.RunTest();
|
| - client.WaitForResult(OK);
|
| - ASSERT_TRUE(client.fetcher_->DidFinish());
|
| - EXPECT_EQ(OK, client.fetcher_->GetResult());
|
| - EXPECT_EQ(string16(L"-downloadable.pac-\n"), client.fetcher_->GetPacScript());
|
| - EXPECT_EQ(configured_url,
|
| - client.fetcher_->GetPacURL());
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -} // namespace net
|
|
|