Index: chrome/browser/chromeos/power/power_prefs_unittest.cc |
diff --git a/chrome/browser/chromeos/power/power_prefs_unittest.cc b/chrome/browser/chromeos/power/power_prefs_unittest.cc |
index ff879e46c2902bff9ecb8f8c92e17b623e02b2ee..3e60eb7bdb9a223f136d29a1cc6ae7fd1bf109ac 100644 |
--- a/chrome/browser/chromeos/power/power_prefs_unittest.cc |
+++ b/chrome/browser/chromeos/power/power_prefs_unittest.cc |
@@ -23,7 +23,7 @@ |
#include "chrome/test/base/testing_profile.h" |
#include "chrome/test/base/testing_profile_manager.h" |
#include "chromeos/chromeos_switches.h" |
-#include "chromeos/dbus/fake_dbus_thread_manager.h" |
+#include "chromeos/dbus/dbus_thread_manager.h" |
#include "chromeos/dbus/fake_power_manager_client.h" |
#include "chromeos/dbus/power_manager/policy.pb.h" |
#include "chromeos/dbus/power_policy_controller.h" |
@@ -51,7 +51,6 @@ class PowerPrefsTest : public testing::Test { |
bool GetCurrentAllowScreenWakeLocks() const; |
TestingProfileManager profile_manager_; |
- FakeDBusThreadManager fake_dbus_thread_manager_; |
PowerPolicyController* power_policy_controller_; // Not owned. |
FakePowerManagerClient* fake_power_manager_client_; // Not owned. |
@@ -64,15 +63,19 @@ PowerPrefsTest::PowerPrefsTest() |
: profile_manager_(TestingBrowserProcess::GetGlobal()), |
power_policy_controller_(new PowerPolicyController), |
fake_power_manager_client_(new FakePowerManagerClient) { |
- fake_dbus_thread_manager_.SetPowerManagerClient( |
- scoped_ptr<PowerManagerClient>(fake_power_manager_client_)); |
- fake_dbus_thread_manager_.SetPowerPolicyController( |
- scoped_ptr<PowerPolicyController>(power_policy_controller_)); |
- power_policy_controller_->Init(&fake_dbus_thread_manager_); |
} |
void PowerPrefsTest::SetUp() { |
testing::Test::SetUp(); |
+ |
+ scoped_ptr<DBusThreadManagerTestHelper> dbus_helper = |
+ chromeos::DBusThreadManager::GetDBusThreadManagerTestHelper(); |
+ dbus_helper->SetPowerManagerClient( |
+ scoped_ptr<PowerManagerClient>(fake_power_manager_client_)); |
+ dbus_helper->SetPowerPolicyController( |
+ scoped_ptr<PowerPolicyController>(power_policy_controller_)); |
+ power_policy_controller_->Init(chromeos::DBusThreadManager::Get()); |
+ |
ASSERT_TRUE(profile_manager_.SetUp()); |
power_prefs_.reset(new PowerPrefs(power_policy_controller_)); |