Index: chromeos/dbus/shill_client_unittest_base.cc |
diff --git a/chromeos/dbus/shill_client_unittest_base.cc b/chromeos/dbus/shill_client_unittest_base.cc |
index ab5937e2b46d27c462ae6a4c8e6437bd0d582190..ccc481ca1c6db8f8a7861d7334da26428da53a45 100644 |
--- a/chromeos/dbus/shill_client_unittest_base.cc |
+++ b/chromeos/dbus/shill_client_unittest_base.cc |
@@ -22,12 +22,6 @@ namespace chromeos { |
namespace { |
-// Runs the given task. This function is used to implement the mock bus. |
-void RunTask(const tracked_objects::Location& from_here, |
- const base::Closure& task) { |
- task.Run(); |
-} |
- |
// Pops a string-to-string dictionary from the reader. |
base::DictionaryValue* PopStringToStringDictionary( |
dbus::MessageReader* reader) { |
@@ -158,10 +152,10 @@ void ShillClientUnittestBase::SetUp() { |
GetObjectProxy(shill::kFlimflamServiceName, object_path_)) |
.WillOnce(Return(mock_proxy_.get())); |
- // Set an expectation so mock_bus's PostTaskToDBusThread() will run the |
- // given task. |
- EXPECT_CALL(*mock_bus_.get(), PostTaskToDBusThread(_, _)) |
- .WillRepeatedly(Invoke(&RunTask)); |
+ // Set an expectation so mock_bus's GetDBusTaskRunner will return the current |
+ // task runner. |
+ EXPECT_CALL(*mock_bus_.get(), GetDBusTaskRunner()) |
+ .WillRepeatedly(Return(message_loop_.message_loop_proxy())); |
// ShutdownAndBlock() will be called in TearDown(). |
EXPECT_CALL(*mock_bus_.get(), ShutdownAndBlock()).WillOnce(Return()); |