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_; |