Chromium Code Reviews| Index: remoting/host/security_key/gnubby_auth_handler_linux_unittest.cc |
| diff --git a/remoting/host/security_key/gnubby_auth_handler_linux_unittest.cc b/remoting/host/security_key/gnubby_auth_handler_linux_unittest.cc |
| index 74ba1b7ee6661d14248e644cc95db30ca29ffa65..c9888ca073de84aa0ddcb7819c2b026f311b1b25 100644 |
| --- a/remoting/host/security_key/gnubby_auth_handler_linux_unittest.cc |
| +++ b/remoting/host/security_key/gnubby_auth_handler_linux_unittest.cc |
| @@ -57,7 +57,10 @@ class GnubbyAuthHandlerLinuxTest : public testing::Test { |
| send_message_callback_ = |
| base::Bind(&GnubbyAuthHandlerLinuxTest::SendMessageToClient, |
| base::Unretained(this)); |
| - auth_handler_ = remoting::GnubbyAuthHandler::Create(send_message_callback_); |
| + session_id_callback_ = base::Bind(&GnubbyAuthHandlerLinuxTest::GetSessionId, |
| + base::Unretained(this)); |
| + auth_handler_ = remoting::GnubbyAuthHandler::Create(send_message_callback_, |
| + session_id_callback_); |
| } |
| void WaitForSendMessageToClient() { |
| @@ -71,6 +74,8 @@ class GnubbyAuthHandlerLinuxTest : public testing::Test { |
| run_loop_->Quit(); |
| } |
| + uint32_t GetSessionId() const { return UINT32_MAX; } |
|
Sergey Ulanov
2016/06/27 23:51:58
can be a static function.
|
| + |
| void CheckHostDataMessage(int id, const std::string& expected_data) { |
| ASSERT_EQ(id, last_connection_id_received_); |
| ASSERT_EQ(expected_data.length(), last_message_received_.length()); |
| @@ -120,6 +125,7 @@ class GnubbyAuthHandlerLinuxTest : public testing::Test { |
| std::unique_ptr<GnubbyAuthHandler> auth_handler_; |
| GnubbyAuthHandler::SendMessageCallback send_message_callback_; |
| + GnubbyAuthHandler::SessionIdCallback session_id_callback_; |
| int last_connection_id_received_; |
| std::string last_message_received_; |