chromeos: 1st draft of ProxyConfigService for chromeos...

chrome/browser/chromeos/proxy_config_service_impl_unittest.cc

+// Copyright (c) 2010 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/chromeos/proxy_config_service_impl.h"
+
+#include <map>
+#include <string>
+#include <vector>
+
+#include "base/format_macros.h"
+#include "base/logging.h"
+#include "base/string_util.h"
+#include "chrome/browser/chrome_thread.h"
+#include "net/proxy/proxy_config_service_common_unittest.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/platform_test.h"
+
+namespace chromeos {
+
+namespace {
+
+struct Input {  // Fields of chromeos::ProxyConfigServiceImpl::ProxyConfig.
+  ProxyConfigServiceImpl::ProxyConfig::Mode mode;
+  const char* pac_url;
+  const char* single_uri;
+  const char* http_uri;
+  const char* https_uri;
+  const char* ftp_uri;
+  const char* socks_uri;
+  const char* bypass_rules;
+};
+
+// Builds an identifier for each test in an array.
+#define TEST_DESC(desc) StringPrintf("at line %d <%s>", __LINE__, desc)
+
+// Shortcuts to declare enums within chromeos's ProxyConfig.
+#define MK_MODE(mode) ProxyConfigServiceImpl::ProxyConfig::MODE_##mode
+#define MK_SRC(src) ProxyConfigServiceImpl::ProxyConfig::SOURCE_##src
+
+// Inspired from net/proxy/proxy_config_service_linux_unittest.cc.
+const struct {
+  // Short description to identify the test
+  std::string description;
+
+  Input input;
+
+  // Expected outputs from fields of net::ProxyConfig (via IO).
+  bool auto_detect;
+  GURL pac_url;
+  net::ProxyRulesExpectation proxy_rules;
+} tests[] = {
+  {
+    TEST_DESC("No proxying"),
+    { // Input.
+      MK_MODE(DIRECT),  // mode
+    },
+
+    // Expected result.
+    false,                                   // auto_detect
+    GURL(),                                  // pac_url
+    net::ProxyRulesExpectation::Empty(),     // proxy_rules
+  },
+
+  {
+    TEST_DESC("Auto detect"),
+    { // Input.
+      MK_MODE(AUTO_DETECT),  // mode
+    },
+
+    // Expected result.
+    true,                                    // auto_detect
+    GURL(),                                  // pac_url
+    net::ProxyRulesExpectation::Empty(),     // proxy_rules
+  },
+
+  {
+    TEST_DESC("Valid PAC URL"),
+    { // Input.
+      MK_MODE(PAC_SCRIPT),     // mode
+      "http://wpad/wpad.dat",  // pac_url
+    },
+
+    // Expected result.
+    false,                                   // auto_detect
+    GURL("http://wpad/wpad.dat"),            // pac_url
+    net::ProxyRulesExpectation::Empty(),     // proxy_rules
+  },
+
+  {
+    TEST_DESC("Invalid PAC URL"),
+    { // Input.
+      MK_MODE(PAC_SCRIPT),  // mode
+      "wpad.dat",           // pac_url
+    },
+
+    // Expected result.
+    false,                                   // auto_detect
+    GURL(),                                  // pac_url
+    net::ProxyRulesExpectation::Empty(),     // proxy_rules
+  },
+
+  {
+    TEST_DESC("Single-host in proxy list"),
+    { // Input.
+      MK_MODE(SINGLE_PROXY),  // mode
+      NULL,                   // pac_url
+      "www.google.com",       // single_uri
+    },
+
+    // Expected result.
+    false,                                   // auto_detect
+    GURL(),                                  // pac_url
+    net::ProxyRulesExpectation::Single(      // proxy_rules
+        "www.google.com:80",  // single proxy
+        ""),                  // bypass rules
+  },
+
+  {
+    TEST_DESC("Single-host, different port"),
+    { // Input.
+      MK_MODE(SINGLE_PROXY),  // mode
+      NULL,                   // pac_url
+      "www.google.com:99",    // single_uri
+    },
+
+    // Expected result.
+    false,                                   // auto_detect
+    GURL(),                                  // pac_url
+    net::ProxyRulesExpectation::Single(      // proxy_rules
+        "www.google.com:99",  // single
+        ""),                  // bypass rules
+  },
+
+  {
+    TEST_DESC("Tolerate a scheme"),
+    { // Input.
+      MK_MODE(SINGLE_PROXY),       // mode
+      NULL,                        // pac_url
+      "http://www.google.com:99",  // single_uri
+    },
+
+    // Expected result.
+    false,                                   // auto_detect
+    GURL(),                                  // pac_url
+    net::ProxyRulesExpectation::Single(      // proxy_rules
+        "www.google.com:99",  // single proxy
+        ""),                  // bypass rules
+  },
+
+  {
+    TEST_DESC("Per-scheme proxy rules"),
+    { // Input.
+      MK_MODE(PROXY_PER_SCHEME),  // mode
+      NULL,                       // pac_url
+      NULL,                       // single_uri
+      "www.google.com:80", "www.foo.com:110", "ftp.foo.com:121", // per-proto
+    },
+
+    // Expected result.
+    false,                                    // auto_detect
+    GURL(),                                   // pac_url
+    net::ProxyRulesExpectation::PerScheme(    // proxy_rules
+        "www.google.com:80",  // http
+        "www.foo.com:110",    // https
+        "ftp.foo.com:121",    // ftp
+        ""),                  // bypass rules
+  },
+
+// TODO(kuan): enable these.
+#if defined(TO_ENABLE_SOON)
+  {
+    TEST_DESC("socks"),
+    { // Input.
+      MK_MODE(PROXY_PER_SCHEME),  // mode
+      NULL,                       // pac_url
+      NULL,                       // single_uri
+      NULL, NULL, NULL,           // per-proto proxies
+      "socks.com:888",            // socks_uri
+    },
+
+    // Expected result.
+    false,                                   // auto_detect
+    GURL(),                                  // pac_url
+    net::ProxyRulesExpectation::Single(      // proxy_rules
+        "socks4://socks.com:888",  // single proxy
+        ""),                       // bypass rules
+  },
+
+  {
+    TEST_DESC("socks5"),
+    { // Input.
+      NULL,  // auto_proxy
+      "",  // all_proxy
+      NULL, NULL, NULL,  // per-proto proxies
+      "socks.com:888", "5",  // SOCKS
+      NULL,  // no_proxy
+    },
+
+    // Expected result.
+    false,                                   // auto_detect
+    GURL(),                                  // pac_url
+    net::ProxyRulesExpectation::Single(      // proxy_rules
+        "socks5://socks.com:888",  // single proxy
+        ""),                       // bypass rules
+    ProxyConfigServiceImpl::READ_ONLY_MAIN,  // readonly for owner
+    ProxyConfigServiceImpl::READ_ONLY_MAIN,  // readonly for non-owner
+  },
+
+  {
+    TEST_DESC("socks default port"),
+    { // Input.
+      NULL,  // auto_proxy
+      "",  // all_proxy
+      NULL, NULL, NULL,  // per-proto proxies
+      "socks.com", NULL,  // SOCKS
+      NULL,  // no_proxy
+    },
+
+    // Expected result.
+    false,                                   // auto_detect
+    GURL(),                                  // pac_url
+    net::ProxyRulesExpectation::Single(      // proxy_rules
+        "socks4://socks.com:1080",  // single proxy
+        ""),                        // bypass rules
+  },
+
+  {
+    TEST_DESC("bypass"),
+    { // Input.
+      MK_MODE(PROXY_PER_SCHEME),  // mode
+      NULL,                       // pac_url
+      "www.google.com",           // single_uri
+      NULL, NULL, NULL, NULL,     // per-proto & socks proxies
+      ".google.com, foo.com:99, 1.2.3.4:22, 127.0.0.1/8",  // bypass_rules
+    },
+
+    // Expected result;
+    false,                                   // auto_detect
+    GURL(),                                  // pac_url
+    net::ProxyRulesExpectation::Single(      // proxy_rules
+        "www.google.com:80",
+        "*.google.com,*foo.com:99,1.2.3.4:22,127.0.0.1/8"),
+  },
+#endif  // TO_ENABLE_SOON
+};  // tests
+
+}  // namespace
+
+class ProxyConfigServiceImplTest : public PlatformTest {
+ protected:
+  ProxyConfigServiceImplTest()
+      : ui_thread_(ChromeThread::UI, &message_loop_),
+        io_thread_(ChromeThread::IO, &message_loop_) {
+  }
+
+  virtual ~ProxyConfigServiceImplTest() {
+    config_service_ = NULL;
+    MessageLoop::current()->RunAllPending();
+  }
+
+  void CreateConfigService(
+      const ProxyConfigServiceImpl::ProxyConfig& init_config) {
+    // Instantiate proxy config service with |init_config|.
+    config_service_ = new ProxyConfigServiceImpl(init_config);
+  }
+
+  void SetAutomaticProxy(
+      ProxyConfigServiceImpl::ProxyConfig::Mode mode,
+      ProxyConfigServiceImpl::ProxyConfig::Source source,
+      const char* pac_url,
+      ProxyConfigServiceImpl::ProxyConfig* config,
+      ProxyConfigServiceImpl::ProxyConfig::AutomaticProxy* automatic_proxy) {
+    config->mode = mode;
+    automatic_proxy->source = source;
+    if (pac_url)
+      automatic_proxy->pac_url = GURL(pac_url);
+  }
+
+  void SetManualProxy(
+      ProxyConfigServiceImpl::ProxyConfig::Mode mode,
+      ProxyConfigServiceImpl::ProxyConfig::Source source,
+      const char* server_uri,
+      ProxyConfigServiceImpl::ProxyConfig* config,
+      ProxyConfigServiceImpl::ProxyConfig::ManualProxy* manual_proxy) {
+    if (!server_uri)
+      return;
+    config->mode = mode;
+    manual_proxy->source = source;
+    manual_proxy->server = net::ProxyServer::FromURI(server_uri,
+        net::ProxyServer::SCHEME_HTTP);
+  }
+
+  void InitConfigWithTestInput(
+      const Input& input,
+      ProxyConfigServiceImpl::ProxyConfig* init_config) {
+    ProxyConfigServiceImpl::ProxyConfig::Source source = MK_SRC(OWNER);
+    switch (input.mode) {
+      case MK_MODE(DIRECT):
+      case MK_MODE(AUTO_DETECT):
+      case MK_MODE(PAC_SCRIPT):
+        SetAutomaticProxy(input.mode, source, input.pac_url, init_config,
+                          &init_config->automatic_proxy);
+        return;
+      case MK_MODE(SINGLE_PROXY):
+        SetManualProxy(input.mode, source, input.single_uri, init_config,
+                       &init_config->single_proxy);
+        break;
+      case MK_MODE(PROXY_PER_SCHEME):
+        SetManualProxy(input.mode, source, input.http_uri, init_config,
+                       &init_config->http_proxy);
+        SetManualProxy(input.mode, source, input.https_uri, init_config,
+                       &init_config->https_proxy);
+        SetManualProxy(input.mode, source, input.ftp_uri, init_config,
+                       &init_config->ftp_proxy);
+        SetManualProxy(input.mode, source, input.socks_uri, init_config,
+                       &init_config->socks_proxy);
+        break;
+    }
+    if (input.bypass_rules) {
+      init_config->bypass_rules.ParseFromStringUsingSuffixMatching(
+          input.bypass_rules);
+    }
+  }
+
+  // Synchronously gets the latest proxy config.
+  bool SyncGetLatestProxyConfig(net::ProxyConfig* config) {
+    // Let message loop process all messages.
+    MessageLoop::current()->RunAllPending();
+    // Calls IOGetProxyConfig (which is called from
+    // ProxyConfigService::GetLatestProxyConfig), running on faked IO thread.
+    return config_service_->IOGetProxyConfig(config);
+  }
+
+  ProxyConfigServiceImpl* config_service() const {
+    return config_service_;
+  }
+
+ private:
+  MessageLoop message_loop_;
+  ChromeThread ui_thread_;
+  ChromeThread io_thread_;
+
+  scoped_refptr<ProxyConfigServiceImpl> config_service_;
+};
+
+TEST_F(ProxyConfigServiceImplTest, ChromeosProxyConfigToNetProxyConfig) {
+  for (size_t i = 0; i < ARRAYSIZE_UNSAFE(tests); ++i) {
+    SCOPED_TRACE(StringPrintf("Test[%" PRIuS "] %s", i,
+                              tests[i].description.c_str()));
+
+    ProxyConfigServiceImpl::ProxyConfig init_config;
+    InitConfigWithTestInput(tests[i].input, &init_config);
+    CreateConfigService(init_config);
+
+    net::ProxyConfig config;
+    SyncGetLatestProxyConfig(&config);
+
+    EXPECT_EQ(tests[i].auto_detect, config.auto_detect());
+    EXPECT_EQ(tests[i].pac_url, config.pac_url());
+    EXPECT_TRUE(tests[i].proxy_rules.Matches(config.proxy_rules()));
+  }
+}
+
+TEST_F(ProxyConfigServiceImplTest, ReadWriteAccess) {
+  static const char* pac_url = "http://wpad.dat";
+
+  { // Init with pac script from policy.
+    ProxyConfigServiceImpl::ProxyConfig init_config;
+    SetAutomaticProxy(MK_MODE(PAC_SCRIPT), MK_SRC(POLICY), pac_url,
+                      &init_config, &init_config.automatic_proxy);
+    CreateConfigService(init_config);
+
+    ProxyConfigServiceImpl::ProxyConfig config;
+    config_service()->UIGetProxyConfig(&config);
+
+    EXPECT_EQ(MK_SRC(POLICY), config.automatic_proxy.source);
+    // Setting should be not be writeable by owner.
+    EXPECT_FALSE(config.automatic_proxy.CanBeWrittenByUser(true));
+    // Setting should be not be writeable by non-owner.
+    EXPECT_FALSE(config.automatic_proxy.CanBeWrittenByUser(false));
+  }
+
+  { // Init with pac script from owner.
+    ProxyConfigServiceImpl::ProxyConfig init_config;
+    SetAutomaticProxy(MK_MODE(PAC_SCRIPT), MK_SRC(OWNER), pac_url,
+                      &init_config, &init_config.automatic_proxy);
+    CreateConfigService(init_config);
+
+    ProxyConfigServiceImpl::ProxyConfig config;
+    config_service()->UIGetProxyConfig(&config);
+
+    EXPECT_EQ(MK_SRC(OWNER), config.automatic_proxy.source);
+    // Setting should be writeable by owner.
+    EXPECT_TRUE(config.automatic_proxy.CanBeWrittenByUser(true));
+    // Setting should not be writeable by non-owner.
+    EXPECT_FALSE(config.automatic_proxy.CanBeWrittenByUser(false));
+  }
+}
+
+TEST_F(ProxyConfigServiceImplTest, ModifyFromUI) {
+  // Init with direct.
+  ProxyConfigServiceImpl::ProxyConfig init_config;
+  SetAutomaticProxy(MK_MODE(DIRECT), MK_SRC(OWNER), NULL, &init_config,
+                    &init_config.automatic_proxy);
+  CreateConfigService(init_config);
+
+  for (size_t i = 0; i < ARRAYSIZE_UNSAFE(tests); ++i) {
+    SCOPED_TRACE(StringPrintf("Test[%" PRIuS "] %s", i,
+                              tests[i].description.c_str()));
+
+    // Set config to tests[i].input via UI.
+    net::ProxyBypassRules bypass_rules;
+    net::ProxyServer::Scheme scheme = net::ProxyServer::SCHEME_HTTP;
+    const Input& input = tests[i].input;
+    switch (input.mode) {
+      case MK_MODE(DIRECT) :
+        config_service()->UISetProxyConfigToDirect();
+        break;
+      case MK_MODE(AUTO_DETECT) :
+        config_service()->UISetProxyConfigToAutoDetect();
+        break;
+      case MK_MODE(PAC_SCRIPT) :
+        config_service()->UISetProxyConfigToPACScript(GURL(input.pac_url));
+        break;
+      case MK_MODE(SINGLE_PROXY) :
+        config_service()->UISetProxyConfigToSingleProxy(
+            net::ProxyServer::FromURI(input.single_uri,
+                                      net::ProxyServer::SCHEME_HTTP));
+        if (input.bypass_rules) {
+          bypass_rules.ParseFromStringUsingSuffixMatching(input.bypass_rules);
+          config_service()->UISetProxyConfigBypassRules(bypass_rules);
+        }
+        break;
+      case MK_MODE(PROXY_PER_SCHEME) :
+        if (input.http_uri) {
+          config_service()->UISetProxyConfigToProxyPerScheme(
+              "http",
+              net::ProxyServer::FromURI(input.http_uri, scheme));
+        }
+        if (input.https_uri) {
+          config_service()->UISetProxyConfigToProxyPerScheme(
+              "https",
+              net::ProxyServer::FromURI(input.https_uri, scheme));
+        }
+        if (input.ftp_uri) {
+          config_service()->UISetProxyConfigToProxyPerScheme(
+              "ftp",
+              net::ProxyServer::FromURI(input.ftp_uri, scheme));
+        }
+        if (input.socks_uri) {
+          config_service()->UISetProxyConfigToProxyPerScheme(
+              "socks",
+              net::ProxyServer::FromURI(input.socks_uri, scheme));
+        }
+        if (input.bypass_rules) {
+          bypass_rules.ParseFromStringUsingSuffixMatching(input.bypass_rules);
+          config_service()->UISetProxyConfigBypassRules(bypass_rules);
+        }
+        break;
+    }
+
+    // Retrieve config from IO thread.
+    net::ProxyConfig io_config;
+    SyncGetLatestProxyConfig(&io_config);
+    EXPECT_EQ(tests[i].auto_detect, io_config.auto_detect());
+    EXPECT_EQ(tests[i].pac_url, io_config.pac_url());
+    EXPECT_TRUE(tests[i].proxy_rules.Matches(io_config.proxy_rules()));
+
+    // Retrieve config from UI thread.
+    ProxyConfigServiceImpl::ProxyConfig ui_config;
+    config_service()->UIGetProxyConfig(&ui_config);
+    EXPECT_EQ(input.mode, ui_config.mode);
+    if (input.pac_url)
+      EXPECT_EQ(GURL(input.pac_url), ui_config.automatic_proxy.pac_url);
+    const net::ProxyRulesExpectation& proxy_rules = tests[i].proxy_rules;
+    if (input.single_uri)
+      EXPECT_EQ(proxy_rules.single_proxy,
+                ui_config.single_proxy.server.ToURI());
+    if (input.http_uri)
+      EXPECT_EQ(proxy_rules.proxy_for_http,
+                ui_config.http_proxy.server.ToURI());
+    if (input.https_uri)
+      EXPECT_EQ(proxy_rules.proxy_for_https,
+                ui_config.https_proxy.server.ToURI());
+    if (input.ftp_uri)
+      EXPECT_EQ(proxy_rules.proxy_for_ftp, ui_config.ftp_proxy.server.ToURI());
+    if (input.socks_uri)
+      EXPECT_EQ(proxy_rules.socks_proxy, ui_config.socks_proxy.server.ToURI());
+    if (input.bypass_rules)
+      EXPECT_TRUE(bypass_rules.Equals(ui_config.bypass_rules));
+  }
+}
+
+TEST_F(ProxyConfigServiceImplTest, ProxyChangedObserver) {
+  // This is used to observe for OnProxyConfigChanged notification.
+  class ProxyChangedObserver : public net::ProxyConfigService::Observer {
+   public:
+    explicit ProxyChangedObserver(
+        const scoped_refptr<ProxyConfigServiceImpl>& config_service)
+        : config_service_(config_service) {
+      config_service_->AddObserver(this);
+    }
+    virtual ~ProxyChangedObserver() {
+      config_service_->RemoveObserver(this);
+    }
+    const net::ProxyConfig& config() const {
+      return config_;
+    }
+
+   private:
+    virtual void OnProxyConfigChanged(const net::ProxyConfig& config) {
+      config_ = config;
+    }
+
+    scoped_refptr<ProxyConfigServiceImpl> config_service_;
+    net::ProxyConfig config_;
+  };
+
+  // Init with direct.
+  ProxyConfigServiceImpl::ProxyConfig init_config;
+  SetAutomaticProxy(MK_MODE(DIRECT), MK_SRC(OWNER), NULL, &init_config,
+                    &init_config.automatic_proxy);
+  CreateConfigService(init_config);
+
+  ProxyChangedObserver observer(config_service());
+
+  // Set to pac script from UI.
+  config_service()->UISetProxyConfigToPACScript(GURL("http://wpad.dat"));
+
+  // Retrieve config from IO thread.
+  net::ProxyConfig io_config;
+  SyncGetLatestProxyConfig(&io_config);
+
+  // Observer should have gotten the same new proxy config.
+  EXPECT_TRUE(io_config.Equals(observer.config()));
+}
+
+}  // namespace chromeos