Fixing Intermittent SecurityKey Unittest Failures

remoting/host/security_key/fake_security_key_ipc_client.h

 // 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.
 
 #ifndef REMOTING_HOST_SECURITY_KEY_FAKE_SECURITY_KEY_IPC_CLIENT_H_
 #define REMOTING_HOST_SECURITY_KEY_FAKE_SECURITY_KEY_IPC_CLIENT_H_
 
 #include <memory>
 #include <string>
 
@@ skipping 57 matching lines @@
   }
 
   void set_send_security_request_should_succeed(bool should_succeed) {
     send_security_request_should_succeed_ = should_succeed;
   }
 
   void set_security_key_response_payload(const std::string& response_payload) {
     security_key_response_payload_ = response_payload;
   }
 
+  void set_on_channel_connected_callback(const base::Closure& callback) {
+    on_channel_connected_callback_ = callback;
+  }
+
  private:
   // IPC::Listener implementation.
   bool OnMessageReceived(const IPC::Message& message) override;
   void OnChannelConnected(int32_t peer_pid) override;
   void OnChannelError() override;
 
   // Handles security key response IPC messages.
   void OnSecurityKeyResponse(const std::string& request_data);
 
   // Handles the ConnectionReady IPC message.
   void OnConnectionReady();
 
   // Handles the InvalidSession IPC message.
   void OnInvalidSession();
 
   // Called when a change in the IPC channel state has occurred.
   base::Closure channel_event_callback_;
 
+  // Called when the IPC Channel is connected.
+  base::Closure on_channel_connected_callback_;
+
   // Used for sending/receiving security key messages between processes.
   std::unique_ptr<IPC::Channel> client_channel_;
 
   // Provides the contents of the last IPC message received.
   std::string last_message_received_;
 
   // Determines whether EstablishIpcConnection() returns success or failure.
   bool establish_ipc_connection_should_succeed_ = true;
 
   // Determines whether SendSecurityKeyRequest() returns success or failure.
@@ skipping 15 matching lines @@
   std::string security_key_response_payload_;
 
   base::WeakPtrFactory<FakeSecurityKeyIpcClient> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(FakeSecurityKeyIpcClient);
 };
 
 }  // namespace remoting
 
 #endif  // REMOTING_HOST_SECURITY_KEY_FAKE_SECURITY_KEY_IPC_CLIENT_H_