| Index: net/proxy/proxy_service_mojo_unittest.cc
|
| diff --git a/net/proxy/proxy_service_mojo_unittest.cc b/net/proxy/proxy_service_mojo_unittest.cc
|
| index bb9f41eb8d247cede24b0a36faafe9406addf472..5caf2a5ffd429f33874fa5bf7cc9a80e45f29174 100644
|
| --- a/net/proxy/proxy_service_mojo_unittest.cc
|
| +++ b/net/proxy/proxy_service_mojo_unittest.cc
|
| @@ -6,13 +6,16 @@
|
|
|
| #include <string>
|
|
|
| +#include "base/callback_helpers.h"
|
| #include "base/memory/scoped_ptr.h"
|
| #include "net/base/load_flags.h"
|
| #include "net/base/test_completion_callback.h"
|
| #include "net/dns/mock_host_resolver.h"
|
| #include "net/log/net_log.h"
|
| #include "net/proxy/dhcp_proxy_script_fetcher.h"
|
| +#include "net/proxy/in_process_mojo_proxy_resolver_factory.h"
|
| #include "net/proxy/mock_proxy_script_fetcher.h"
|
| +#include "net/proxy/mojo_proxy_resolver_factory.h"
|
| #include "net/proxy/proxy_config_service_fixed.h"
|
| #include "net/proxy/proxy_service.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| @@ -36,22 +39,35 @@ const char kDnsResolvePacScript[] =
|
|
|
| } // namespace
|
|
|
| -class ProxyServiceMojoTest : public testing::Test {
|
| +class ProxyServiceMojoTest : public testing::Test,
|
| + public MojoProxyResolverFactory {
|
| protected:
|
| void SetUp() override {
|
| mock_host_resolver_.rules()->AddRule("example.com", "1.2.3.4");
|
|
|
| fetcher_ = new MockProxyScriptFetcher;
|
| - proxy_service_.reset(CreateProxyServiceUsingMojoInProcess(
|
| - new ProxyConfigServiceFixed(
|
| - ProxyConfig::CreateFromCustomPacURL(GURL(kPacUrl))),
|
| + proxy_service_.reset(CreateProxyServiceUsingMojoFactory(
|
| + this, new ProxyConfigServiceFixed(
|
| + ProxyConfig::CreateFromCustomPacURL(GURL(kPacUrl))),
|
| fetcher_, new DoNothingDhcpProxyScriptFetcher(), &mock_host_resolver_,
|
| nullptr /* NetLog* */, nullptr /* NetworkDelegate* */));
|
| }
|
|
|
| + scoped_ptr<base::ScopedClosureRunner> CreateResolver(
|
| + const mojo::String& pac_script,
|
| + mojo::InterfaceRequest<interfaces::ProxyResolver> req,
|
| + interfaces::HostResolverPtr host_resolver,
|
| + interfaces::ProxyResolverFactoryRequestClientPtr client) override {
|
| + InProcessMojoProxyResolverFactory::GetInstance()->CreateResolver(
|
| + pac_script, req.Pass(), host_resolver.Pass(), client.Pass());
|
| + return make_scoped_ptr(
|
| + new base::ScopedClosureRunner(on_delete_closure_.closure()));
|
| + }
|
| +
|
| MockHostResolver mock_host_resolver_;
|
| MockProxyScriptFetcher* fetcher_; // Owned by |proxy_service_|.
|
| scoped_ptr<ProxyService> proxy_service_;
|
| + TestClosure on_delete_closure_;
|
| };
|
|
|
| TEST_F(ProxyServiceMojoTest, Basic) {
|
| @@ -71,6 +87,8 @@ TEST_F(ProxyServiceMojoTest, Basic) {
|
| EXPECT_EQ(OK, callback.WaitForResult());
|
| EXPECT_EQ("PROXY foo:1234", info.ToPacString());
|
| EXPECT_EQ(0u, mock_host_resolver_.num_resolve());
|
| + proxy_service_.reset();
|
| + on_delete_closure_.WaitForResult();
|
| }
|
|
|
| TEST_F(ProxyServiceMojoTest, DnsResolution) {
|
| @@ -90,6 +108,8 @@ TEST_F(ProxyServiceMojoTest, DnsResolution) {
|
| EXPECT_EQ(OK, callback.WaitForResult());
|
| EXPECT_EQ("QUIC bar:4321", info.ToPacString());
|
| EXPECT_EQ(1u, mock_host_resolver_.num_resolve());
|
| + proxy_service_.reset();
|
| + on_delete_closure_.WaitForResult();
|
| }
|
|
|
| } // namespace net
|
|
|