1 files changed, 22 insertions, 19 deletions

diff --git a/mojo/shell/application_manager_unittest.cc b/mojo/shell/application_manager_unittest.cc
index 77a959c..c626b01 100644
--- a/mojo/shell/application_manager_unittest.cc
+++ b/mojo/shell/application_manager_unittest.cc
@@ -17,6 +17,7 @@
 #include "mojo/public/cpp/bindings/strong_binding.h"
 #include "mojo/shell/application_loader.h"
 #include "mojo/shell/application_manager.h"
+#include "mojo/shell/connect_util.h"
 #include "mojo/shell/fetcher.h"
 #include "mojo/shell/test.mojom.h"
 #include "mojo/shell/test_package_manager.h"
@@ -512,8 +513,8 @@ class ApplicationManagerTest : public testing::Test {
         scoped_ptr<ApplicationLoader>(test_loader_));
 
     TestServicePtr service_proxy;
-    application_manager_->ConnectToService(GURL(kTestURLString),
-                                           &service_proxy);
+    ConnectToService(application_manager_.get(), GURL(kTestURLString),
+                     &service_proxy);
     test_client_.reset(new TestClient(service_proxy.Pass()));
   }
 
@@ -589,12 +590,14 @@ TEST_F(ApplicationManagerTest, SetLoaders) {
 
   // test::test1 should go to url_loader.
   TestServicePtr test_service;
-  application_manager_->ConnectToService(GURL("test:test1"), &test_service);
+  ConnectToService(application_manager_.get(), GURL("test:test1"),
+                   &test_service);
   EXPECT_EQ(1, url_loader->num_loads());
   EXPECT_EQ(0, default_loader->num_loads());
 
   // http::test1 should go to default loader.
-  application_manager_->ConnectToService(GURL("http:test1"), &test_service);
+  ConnectToService(application_manager_.get(), GURL("http:test1"),
+                   &test_service);
   EXPECT_EQ(1, url_loader->num_loads());
   EXPECT_EQ(1, default_loader->num_loads());
 }
@@ -609,7 +612,7 @@ TEST_F(ApplicationManagerTest, ACallB) {
   AddLoaderForURL(GURL(kTestBURLString), kTestAURLString);
 
   TestAPtr a;
-  application_manager_->ConnectToService(GURL(kTestAURLString), &a);
+  ConnectToService(application_manager_.get(), GURL(kTestAURLString), &a);
   a->CallB();
   loop_.Run();
   EXPECT_EQ(1, tester_context_.num_b_calls());
@@ -625,7 +628,7 @@ TEST_F(ApplicationManagerTest, BCallC) {
   AddLoaderForURL(GURL(kTestBURLString), kTestAURLString);
 
   TestAPtr a;
-  application_manager_->ConnectToService(GURL(kTestAURLString), &a);
+  ConnectToService(application_manager_.get(), GURL(kTestAURLString), &a);
   a->CallCFromB();
   loop_.Run();
 
@@ -641,7 +644,7 @@ TEST_F(ApplicationManagerTest, BDeleted) {
   AddLoaderForURL(GURL(kTestBURLString), std::string());
 
   TestAPtr a;
-  application_manager_->ConnectToService(GURL(kTestAURLString), &a);
+  ConnectToService(application_manager_.get(), GURL(kTestAURLString), &a);
 
   a->CallB();
   loop_.Run();
@@ -664,7 +667,7 @@ TEST_F(ApplicationManagerTest, ANoLoadB) {
   AddLoaderForURL(GURL(kTestBURLString), "test:TestC");
 
   TestAPtr a;
-  application_manager_->ConnectToService(GURL(kTestAURLString), &a);
+  ConnectToService(application_manager_.get(), GURL(kTestAURLString), &a);
   a->CallB();
   loop_.Run();
   EXPECT_EQ(0, tester_context_.num_b_calls());
@@ -679,7 +682,7 @@ TEST_F(ApplicationManagerTest, NoServiceNoLoad) {
   // There is no TestC service implementation registered with
   // ApplicationManager, so this cannot succeed (but also shouldn't crash).
   TestCPtr c;
-  application_manager_->ConnectToService(GURL(kTestAURLString), &c);
+  ConnectToService(application_manager_.get(), GURL(kTestAURLString), &c);
   c.set_connection_error_handler(
       []() { base::MessageLoop::current()->QuitWhenIdle(); });
 
@@ -742,29 +745,29 @@ TEST_F(ApplicationManagerTest, SameIdentityShouldNotCauseDuplicateLoad) {
   EXPECT_EQ(1, test_loader_->num_loads());
 
   TestServicePtr test_service;
-  application_manager_->ConnectToService(GURL("http://www.example.org/abc?def"),
-                                         &test_service);
+  ConnectToService(application_manager_.get(),
+                   GURL("http://www.example.org/abc?def"), &test_service);
   EXPECT_EQ(2, test_loader_->num_loads());
 
   // Exactly the same URL as above.
-  application_manager_->ConnectToService(GURL("http://www.example.org/abc?def"),
-                                         &test_service);
+  ConnectToService(application_manager_.get(),
+                   GURL("http://www.example.org/abc?def"), &test_service);
   EXPECT_EQ(2, test_loader_->num_loads());
 
   // The same identity as the one above because only the query string is
   // different.
-  application_manager_->ConnectToService(GURL("http://www.example.org/abc"),
-                                         &test_service);
+  ConnectToService(application_manager_.get(),
+                   GURL("http://www.example.org/abc"), &test_service);
   EXPECT_EQ(2, test_loader_->num_loads());
 
   // A different identity because the path is different.
-  application_manager_->ConnectToService(
-      GURL("http://www.example.org/another_path"), &test_service);
+  ConnectToService(application_manager_.get(),
+                   GURL("http://www.example.org/another_path"), &test_service);
   EXPECT_EQ(3, test_loader_->num_loads());
 
   // A different identity because the domain is different.
-  application_manager_->ConnectToService(
-      GURL("http://www.another_domain.org/abc"), &test_service);
+  ConnectToService(application_manager_.get(),
+                   GURL("http://www.another_domain.org/abc"), &test_service);
   EXPECT_EQ(4, test_loader_->num_loads());
 }