Better match IE's proxy settings....

net/proxy/init_proxy_resolver_unittest.cc

+// 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 <vector>
+
+#include "net/base/net_errors.h"
+#include "net/base/test_completion_callback.h"
+#include "net/proxy/init_proxy_resolver.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) {
+    }
+
+    std::string bytes() const {
+      if (set_pac_error == OK)
+        return url.spec() + "!valid-script";
+      if (fetch_error == OK)
+        return url.spec() + "!invalid-script";
+      return std::string();
+    }
+
+    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;
+    }
+    CHECK(false) << "Rule not found for " << url;
+    return rules_[0];
+  }
+
+  const Rule& GetRuleByBytes(const std::string& bytes) const {
+    for (RuleList::const_iterator it = rules_.begin(); it != rules_.end();
+         ++it) {
+      if (it->bytes() == bytes)
+        return *it;
+    }
+    CHECK(false) << "Rule not found for " << bytes;
+    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,
+                    std::string* bytes,
+                    CompletionCallback* callback) {
+    const Rules::Rule& rule = rules_->GetRuleByUrl(url);
+    int rv = rule.fetch_error;
+    EXPECT_NE(ERR_UNEXPECTED, rv);
+    if (rv == OK)
+      *bytes = rule.bytes();
+    return rv;
+  }
+
+  virtual void Cancel() {}
+
+ 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*/,
+                             CompletionCallback* /*callback*/,
+                             RequestHandle* /*request_handle*/) {
+    NOTREACHED();
+    return ERR_UNEXPECTED;
+  }
+
+  virtual void CancelRequest(RequestHandle request_handle) {
+    NOTREACHED();
+  }
+
+  virtual int SetPacScript(const GURL& pac_url,
+                           const std::string& pac_bytes,
+                           CompletionCallback* callback) {
+    const Rules::Rule& rule = expects_pac_bytes() ?
+        rules_->GetRuleByBytes(pac_bytes) :
+        rules_->GetRuleByUrl(pac_url);
+
+    int rv = rule.set_pac_error;
+    EXPECT_NE(ERR_UNEXPECTED, rv);
+
+    if (expects_pac_bytes())
+      EXPECT_EQ(rule.bytes(), pac_bytes);
+    else
+      EXPECT_EQ(rule.url, pac_url);
+
+    if (rv == OK) {
+      pac_bytes_ = pac_bytes;
+      pac_url_ = pac_url;
+    }
+    return rv;
+  }
+
+  const std::string& pac_bytes() const { return pac_bytes_; }
+  const GURL& pac_url() const { return pac_url_; }
+
+ private:
+  const Rules* rules_;
+  std::string pac_bytes_;
+  GURL pac_url_;
+};
+
+// Succeed using custom PAC script.
+TEST(InitProxyResolverTest, CustomPacSucceeds) {
+  Rules rules;
+  RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
+  RuleBasedProxyScriptFetcher fetcher(&rules);
+
+  ProxyConfig config;
+  config.pac_url = GURL("http://custom/proxy.pac");
+
+  Rules::Rule rule = rules.AddSuccessRule("http://custom/proxy.pac");
+
+  TestCompletionCallback callback;
+  InitProxyResolver init(&resolver, &fetcher);
+  EXPECT_EQ(OK, init.Init(config, &callback));
+  EXPECT_EQ(rule.bytes(), resolver.pac_bytes());
+}
+
+// Fail downloading the custom PAC script.
+TEST(InitProxyResolverTest, CustomPacFails1) {
+  Rules rules;
+  RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
+  RuleBasedProxyScriptFetcher fetcher(&rules);
+
+  ProxyConfig config;
+  config.pac_url = GURL("http://custom/proxy.pac");
+
+  rules.AddFailDownloadRule("http://custom/proxy.pac");
+
+  TestCompletionCallback callback;
+  InitProxyResolver init(&resolver, &fetcher);
+  EXPECT_EQ(kFailedDownloading, init.Init(config, &callback));
+  EXPECT_EQ("", resolver.pac_bytes());
+}
+
+// Fail parsing the custom PAC script.
+TEST(InitProxyResolverTest, CustomPacFails2) {
+  Rules rules;
+  RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
+  RuleBasedProxyScriptFetcher fetcher(&rules);
+
+  ProxyConfig config;
+  config.pac_url = GURL("http://custom/proxy.pac");
+
+  rules.AddFailParsingRule("http://custom/proxy.pac");
+
+  TestCompletionCallback callback;
+  InitProxyResolver init(&resolver, &fetcher);
+  EXPECT_EQ(kFailedParsing, init.Init(config, &callback));
+  EXPECT_EQ("", resolver.pac_bytes());
+}
+
+// Succeeds in choosing autodetect (wpad).
+TEST(InitProxyResolverTest, AutodetectSuccess) {
+  Rules rules;
+  RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
+  RuleBasedProxyScriptFetcher fetcher(&rules);
+
+  ProxyConfig config;
+  config.auto_detect = true;
+
+  Rules::Rule rule = rules.AddSuccessRule("http://wpad/wpad.dat");
+
+  TestCompletionCallback callback;
+  InitProxyResolver init(&resolver, &fetcher);
+  EXPECT_EQ(OK, init.Init(config, &callback));
+  EXPECT_EQ(rule.bytes(), resolver.pac_bytes());
+}
+
+// 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);
+
+  ProxyConfig config;
+  config.auto_detect = true;
+  config.pac_url = GURL("http://custom/proxy.pac");
+
+  rules.AddFailDownloadRule("http://wpad/wpad.dat");
+  Rules::Rule rule = rules.AddSuccessRule("http://custom/proxy.pac");
+
+  TestCompletionCallback callback;
+  InitProxyResolver init(&resolver, &fetcher);
+  EXPECT_EQ(OK, init.Init(config, &callback));
+  EXPECT_EQ(rule.bytes(), resolver.pac_bytes());
+}
+
+// Fails at WPAD (parsing), but succeeds in choosing the custom PAC.
+TEST(InitProxyResolverTest, AutodetectFailCustomSuccess2) {
+  Rules rules;
+  RuleBasedProxyResolver resolver(&rules, true /*expects_pac_bytes*/);
+  RuleBasedProxyScriptFetcher fetcher(&rules);
+
+  ProxyConfig config;
+  config.auto_detect = true;
+  config.pac_url = GURL("http://custom/proxy.pac");
+
+  rules.AddFailParsingRule("http://wpad/wpad.dat");
+  Rules::Rule rule = rules.AddSuccessRule("http://custom/proxy.pac");
+
+  TestCompletionCallback callback;
+  InitProxyResolver init(&resolver, &fetcher);
+  EXPECT_EQ(OK, init.Init(config, &callback));
+  EXPECT_EQ(rule.bytes(), resolver.pac_bytes());
+}
+
+// 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);
+
+  ProxyConfig config;
+  config.auto_detect = true;
+  config.pac_url = GURL("http://custom/proxy.pac");
+
+  rules.AddFailDownloadRule("http://wpad/wpad.dat");
+  rules.AddFailDownloadRule("http://custom/proxy.pac");
+
+  TestCompletionCallback callback;
+  InitProxyResolver init(&resolver, &fetcher);
+  EXPECT_EQ(kFailedDownloading, init.Init(config, &callback));
+  EXPECT_EQ("", resolver.pac_bytes());
+}
+
+// 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);
+
+  ProxyConfig config;
+  config.auto_detect = true;
+  config.pac_url = GURL("http://custom/proxy.pac");
+
+  rules.AddFailDownloadRule("http://wpad/wpad.dat");
+  rules.AddFailParsingRule("http://custom/proxy.pac");
+
+  TestCompletionCallback callback;
+  InitProxyResolver init(&resolver, &fetcher);
+  EXPECT_EQ(kFailedParsing, init.Init(config, &callback));
+  EXPECT_EQ("", resolver.pac_bytes());
+}
+
+// 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);
+
+  ProxyConfig config;
+  config.auto_detect = true;
+  config.pac_url = GURL("http://custom/proxy.pac");
+
+  rules.AddFailParsingRule("");  // Autodetect.
+  Rules::Rule rule = rules.AddSuccessRule("http://custom/proxy.pac");
+
+  TestCompletionCallback callback;
+  InitProxyResolver init(&resolver, &fetcher);
+  EXPECT_EQ(OK, init.Init(config, &callback));
+  EXPECT_EQ(rule.url, resolver.pac_url());
+}
+
+}  // namespace
+}  // namespace net