| Index: net/proxy/init_proxy_resolver_unittest.cc
|
| ===================================================================
|
| --- net/proxy/init_proxy_resolver_unittest.cc (revision 113525)
|
| +++ net/proxy/init_proxy_resolver_unittest.cc (working copy)
|
| @@ -1,715 +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 <vector>
|
| -
|
| -#include "base/message_loop.h"
|
| -#include "base/string_util.h"
|
| -#include "base/time.h"
|
| -#include "base/utf_string_conversions.h"
|
| -#include "net/base/net_errors.h"
|
| -#include "net/base/net_log.h"
|
| -#include "net/base/net_log_unittest.h"
|
| -#include "net/base/test_completion_callback.h"
|
| -#include "net/proxy/init_proxy_resolver.h"
|
| -#include "net/proxy/dhcp_proxy_script_fetcher.h"
|
| -#include "net/proxy/proxy_config.h"
|
| -#include "net/proxy/proxy_resolver.h"
|
| -#include "net/proxy/proxy_script_fetcher.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -namespace net {
|
| -namespace {
|
| -
|
| -enum Error {
|
| - kFailedDownloading = -100,
|
| - kFailedParsing = -200,
|
| -};
|
| -
|
| -class Rules {
|
| - public:
|
| - struct Rule {
|
| - Rule(const GURL& url,
|
| - int fetch_error,
|
| - int set_pac_error)
|
| - : url(url),
|
| - fetch_error(fetch_error),
|
| - set_pac_error(set_pac_error) {
|
| - }
|
| -
|
| - string16 text() const {
|
| - if (set_pac_error == OK)
|
| - return UTF8ToUTF16(url.spec() + "!valid-script");
|
| - if (fetch_error == OK)
|
| - return UTF8ToUTF16(url.spec() + "!invalid-script");
|
| - return string16();
|
| - }
|
| -
|
| - GURL url;
|
| - int fetch_error;
|
| - int set_pac_error;
|
| - };
|
| -
|
| - Rule AddSuccessRule(const char* url) {
|
| - Rule rule(GURL(url), OK /*fetch_error*/, OK /*set_pac_error*/);
|
| - rules_.push_back(rule);
|
| - return rule;
|
| - }
|
| -
|
| - void AddFailDownloadRule(const char* url) {
|
| - rules_.push_back(Rule(GURL(url), kFailedDownloading /*fetch_error*/,
|
| - ERR_UNEXPECTED /*set_pac_error*/));
|
| - }
|
| -
|
| - void AddFailParsingRule(const char* url) {
|
| - rules_.push_back(Rule(GURL(url), OK /*fetch_error*/,
|
| - kFailedParsing /*set_pac_error*/));
|
| - }
|
| -
|
| - const Rule& GetRuleByUrl(const GURL& url) const {
|
| - for (RuleList::const_iterator it = rules_.begin(); it != rules_.end();
|
| - ++it) {
|
| - if (it->url == url)
|
| - return *it;
|
| - }
|
| - LOG(FATAL) << "Rule not found for " << url;
|
| - return rules_[0];
|
| - }
|
| -
|
| - const Rule& GetRuleByText(const string16& text) const {
|
| - for (RuleList::const_iterator it = rules_.begin(); it != rules_.end();
|
| - ++it) {
|
| - if (it->text() == text)
|
| - return *it;
|
| - }
|
| - LOG(FATAL) << "Rule not found for " << text;
|
| - return rules_[0];
|
| - }
|
| -
|
| - private:
|
| - typedef std::vector<Rule> RuleList;
|
| - RuleList rules_;
|
| -};
|
| -
|
| -class RuleBasedProxyScriptFetcher : public ProxyScriptFetcher {
|
| - public:
|
| - explicit RuleBasedProxyScriptFetcher(const Rules* rules) : rules_(rules) {}
|
| -
|
| - // ProxyScriptFetcher implementation.
|
| - virtual int Fetch(const GURL& url,
|
| - string16* text,
|
| - OldCompletionCallback* callback) {
|
| - const Rules::Rule& rule = rules_->GetRuleByUrl(url);
|
| - int rv = rule.fetch_error;
|
| - EXPECT_NE(ERR_UNEXPECTED, rv);
|
| - if (rv == OK)
|
| - *text = rule.text();
|
| - return rv;
|
| - }
|
| -
|
| - virtual void Cancel() {}
|
| -
|
| - virtual URLRequestContext* GetRequestContext() const { return NULL; }
|
| -
|
| - private:
|
| - const Rules* rules_;
|
| -};
|
| -
|
| -class RuleBasedProxyResolver : public ProxyResolver {
|
| - public:
|
| - RuleBasedProxyResolver(const Rules* rules, bool expects_pac_bytes)
|
| - : ProxyResolver(expects_pac_bytes), rules_(rules) {}
|
| -
|
| - // ProxyResolver implementation:
|
| - virtual int GetProxyForURL(const GURL& /*url*/,
|
| - ProxyInfo* /*results*/,
|
| - OldCompletionCallback* /*callback*/,
|
| - RequestHandle* /*request_handle*/,
|
| - const BoundNetLog& /*net_log*/) {
|
| - NOTREACHED();
|
| - return ERR_UNEXPECTED;
|
| - }
|
| -
|
| - virtual void CancelRequest(RequestHandle request_handle) {
|
| - NOTREACHED();
|
| - }
|
| -
|
| - virtual LoadState GetLoadState(RequestHandle request_handle) const {
|
| - NOTREACHED();
|
| - return LOAD_STATE_IDLE;
|
| - }
|
| -
|
| - virtual LoadState GetLoadStateThreadSafe(
|
| - RequestHandle request) const OVERRIDE {
|
| - NOTREACHED();
|
| - return LOAD_STATE_IDLE;
|
| - }
|
| -
|
| - virtual void CancelSetPacScript() {
|
| - NOTREACHED();
|
| - }
|
| -
|
| - virtual int SetPacScript(
|
| - const scoped_refptr<ProxyResolverScriptData>& script_data,
|
| - OldCompletionCallback* callback) {
|
| -
|
| - const GURL url =
|
| - script_data->type() == ProxyResolverScriptData::TYPE_SCRIPT_URL ?
|
| - script_data->url() : GURL();
|
| -
|
| - const Rules::Rule& rule = expects_pac_bytes() ?
|
| - rules_->GetRuleByText(script_data->utf16()) :
|
| - rules_->GetRuleByUrl(url);
|
| -
|
| - int rv = rule.set_pac_error;
|
| - EXPECT_NE(ERR_UNEXPECTED, rv);
|
| -
|
| - if (expects_pac_bytes()) {
|
| - EXPECT_EQ(rule.text(), script_data->utf16());
|
| - } else {
|
| - EXPECT_EQ(rule.url, url);
|
| - }
|
| -
|
| - if (rv == OK)
|
| - script_data_ = script_data;
|
| - return rv;
|
| - }
|
| -
|
| - const ProxyResolverScriptData* script_data() const { return script_data_; }
|
| -
|
| - private:
|
| - const Rules* rules_;
|
| - scoped_refptr<ProxyResolverScriptData> script_data_;
|
| -};
|
| -
|
| -// Succeed using custom PAC script.
|
| -TEST(InitProxyResolverTest, CustomPacSucceeds) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| -
|
| - Rules::Rule rule = rules.AddSuccessRule("http://custom/proxy.pac");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - CapturingNetLog log(CapturingNetLog::kUnbounded);
|
| - ProxyConfig effective_config;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, &log);
|
| - EXPECT_EQ(OK, init.Init(
|
| - config, base::TimeDelta(), &effective_config, &callback));
|
| - EXPECT_EQ(rule.text(), resolver.script_data()->utf16());
|
| -
|
| - // Check the NetLog was filled correctly.
|
| - CapturingNetLog::EntryList entries;
|
| - log.GetEntries(&entries);
|
| -
|
| - EXPECT_EQ(6u, entries.size());
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 0, NetLog::TYPE_INIT_PROXY_RESOLVER));
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 1, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 2, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 3, NetLog::TYPE_INIT_PROXY_RESOLVER_SET_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 4, NetLog::TYPE_INIT_PROXY_RESOLVER_SET_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 5, NetLog::TYPE_INIT_PROXY_RESOLVER));
|
| -
|
| - EXPECT_TRUE(effective_config.has_pac_url());
|
| - EXPECT_EQ(config.pac_url(), effective_config.pac_url());
|
| -}
|
| -
|
| -// Fail downloading the custom PAC script.
|
| -TEST(InitProxyResolverTest, CustomPacFails1) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| -
|
| - rules.AddFailDownloadRule("http://custom/proxy.pac");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - CapturingNetLog log(CapturingNetLog::kUnbounded);
|
| - ProxyConfig effective_config;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, &log);
|
| - EXPECT_EQ(kFailedDownloading,
|
| - init.Init(config, base::TimeDelta(), &effective_config, &callback));
|
| - EXPECT_EQ(NULL, resolver.script_data());
|
| -
|
| - // Check the NetLog was filled correctly.
|
| - CapturingNetLog::EntryList entries;
|
| - log.GetEntries(&entries);
|
| -
|
| - EXPECT_EQ(4u, entries.size());
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 0, NetLog::TYPE_INIT_PROXY_RESOLVER));
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 1, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 2, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 3, NetLog::TYPE_INIT_PROXY_RESOLVER));
|
| -
|
| - EXPECT_FALSE(effective_config.has_pac_url());
|
| -}
|
| -
|
| -// Fail parsing the custom PAC script.
|
| -TEST(InitProxyResolverTest, CustomPacFails2) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| -
|
| - rules.AddFailParsingRule("http://custom/proxy.pac");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, NULL);
|
| - EXPECT_EQ(kFailedParsing,
|
| - init.Init(config, base::TimeDelta(), NULL, &callback));
|
| - EXPECT_EQ(NULL, resolver.script_data());
|
| -}
|
| -
|
| -// Fail downloading the custom PAC script, because the fetcher was NULL.
|
| -TEST(InitProxyResolverTest, HasNullProxyScriptFetcher) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| -
|
| - TestOldCompletionCallback callback;
|
| - InitProxyResolver init(&resolver, NULL, &dhcp_fetcher, NULL);
|
| - EXPECT_EQ(ERR_UNEXPECTED,
|
| - init.Init(config, base::TimeDelta(), NULL, &callback));
|
| - EXPECT_EQ(NULL, resolver.script_data());
|
| -}
|
| -
|
| -// Succeeds in choosing autodetect (WPAD DNS).
|
| -TEST(InitProxyResolverTest, AutodetectSuccess) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_auto_detect(true);
|
| -
|
| - Rules::Rule rule = rules.AddSuccessRule("http://wpad/wpad.dat");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - ProxyConfig effective_config;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, NULL);
|
| - EXPECT_EQ(OK, init.Init(
|
| - config, base::TimeDelta(), &effective_config, &callback));
|
| - EXPECT_EQ(rule.text(), resolver.script_data()->utf16());
|
| -
|
| - EXPECT_TRUE(effective_config.has_pac_url());
|
| - EXPECT_EQ(rule.url, effective_config.pac_url());
|
| -}
|
| -
|
| -// Fails at WPAD (downloading), but succeeds in choosing the custom PAC.
|
| -TEST(InitProxyResolverTest, AutodetectFailCustomSuccess1) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_auto_detect(true);
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| -
|
| - rules.AddFailDownloadRule("http://wpad/wpad.dat");
|
| - Rules::Rule rule = rules.AddSuccessRule("http://custom/proxy.pac");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - ProxyConfig effective_config;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, NULL);
|
| - EXPECT_EQ(OK, init.Init(
|
| - config, base::TimeDelta(), &effective_config, &callback));
|
| - EXPECT_EQ(rule.text(), resolver.script_data()->utf16());
|
| -
|
| - EXPECT_TRUE(effective_config.has_pac_url());
|
| - EXPECT_EQ(rule.url, effective_config.pac_url());
|
| -}
|
| -
|
| -// Fails at WPAD (no DHCP config, DNS PAC fails parsing), but succeeds in
|
| -// choosing the custom PAC.
|
| -TEST(InitProxyResolverTest, AutodetectFailCustomSuccess2) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_auto_detect(true);
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| - config.proxy_rules().ParseFromString("unused-manual-proxy:99");
|
| -
|
| - rules.AddFailParsingRule("http://wpad/wpad.dat");
|
| - Rules::Rule rule = rules.AddSuccessRule("http://custom/proxy.pac");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - CapturingNetLog log(CapturingNetLog::kUnbounded);
|
| -
|
| - ProxyConfig effective_config;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, &log);
|
| - EXPECT_EQ(OK, init.Init(config, base::TimeDelta(),
|
| - &effective_config, &callback));
|
| - EXPECT_EQ(rule.text(), resolver.script_data()->utf16());
|
| -
|
| - // Verify that the effective configuration no longer contains auto detect or
|
| - // any of the manual settings.
|
| - EXPECT_TRUE(effective_config.Equals(
|
| - ProxyConfig::CreateFromCustomPacURL(GURL("http://custom/proxy.pac"))));
|
| -
|
| - // Check the NetLog was filled correctly.
|
| - // (Note that various states are repeated since both WPAD and custom
|
| - // PAC scripts are tried).
|
| - CapturingNetLog::EntryList entries;
|
| - log.GetEntries(&entries);
|
| -
|
| - EXPECT_EQ(14u, entries.size());
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 0, NetLog::TYPE_INIT_PROXY_RESOLVER));
|
| - // This is the DHCP phase, which fails fetching rather than parsing, so
|
| - // there is no pair of SET_PAC_SCRIPT events.
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 1, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 2, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEvent(
|
| - entries, 3,
|
| - NetLog::TYPE_INIT_PROXY_RESOLVER_FALLING_BACK_TO_NEXT_PAC_SOURCE,
|
| - NetLog::PHASE_NONE));
|
| - // This is the DNS phase, which attempts a fetch but fails.
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 4, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 5, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 6, NetLog::TYPE_INIT_PROXY_RESOLVER_SET_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 7, NetLog::TYPE_INIT_PROXY_RESOLVER_SET_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEvent(
|
| - entries, 8,
|
| - NetLog::TYPE_INIT_PROXY_RESOLVER_FALLING_BACK_TO_NEXT_PAC_SOURCE,
|
| - NetLog::PHASE_NONE));
|
| - // Finally, the custom PAC URL phase.
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 9, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 10, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 11, NetLog::TYPE_INIT_PROXY_RESOLVER_SET_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 12, NetLog::TYPE_INIT_PROXY_RESOLVER_SET_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 13, NetLog::TYPE_INIT_PROXY_RESOLVER));
|
| -}
|
| -
|
| -// Fails at WPAD (downloading), and fails at custom PAC (downloading).
|
| -TEST(InitProxyResolverTest, AutodetectFailCustomFails1) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_auto_detect(true);
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| -
|
| - rules.AddFailDownloadRule("http://wpad/wpad.dat");
|
| - rules.AddFailDownloadRule("http://custom/proxy.pac");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, NULL);
|
| - EXPECT_EQ(kFailedDownloading,
|
| - init.Init(config, base::TimeDelta(), NULL, &callback));
|
| - EXPECT_EQ(NULL, resolver.script_data());
|
| -}
|
| -
|
| -// Fails at WPAD (downloading), and fails at custom PAC (parsing).
|
| -TEST(InitProxyResolverTest, AutodetectFailCustomFails2) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_auto_detect(true);
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| -
|
| - rules.AddFailDownloadRule("http://wpad/wpad.dat");
|
| - rules.AddFailParsingRule("http://custom/proxy.pac");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, NULL);
|
| - EXPECT_EQ(kFailedParsing,
|
| - init.Init(config, base::TimeDelta(), NULL, &callback));
|
| - EXPECT_EQ(NULL, resolver.script_data());
|
| -}
|
| -
|
| -// Fails at WPAD (parsing), but succeeds in choosing the custom PAC.
|
| -// This is the same as AutodetectFailCustomSuccess2, but using a ProxyResolver
|
| -// that doesn't |expects_pac_bytes|.
|
| -TEST(InitProxyResolverTest, AutodetectFailCustomSuccess2_NoFetch) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, false /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_auto_detect(true);
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| -
|
| - rules.AddFailParsingRule(""); // Autodetect.
|
| - Rules::Rule rule = rules.AddSuccessRule("http://custom/proxy.pac");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, NULL);
|
| - EXPECT_EQ(OK, init.Init(config, base::TimeDelta(), NULL, &callback));
|
| - EXPECT_EQ(rule.url, resolver.script_data()->url());
|
| -}
|
| -
|
| -// This is a copy-paste of CustomPacFails1, with the exception that we give it
|
| -// a 1 millisecond delay. This means it will now complete asynchronously.
|
| -// Moreover, we test the NetLog to make sure it logged the pause.
|
| -TEST(InitProxyResolverTest, CustomPacFails1_WithPositiveDelay) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| -
|
| - rules.AddFailDownloadRule("http://custom/proxy.pac");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - CapturingNetLog log(CapturingNetLog::kUnbounded);
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, &log);
|
| - EXPECT_EQ(ERR_IO_PENDING,
|
| - init.Init(config, base::TimeDelta::FromMilliseconds(1),
|
| - NULL, &callback));
|
| -
|
| - EXPECT_EQ(kFailedDownloading, callback.WaitForResult());
|
| - EXPECT_EQ(NULL, resolver.script_data());
|
| -
|
| - // Check the NetLog was filled correctly.
|
| - CapturingNetLog::EntryList entries;
|
| - log.GetEntries(&entries);
|
| -
|
| - EXPECT_EQ(6u, entries.size());
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 0, NetLog::TYPE_INIT_PROXY_RESOLVER));
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 1, NetLog::TYPE_INIT_PROXY_RESOLVER_WAIT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 2, NetLog::TYPE_INIT_PROXY_RESOLVER_WAIT));
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 3, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 4, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 5, NetLog::TYPE_INIT_PROXY_RESOLVER));
|
| -}
|
| -
|
| -// This is a copy-paste of CustomPacFails1, with the exception that we give it
|
| -// a -5 second delay instead of a 0 ms delay. This change should have no effect
|
| -// so the rest of the test is unchanged.
|
| -TEST(InitProxyResolverTest, CustomPacFails1_WithNegativeDelay) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - DoNothingDhcpProxyScriptFetcher dhcp_fetcher;
|
| -
|
| - ProxyConfig config;
|
| - config.set_pac_url(GURL("http://custom/proxy.pac"));
|
| -
|
| - rules.AddFailDownloadRule("http://custom/proxy.pac");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - CapturingNetLog log(CapturingNetLog::kUnbounded);
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, &log);
|
| - EXPECT_EQ(kFailedDownloading,
|
| - init.Init(config, base::TimeDelta::FromSeconds(-5),
|
| - NULL, &callback));
|
| - EXPECT_EQ(NULL, resolver.script_data());
|
| -
|
| - // Check the NetLog was filled correctly.
|
| - CapturingNetLog::EntryList entries;
|
| - log.GetEntries(&entries);
|
| -
|
| - EXPECT_EQ(4u, entries.size());
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 0, NetLog::TYPE_INIT_PROXY_RESOLVER));
|
| - EXPECT_TRUE(LogContainsBeginEvent(
|
| - entries, 1, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 2, NetLog::TYPE_INIT_PROXY_RESOLVER_FETCH_PAC_SCRIPT));
|
| - EXPECT_TRUE(LogContainsEndEvent(
|
| - entries, 3, NetLog::TYPE_INIT_PROXY_RESOLVER));
|
| -}
|
| -
|
| -class SynchronousSuccessDhcpFetcher : public DhcpProxyScriptFetcher {
|
| - public:
|
| - explicit SynchronousSuccessDhcpFetcher(const string16& expected_text)
|
| - : gurl_("http://dhcppac/"), expected_text_(expected_text) {
|
| - }
|
| -
|
| - int Fetch(string16* utf16_text, OldCompletionCallback* callback) OVERRIDE {
|
| - *utf16_text = expected_text_;
|
| - return OK;
|
| - }
|
| -
|
| - void Cancel() OVERRIDE {
|
| - }
|
| -
|
| - const GURL& GetPacURL() const OVERRIDE {
|
| - return gurl_;
|
| - }
|
| -
|
| - const string16& expected_text() const {
|
| - return expected_text_;
|
| - }
|
| -
|
| - private:
|
| - GURL gurl_;
|
| - string16 expected_text_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(SynchronousSuccessDhcpFetcher);
|
| -};
|
| -
|
| -// All of the tests above that use InitProxyResolver have tested
|
| -// failure to fetch a PAC file via DHCP configuration, so we now test
|
| -// success at downloading and parsing, and then success at downloading,
|
| -// failure at parsing.
|
| -
|
| -TEST(InitProxyResolverTest, AutodetectDhcpSuccess) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - SynchronousSuccessDhcpFetcher dhcp_fetcher(
|
| - WideToUTF16(L"http://bingo/!valid-script"));
|
| -
|
| - ProxyConfig config;
|
| - config.set_auto_detect(true);
|
| -
|
| - rules.AddSuccessRule("http://bingo/");
|
| - rules.AddFailDownloadRule("http://wpad/wpad.dat");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - ProxyConfig effective_config;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, NULL);
|
| - EXPECT_EQ(OK, init.Init(
|
| - config, base::TimeDelta(), &effective_config, &callback));
|
| - EXPECT_EQ(dhcp_fetcher.expected_text(),
|
| - resolver.script_data()->utf16());
|
| -
|
| - EXPECT_TRUE(effective_config.has_pac_url());
|
| - EXPECT_EQ(GURL("http://dhcppac/"), effective_config.pac_url());
|
| -}
|
| -
|
| -TEST(InitProxyResolverTest, AutodetectDhcpFailParse) {
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| - SynchronousSuccessDhcpFetcher dhcp_fetcher(
|
| - WideToUTF16(L"http://bingo/!invalid-script"));
|
| -
|
| - ProxyConfig config;
|
| - config.set_auto_detect(true);
|
| -
|
| - rules.AddFailParsingRule("http://bingo/");
|
| - rules.AddFailDownloadRule("http://wpad/wpad.dat");
|
| -
|
| - TestOldCompletionCallback callback;
|
| - ProxyConfig effective_config;
|
| - InitProxyResolver init(&resolver, &fetcher, &dhcp_fetcher, NULL);
|
| - // Since there is fallback to DNS-based WPAD, the final error will be that
|
| - // it failed downloading, not that it failed parsing.
|
| - EXPECT_EQ(kFailedDownloading,
|
| - init.Init(config, base::TimeDelta(), &effective_config, &callback));
|
| - EXPECT_EQ(NULL, resolver.script_data());
|
| -
|
| - EXPECT_FALSE(effective_config.has_pac_url());
|
| -}
|
| -
|
| -class AsyncFailDhcpFetcher
|
| - : public DhcpProxyScriptFetcher,
|
| - public base::RefCountedThreadSafe<AsyncFailDhcpFetcher> {
|
| - public:
|
| - AsyncFailDhcpFetcher() : callback_(NULL) {
|
| - }
|
| -
|
| - int Fetch(string16* utf16_text, OldCompletionCallback* callback) OVERRIDE {
|
| - callback_ = callback;
|
| - MessageLoop::current()->PostTask(
|
| - FROM_HERE,
|
| - NewRunnableMethod(this, &AsyncFailDhcpFetcher::CallbackWithFailure));
|
| - return ERR_IO_PENDING;
|
| - }
|
| -
|
| - void Cancel() OVERRIDE {
|
| - callback_ = NULL;
|
| - }
|
| -
|
| - const GURL& GetPacURL() const OVERRIDE {
|
| - return dummy_gurl_;
|
| - }
|
| -
|
| - void CallbackWithFailure() {
|
| - if (callback_)
|
| - callback_->Run(ERR_PAC_NOT_IN_DHCP);
|
| - }
|
| -
|
| - private:
|
| - GURL dummy_gurl_;
|
| - OldCompletionCallback* callback_;
|
| -};
|
| -
|
| -TEST(InitProxyResolverTest, DhcpCancelledByDestructor) {
|
| - // This regression test would crash before
|
| - // http://codereview.chromium.org/7044058/
|
| - // Thus, we don't care much about actual results (hence no EXPECT or ASSERT
|
| - // macros below), just that it doesn't crash.
|
| - Rules rules;
|
| - RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
|
| - RuleBasedProxyScriptFetcher fetcher(&rules);
|
| -
|
| - scoped_refptr<AsyncFailDhcpFetcher> dhcp_fetcher(new AsyncFailDhcpFetcher());
|
| -
|
| - ProxyConfig config;
|
| - config.set_auto_detect(true);
|
| - rules.AddFailDownloadRule("http://wpad/wpad.dat");
|
| -
|
| - TestOldCompletionCallback callback;
|
| -
|
| - // Scope so InitProxyResolver gets destroyed early.
|
| - {
|
| - InitProxyResolver init(&resolver, &fetcher, dhcp_fetcher.get(), NULL);
|
| - init.Init(config, base::TimeDelta(), NULL, &callback);
|
| - }
|
| -
|
| - // Run the message loop to let the DHCP fetch complete and post the results
|
| - // back. Before the fix linked to above, this would try to invoke on
|
| - // the callback object provided by InitProxyResolver after it was
|
| - // no longer valid.
|
| - MessageLoop::current()->RunAllPending();
|
| -}
|
| -
|
| -} // namespace
|
| -} // namespace net
|
|
|