Chromium Code Reviews| Index: remoting/host/security_key/security_key_ipc_client_unittest.cc |
| diff --git a/remoting/host/security_key/security_key_ipc_client_unittest.cc b/remoting/host/security_key/security_key_ipc_client_unittest.cc |
| index 68f56ad09d4a0fd4fe1bb722a77b0d02f31099f4..717381d62070f8b97309a75a70fbfedf388c1450 100644 |
| --- a/remoting/host/security_key/security_key_ipc_client_unittest.cc |
| +++ b/remoting/host/security_key/security_key_ipc_client_unittest.cc |
| @@ -12,7 +12,7 @@ |
| #include "base/message_loop/message_loop.h" |
| #include "base/run_loop.h" |
| #include "ipc/ipc_channel.h" |
| -#include "remoting/host/security_key/fake_ipc_security_key_auth_handler.h" |
| +#include "mojo/edk/test/scoped_ipc_support.h" |
| #include "remoting/host/security_key/fake_security_key_ipc_server.h" |
| #include "remoting/host/security_key/security_key_ipc_constants.h" |
|
joedow
2016/11/03 22:25:18
Remove this header since |fake_security_key_auth_h
Sam McNally
2016/11/04 02:51:09
I'm not sure what you mean. The fake auth handler
joedow
2016/11/04 15:19:04
I noticed the files hadn't been deleted in that pa
|
| #include "testing/gtest/include/gtest/gtest.h" |
| @@ -66,6 +66,8 @@ class SecurityKeyIpcClientTest : public testing::Test { |
| // IPC tests require a valid MessageLoop to run. |
| base::MessageLoopForIO message_loop_; |
| + mojo::edk::test::ScopedIPCSupport ipc_support_; |
| + |
| // Used to allow |message_loop_| to run during tests. The instance is reset |
| // after each stage of the tests has been completed. |
| std::unique_ptr<base::RunLoop> run_loop_; |
| @@ -73,10 +75,6 @@ class SecurityKeyIpcClientTest : public testing::Test { |
| // The object under test. |
| SecurityKeyIpcClient security_key_ipc_client_; |
| - // Provides a connection details message to |security_key_ipc_client_| |
| - // for testing. |
| - FakeIpcSecurityKeyAuthHandler fake_security_key_auth_handler_; |
| - |
| // Used to send/receive security key IPC messages for testing. |
| FakeSecurityKeyIpcServer fake_ipc_server_; |
| @@ -98,13 +96,15 @@ class SecurityKeyIpcClientTest : public testing::Test { |
| }; |
| SecurityKeyIpcClientTest::SecurityKeyIpcClientTest() |
| - : run_loop_(new base::RunLoop()), |
| + : ipc_support_(message_loop_.task_runner()), |
| + run_loop_(new base::RunLoop()), |
| fake_ipc_server_( |
| kTestConnectionId, |
| /*client_session_details=*/nullptr, |
| /*initial_connect_timeout=*/base::TimeDelta::FromMilliseconds(500), |
| base::Bind(&SecurityKeyIpcClientTest::SendMessageToClient, |
| base::Unretained(this)), |
| + base::Bind(&base::DoNothing), |
| base::Bind(&SecurityKeyIpcClientTest::OperationComplete, |
| base::Unretained(this), |
| /*failed=*/false)) {} |
| @@ -153,25 +153,13 @@ std::string SecurityKeyIpcClientTest::GenerateUniqueTestChannelName() { |
| void SecurityKeyIpcClientTest::EstablishConnection(bool expect_success) { |
| // Start up the security key forwarding session IPC channel first, that way |
| // we can provide the channel using the fake SecurityKeyAuthHandler later on. |
| - std::string ipc_session_channel_name = GenerateUniqueTestChannelName(); |
| + mojo::edk::NamedPlatformHandle channel_handle( |
| + GenerateUniqueTestChannelName()); |
| + security_key_ipc_client_.SetIpcChannelHandleForTest(channel_handle); |
| ASSERT_TRUE(fake_ipc_server_.CreateChannel( |
| - ipc_session_channel_name, |
| + channel_handle, |
| /*request_timeout=*/base::TimeDelta::FromMilliseconds(500))); |
| - ASSERT_TRUE(IPC::Channel::IsNamedServerInitialized(ipc_session_channel_name)); |
| - fake_security_key_auth_handler_.set_ipc_security_key_channel_name( |
| - ipc_session_channel_name); |
| - |
| - // Set up the channel name for the initial IPC channel. |
| - std::string ipc_server_channel_name = GenerateUniqueTestChannelName(); |
| - fake_security_key_auth_handler_.set_ipc_server_channel_name( |
| - ipc_server_channel_name); |
| - security_key_ipc_client_.SetInitialIpcChannelNameForTest( |
| - ipc_server_channel_name); |
| - |
| - // Create the initial IPC channel and verify it was set up correctly. |
| - ASSERT_FALSE(security_key_ipc_client_.WaitForSecurityKeyIpcServerChannel()); |
| - fake_security_key_auth_handler_.CreateSecurityKeyConnection(); |
| - ASSERT_TRUE(IPC::Channel::IsNamedServerInitialized(ipc_server_channel_name)); |
| + |
| ASSERT_TRUE(security_key_ipc_client_.WaitForSecurityKeyIpcServerChannel()); |
| // Establish the IPC channel so we can begin sending and receiving security |
| @@ -286,37 +274,8 @@ TEST_F(SecurityKeyIpcClientTest, SendRequestBeforeEstablishingConnection) { |
| } |
| TEST_F(SecurityKeyIpcClientTest, NonExistentMainIpcServerChannel) { |
| - std::string ipc_server_channel_name(kNonexistentIpcChannelName); |
| - security_key_ipc_client_.SetInitialIpcChannelNameForTest( |
| - ipc_server_channel_name); |
| - |
| - // Attempt to establish the conection (should fail since the IPC channel does |
| - // not exist). |
| - security_key_ipc_client_.EstablishIpcConnection( |
| - base::Bind(&SecurityKeyIpcClientTest::OperationComplete, |
| - base::Unretained(this), /*failed=*/false), |
| - base::Bind(&SecurityKeyIpcClientTest::OperationComplete, |
| - base::Unretained(this), /*failed=*/true)); |
| - WaitForOperationComplete(); |
| - ASSERT_TRUE(operation_failed_); |
| -} |
| - |
| -TEST_F(SecurityKeyIpcClientTest, NonExistentIpcSessionChannel) { |
| - fake_security_key_auth_handler_.set_ipc_security_key_channel_name( |
| - kNonexistentIpcChannelName); |
| - |
| - // Set up the channel name for the initial IPC channel. |
| - std::string ipc_server_channel_name = GenerateUniqueTestChannelName(); |
| - fake_security_key_auth_handler_.set_ipc_server_channel_name( |
| - ipc_server_channel_name); |
| - security_key_ipc_client_.SetInitialIpcChannelNameForTest( |
| - ipc_server_channel_name); |
| - |
| - // Create the initial IPC channel and verify it was set up correctly. |
| - ASSERT_FALSE(security_key_ipc_client_.WaitForSecurityKeyIpcServerChannel()); |
| - fake_security_key_auth_handler_.CreateSecurityKeyConnection(); |
| - ASSERT_TRUE(IPC::Channel::IsNamedServerInitialized(ipc_server_channel_name)); |
| - ASSERT_TRUE(security_key_ipc_client_.WaitForSecurityKeyIpcServerChannel()); |
| + security_key_ipc_client_.SetIpcChannelHandleForTest( |
| + mojo::edk::NamedPlatformHandle(kNonexistentIpcChannelName)); |
| // Attempt to establish the conection (should fail since the IPC channel does |
| // not exist). |