Remove ShellConnection::WaitForInitialize

mojo/shell/public/cpp/lib/shell_test.cc

 // Copyright 2016 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 "mojo/shell/public/cpp/shell_test.h"
 
 #include "base/message_loop/message_loop.h"
 #include "mojo/shell/background/background_shell.h"
 #include "mojo/shell/public/cpp/shell_client.h"
 
@@ skipping 18 matching lines @@
 }
 
 scoped_ptr<ShellClient> ShellTest::CreateShellClient() {
   return make_scoped_ptr(new ShellTestClient(this));
 }
 
 void ShellTest::InitializeCalled(Connector* connector,
                                  const std::string& name,
                                  const std::string& user_id,
                                  uint32_t id) {
-  connector_ = connector;
+  DCHECK_EQ(connector_, connector);
   initialize_name_ = name;
   initialize_instance_id_ = id;
   initialize_userid_ = user_id;
 }
 
 void ShellTest::SetUp() {
   shell_client_ = CreateShellClient();
   message_loop_.reset(new base::MessageLoop);
   background_shell_.reset(new shell::BackgroundShell);
   background_shell_->Init(nullptr);
   shell_connection_.reset(new ShellConnection(
       shell_client_.get(),
       background_shell_->CreateShellClientRequest(test_name_)));
-  shell_connection_->WaitForInitialize();
+  connector_ = shell_connection_->connector();
 }
 
 void ShellTest::TearDown() {
   shell_connection_.reset();
   background_shell_.reset();
   message_loop_.reset();
   shell_client_.reset();
 }
 
 }  // namespace test
 }  // namespace mojo