Index: remoting/host/chromoting_host_unittest.cc |
diff --git a/remoting/host/chromoting_host_unittest.cc b/remoting/host/chromoting_host_unittest.cc |
index edc5ac535a1584ddd788a9db80906dc233703b42..699efd1d47455983378bfd80aa4a8dcdebc8123b 100644 |
--- a/remoting/host/chromoting_host_unittest.cc |
+++ b/remoting/host/chromoting_host_unittest.cc |
@@ -68,22 +68,19 @@ class ChromotingHostTest : public testing::Test { |
virtual void SetUp() OVERRIDE { |
message_loop_proxy_ = base::MessageLoopProxy::current(); |
- ON_CALL(context_, main_message_loop()) |
- .WillByDefault(Return(&message_loop_)); |
- ON_CALL(context_, encode_message_loop()) |
- .WillByDefault(Return(&message_loop_)); |
- ON_CALL(context_, network_message_loop()) |
- .WillByDefault(Return(message_loop_proxy_.get())); |
- ON_CALL(context_, ui_message_loop()) |
- .WillByDefault(Return(message_loop_proxy_.get())); |
- EXPECT_CALL(context_, main_message_loop()) |
- .Times(AnyNumber()); |
- EXPECT_CALL(context_, encode_message_loop()) |
- .Times(AnyNumber()); |
- EXPECT_CALL(context_, network_message_loop()) |
- .Times(AnyNumber()); |
- EXPECT_CALL(context_, ui_message_loop()) |
- .Times(AnyNumber()); |
+ |
+ EXPECT_CALL(context_, ui_task_runner()) |
+ .Times(AnyNumber()) |
+ .WillRepeatedly(Return(message_loop_proxy_.get())); |
+ EXPECT_CALL(context_, capture_task_runner()) |
+ .Times(AnyNumber()) |
+ .WillRepeatedly(Return(message_loop_proxy_.get())); |
+ EXPECT_CALL(context_, encode_task_runner()) |
+ .Times(AnyNumber()) |
+ .WillRepeatedly(Return(message_loop_proxy_.get())); |
+ EXPECT_CALL(context_, network_task_runner()) |
+ .Times(AnyNumber()) |
+ .WillRepeatedly(Return(message_loop_proxy_.get())); |
scoped_ptr<Capturer> capturer(new CapturerFake()); |
event_executor_ = new MockEventExecutor(); |
@@ -207,22 +204,22 @@ class ChromotingHostTest : public testing::Test { |
desktop_environment_->capturer(), base::TimeDelta()); |
connection_ptr->set_host_stub(client); |
- context_.network_message_loop()->PostTask( |
+ context_.network_task_runner()->PostTask( |
FROM_HERE, base::Bind(&ChromotingHostTest::AddClientToHost, |
host_, client)); |
if (authenticate) { |
- context_.network_message_loop()->PostTask( |
+ context_.network_task_runner()->PostTask( |
FROM_HERE, base::Bind(&ClientSession::OnConnectionAuthenticated, |
base::Unretained(client), connection_ptr)); |
if (!reject) { |
- context_.network_message_loop()->PostTask( |
+ context_.network_task_runner()->PostTask( |
FROM_HERE, |
base::Bind(&ClientSession::OnConnectionChannelsConnected, |
base::Unretained(client), connection_ptr)); |
} |
} else { |
- context_.network_message_loop()->PostTask( |
+ context_.network_task_runner()->PostTask( |
FROM_HERE, base::Bind(&ClientSession::OnConnectionClosed, |
base::Unretained(client), connection_ptr, |
protocol::AUTHENTICATION_FAILED)); |