net/proxy/proxy_service_mojo_unittest.cc

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "net/proxy/proxy_service_mojo.h"
   6
   7 #include <string>
   8
   9 #include "base/memory/scoped_ptr.h"
  10 #include "net/base/load_flags.h"
  11 #include "net/base/test_completion_callback.h"
  12 #include "net/dns/mock_host_resolver.h"
  13 #include "net/log/net_log.h"
  14 #include "net/proxy/dhcp_proxy_script_fetcher.h"
  15 #include "net/proxy/mock_proxy_script_fetcher.h"
  16 #include "net/proxy/proxy_config_service_fixed.h"
  17 #include "net/proxy/proxy_service.h"
  18 #include "testing/gtest/include/gtest/gtest.h"
  19 #include "url/gurl.h"
  20
  21 namespace net {
  22
  23 namespace {
  24
  25 const char kPacUrl[] = "http://example.com/proxy.pac";
  26 const char kSimplePacScript[] =
  27     "function FindProxyForURL(url, host) {\n"
  28     "  return 'PROXY foo:1234';\n"
  29     "}";
  30 const char kDnsResolvePacScript[] =
  31     "function FindProxyForURL(url, host) {\n"
  32     "  if (dnsResolveEx('example.com') != '1.2.3.4')\n"
  33     "    return 'DIRECT';\n"
  34     "  return 'QUIC bar:4321';\n"
  35     "}";
  36
  37 }  // namespace
  38
  39 class ProxyServiceMojoTest : public testing::Test {
  40  protected:
  41   void SetUp() override {
  42     mock_host_resolver_.rules()->AddRule("example.com", "1.2.3.4");
  43
  44     fetcher_ = new MockProxyScriptFetcher;
  45     proxy_service_.reset(CreateProxyServiceUsingMojoInProcess(
  46         new ProxyConfigServiceFixed(
  47             ProxyConfig::CreateFromCustomPacURL(GURL(kPacUrl))),
  48         fetcher_, new DoNothingDhcpProxyScriptFetcher(), &mock_host_resolver_,
  49         nullptr /* NetLog* */, nullptr /* NetworkDelegate* */));
  50   }
  51
  52   MockHostResolver mock_host_resolver_;
  53   MockProxyScriptFetcher* fetcher_;  // Owned by |proxy_service_|.
  54   scoped_ptr<ProxyService> proxy_service_;
  55 };
  56
  57 TEST_F(ProxyServiceMojoTest, Basic) {
  58   ProxyInfo info;
  59   TestCompletionCallback callback;
  60   EXPECT_EQ(ERR_IO_PENDING,
  61             proxy_service_->ResolveProxy(GURL("http://foo"), LOAD_NORMAL, &info,
  62                                          callback.callback(), nullptr, nullptr,
  63                                          BoundNetLog()));
  64
  65   // Proxy script fetcher should have a fetch triggered by the first
  66   // |ResolveProxy()| request.
  67   EXPECT_TRUE(fetcher_->has_pending_request());
  68   EXPECT_EQ(GURL(kPacUrl), fetcher_->pending_request_url());
  69   fetcher_->NotifyFetchCompletion(OK, kSimplePacScript);
  70
  71   EXPECT_EQ(OK, callback.WaitForResult());
  72   EXPECT_EQ("PROXY foo:1234", info.ToPacString());
  73   EXPECT_EQ(0u, mock_host_resolver_.num_resolve());
  74 }
  75
  76 TEST_F(ProxyServiceMojoTest, DnsResolution) {
  77   ProxyInfo info;
  78   TestCompletionCallback callback;
  79   EXPECT_EQ(ERR_IO_PENDING,
  80             proxy_service_->ResolveProxy(GURL("http://foo"), LOAD_NORMAL, &info,
  81                                          callback.callback(), nullptr, nullptr,
  82                                          BoundNetLog()));
  83
  84   // Proxy script fetcher should have a fetch triggered by the first
  85   // |ResolveProxy()| request.
  86   EXPECT_TRUE(fetcher_->has_pending_request());
  87   EXPECT_EQ(GURL(kPacUrl), fetcher_->pending_request_url());
  88   fetcher_->NotifyFetchCompletion(OK, kDnsResolvePacScript);
  89
  90   EXPECT_EQ(OK, callback.WaitForResult());
  91   EXPECT_EQ("QUIC bar:4321", info.ToPacString());
  92   EXPECT_EQ(1u, mock_host_resolver_.num_resolve());
  93 }
  94
  95 }  // namespace net