| Index: chrome/browser/chromeos/login/update_screen_browsertest.cc
|
| diff --git a/chrome/browser/chromeos/login/update_screen_browsertest.cc b/chrome/browser/chromeos/login/update_screen_browsertest.cc
|
| index a88e20515bee2c94088f700a0f9c73d0672b83f1..53f64bf50cad76665d91219747f70692fc90c000 100644
|
| --- a/chrome/browser/chromeos/login/update_screen_browsertest.cc
|
| +++ b/chrome/browser/chromeos/login/update_screen_browsertest.cc
|
| @@ -5,7 +5,7 @@
|
| #include "base/message_loop.h"
|
| #include "chrome/browser/chromeos/cros/mock_network_library.h"
|
| #include "chrome/browser/chromeos/cros/mock_update_library.h"
|
| -#include "chrome/browser/chromeos/dbus/dbus_thread_manager.h"
|
| +#include "chrome/browser/chromeos/dbus/mock_dbus_thread_manager.h"
|
| #include "chrome/browser/chromeos/dbus/mock_session_manager_client.h"
|
| #include "chrome/browser/chromeos/login/mock_screen_observer.h"
|
| #include "chrome/browser/chromeos/login/update_screen.h"
|
| @@ -28,22 +28,22 @@ static void RequestUpdateCheckSuccess(UpdateCallback callback, void* userdata) {
|
| class UpdateScreenTest : public WizardInProcessBrowserTest {
|
| public:
|
| UpdateScreenTest() : WizardInProcessBrowserTest("update"),
|
| - mock_session_manager_client_(NULL),
|
| mock_update_library_(NULL),
|
| mock_network_library_(NULL) {}
|
|
|
| protected:
|
| virtual void SetUpInProcessBrowserTestFixture() {
|
| - DBusThreadManager::Initialize();
|
| + MockDBusThreadManager* mock_dbus_thread_manager =
|
| + new MockDBusThreadManager;
|
| + DBusThreadManager::InitializeForTesting(mock_dbus_thread_manager);
|
| WizardInProcessBrowserTest::SetUpInProcessBrowserTestFixture();
|
| cros_mock_->InitStatusAreaMocks();
|
| cros_mock_->SetStatusAreaMocksExpectations();
|
| ASSERT_TRUE(CrosLibrary::Get()->EnsureLoaded());
|
|
|
| - mock_session_manager_client_ = new MockSessionManagerClient();
|
| - DBusThreadManager::Get()->set_session_manager_client_for_testing(
|
| - mock_session_manager_client_);
|
| - EXPECT_CALL(*mock_session_manager_client_, EmitLoginPromptReady())
|
| + MockSessionManagerClient* mock_session_manager_client
|
| + = mock_dbus_thread_manager->mock_session_manager_client();
|
| + EXPECT_CALL(*mock_session_manager_client, EmitLoginPromptReady())
|
| .Times(1);
|
|
|
| mock_update_library_ = new MockUpdateLibrary();
|
| @@ -90,14 +90,11 @@ class UpdateScreenTest : public WizardInProcessBrowserTest {
|
| DBusThreadManager::Shutdown();
|
| }
|
|
|
| - MockSessionManagerClient* mock_session_manager_client_;
|
| MockUpdateLibrary* mock_update_library_;
|
| MockNetworkLibrary* mock_network_library_;
|
|
|
| scoped_ptr<MockScreenObserver> mock_screen_observer_;
|
| UpdateScreen* update_screen_;
|
| - // DBusThreadManager needs a message loop.
|
| - MessageLoopForUI message_loop_;
|
|
|
| private:
|
| DISALLOW_COPY_AND_ASSIGN(UpdateScreenTest);
|
|
|