Inject CanAddURLToHistory into TopSitesImpl

net/proxy/proxy_resolver_mojo_unittest.cc

 // Copyright 2015 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 "net/proxy/proxy_resolver_mojo.h"
 
 #include <list>
 #include <map>
 #include <queue>
 #include <string>
@@ skipping 383 matching lines @@
                                   int result) {
     callback.Run(result);
     mojo_proxy_resolver_factory_.AddFuturePacScriptAction(
         instance, SetPacScriptAction::ReturnResult(OK));
     scoped_refptr<ProxyResolverScriptData> pac_script(
         ProxyResolverScriptData::FromUTF8(kScriptData));
     EXPECT_EQ(ERR_IO_PENDING,
               proxy_resolver_mojo_->SetPacScript(pac_script, pac_callback));
   }
 
+  void DeleteProxyResolverCallback(const CompletionCallback& callback,
+                                   int result) {
+    proxy_resolver_mojo_.reset();
+    callback.Run(result);
+  }
+
   MockHostResolver mock_host_resolver_;
   TestMojoProxyResolverFactory mojo_proxy_resolver_factory_;
   scoped_ptr<ProxyResolverMojo> proxy_resolver_mojo_;
 };
 
 TEST_F(ProxyResolverMojoTest, SetPacScript) {
   SetPacScript(0);
   EXPECT_EQ(kScriptData,
             mojo_proxy_resolver_factory_.GetMockResolver().pac_script_data());
 }
@@ skipping 106 matching lines @@
   EXPECT_EQ(ERR_IO_PENDING, proxy_resolver_mojo_->SetPacScript(
                                 pac_script, callback1.callback()));
   proxy_resolver_mojo_->CancelSetPacScript();
 
   TestCompletionCallback callback2;
   EXPECT_EQ(ERR_IO_PENDING, proxy_resolver_mojo_->SetPacScript(
                                 pac_script, callback2.callback()));
   EXPECT_EQ(ERR_UNEXPECTED, callback2.GetResult(ERR_IO_PENDING));
 }
 
+TEST_F(ProxyResolverMojoTest, SetPacScript_DeleteInCallback) {
+  mojo_proxy_resolver_factory_.AddFuturePacScriptAction(
+      0, SetPacScriptAction::ReturnResult(OK));
+  scoped_refptr<ProxyResolverScriptData> pac_script(
+      ProxyResolverScriptData::FromUTF8(kScriptData));
+  TestCompletionCallback callback;
+  EXPECT_EQ(OK,
+            callback.GetResult(proxy_resolver_mojo_->SetPacScript(
+                pac_script,
+                base::Bind(&ProxyResolverMojoTest::DeleteProxyResolverCallback,
+                           base::Unretained(this), callback.callback()))));
+  EXPECT_EQ(1, mojo_proxy_resolver_factory_.num_create_calls());
+}
+
+TEST_F(ProxyResolverMojoTest, SetPacScript_DeleteInCallbackFromDisconnect) {
+  mojo_proxy_resolver_factory_.AddFuturePacScriptAction(
+      0, SetPacScriptAction::Disconnect());
+  scoped_refptr<ProxyResolverScriptData> pac_script(
+      ProxyResolverScriptData::FromUTF8(kScriptData));
+  TestCompletionCallback callback;
+  EXPECT_EQ(ERR_PAC_SCRIPT_TERMINATED,
+            callback.GetResult(proxy_resolver_mojo_->SetPacScript(
+                pac_script,
+                base::Bind(&ProxyResolverMojoTest::DeleteProxyResolverCallback,
+                           base::Unretained(this), callback.callback()))));
+  EXPECT_EQ(1, mojo_proxy_resolver_factory_.num_create_calls());
+}
+
 TEST_F(ProxyResolverMojoTest, GetProxyForURL) {
   mojo_proxy_resolver_factory_.AddFutureGetProxyAction(
       0, GetProxyForUrlAction::ReturnServers(
              GURL(kExampleUrl), ProxyServersFromPacString("DIRECT")));
   SetPacScript(0);
 
   scoped_ptr<Request> request(MakeRequest(GURL(kExampleUrl)));
   EXPECT_EQ(ERR_IO_PENDING, request->Resolve());
   EXPECT_EQ(OK, request->WaitForResult());
 
@@ skipping 185 matching lines @@
   // Setting the PAC script above should have been successful and let us send a
   // resolve request.
   scoped_ptr<Request> request(MakeRequest(GURL(kExampleUrl)));
   EXPECT_EQ(ERR_IO_PENDING, request->Resolve());
   EXPECT_EQ(OK, request->WaitForResult());
   EXPECT_EQ("DIRECT", request->results().ToPacString());
 
   EXPECT_EQ(2, mojo_proxy_resolver_factory_.num_create_calls());
 }
 
+TEST_F(ProxyResolverMojoTest, GetProxyForURL_DeleteInCallback) {
+  mojo_proxy_resolver_factory_.AddFutureGetProxyAction(
+      0, GetProxyForUrlAction::ReturnServers(
+             GURL(kExampleUrl), ProxyServersFromPacString("DIRECT")));
+  SetPacScript(0);
+  ProxyInfo results;
+  TestCompletionCallback callback;
+  ProxyResolver::RequestHandle handle;
+  BoundNetLog net_log;
+  EXPECT_EQ(OK,
+            callback.GetResult(proxy_resolver_mojo_->GetProxyForURL(
+                GURL(kExampleUrl), &results,
+                base::Bind(&ProxyResolverMojoTest::DeleteProxyResolverCallback,
+                           base::Unretained(this), callback.callback()),
+                &handle, net_log)));
+}
+
+TEST_F(ProxyResolverMojoTest, GetProxyForURL_DeleteInCallbackFromDisconnect) {
+  mojo_proxy_resolver_factory_.AddFutureGetProxyAction(
+      0, GetProxyForUrlAction::Disconnect(GURL(kExampleUrl)));
+  SetPacScript(0);
+  ProxyInfo results;
+  TestCompletionCallback callback;
+  ProxyResolver::RequestHandle handle;
+  BoundNetLog net_log;
+  EXPECT_EQ(ERR_PAC_SCRIPT_TERMINATED,
+            callback.GetResult(proxy_resolver_mojo_->GetProxyForURL(
+                GURL(kExampleUrl), &results,
+                base::Bind(&ProxyResolverMojoTest::DeleteProxyResolverCallback,
+                           base::Unretained(this), callback.callback()),
+                &handle, net_log)));
+}
+
 }  // namespace net