| 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 f51b9aca497f4a7eeff1dd61894bee135fecbd7c..445fd07c3faf24fea02269b41b0daad29aa7a7fb 100644
|
| --- a/chrome/browser/chromeos/login/update_screen_browsertest.cc
|
| +++ b/chrome/browser/chromeos/login/update_screen_browsertest.cc
|
| @@ -2,9 +2,10 @@
|
| // Use of this source code is governed by a BSD-style license that can be
|
| // found in the LICENSE file.
|
|
|
| -#include "chrome/browser/chromeos/cros/mock_login_library.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_session_manager_client.h"
|
| #include "chrome/browser/chromeos/login/mock_screen_observer.h"
|
| #include "chrome/browser/chromeos/login/update_screen.h"
|
| #include "chrome/browser/chromeos/login/wizard_controller.h"
|
| @@ -26,20 +27,22 @@ static void RequestUpdateCheckSuccess(UpdateCallback callback, void* userdata) {
|
| class UpdateScreenTest : public WizardInProcessBrowserTest {
|
| public:
|
| UpdateScreenTest() : WizardInProcessBrowserTest("update"),
|
| - mock_login_library_(NULL),
|
| + mock_session_manager_client_(NULL),
|
| mock_update_library_(NULL),
|
| mock_network_library_(NULL) {}
|
|
|
| protected:
|
| virtual void SetUpInProcessBrowserTestFixture() {
|
| + DBusThreadManager::Initialize();
|
| WizardInProcessBrowserTest::SetUpInProcessBrowserTestFixture();
|
| cros_mock_->InitStatusAreaMocks();
|
| cros_mock_->SetStatusAreaMocksExpectations();
|
| ASSERT_TRUE(CrosLibrary::Get()->EnsureLoaded());
|
|
|
| - mock_login_library_ = new MockLoginLibrary();
|
| - cros_mock_->test_api()->SetLoginLibrary(mock_login_library_, true);
|
| - EXPECT_CALL(*mock_login_library_, EmitLoginPromptReady())
|
| + 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())
|
| .Times(1);
|
|
|
| mock_update_library_ = new MockUpdateLibrary();
|
| @@ -83,9 +86,10 @@ class UpdateScreenTest : public WizardInProcessBrowserTest {
|
| update_screen_->screen_observer_ = (controller());
|
| cros_mock_->test_api()->SetUpdateLibrary(NULL, true);
|
| WizardInProcessBrowserTest::TearDownInProcessBrowserTestFixture();
|
| + DBusThreadManager::Shutdown();
|
| }
|
|
|
| - MockLoginLibrary* mock_login_library_;
|
| + MockSessionManagerClient* mock_session_manager_client_;
|
| MockUpdateLibrary* mock_update_library_;
|
| MockNetworkLibrary* mock_network_library_;
|
|
|
|
|