Index: chrome/browser/chromeos/login/network_screen_browsertest.cc |
diff --git a/chrome/browser/chromeos/login/network_screen_browsertest.cc b/chrome/browser/chromeos/login/network_screen_browsertest.cc |
index e645c912685d141aa16b6ecedc44c3e61b608d17..fbe2168a07bc0025792787f2dd1606f1b8497647 100644 |
--- a/chrome/browser/chromeos/login/network_screen_browsertest.cc |
+++ b/chrome/browser/chromeos/login/network_screen_browsertest.cc |
@@ -10,7 +10,8 @@ |
#include "base/string_number_conversions.h" |
#include "base/utf_string_conversions.h" |
#include "chrome/browser/chromeos/cros/cros_library.h" |
-#include "chrome/browser/chromeos/cros/mock_login_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/cros/mock_network_library.h" |
#include "chrome/browser/chromeos/cros/network_library.h" |
#include "chrome/browser/chromeos/login/mock_screen_observer.h" |
@@ -46,20 +47,22 @@ class DummyButtonListener : public views::ButtonListener { |
class NetworkScreenTest : public WizardInProcessBrowserTest { |
public: |
NetworkScreenTest(): WizardInProcessBrowserTest("network"), |
- mock_login_library_(NULL), |
+ mock_session_manager_client_(NULL), |
mock_network_library_(NULL) { |
} |
protected: |
virtual void SetUpInProcessBrowserTestFixture() { |
+ DBusThreadManager::Initialize(); |
cros_mock_->InitStatusAreaMocks(); |
mock_network_library_ = cros_mock_->mock_network_library(); |
- mock_login_library_ = new MockLoginLibrary(); |
+ mock_session_manager_client_ = new MockSessionManagerClient(); |
+ DBusThreadManager::Get()->set_session_manager_client_for_testing( |
+ mock_session_manager_client_); |
cellular_.reset(new NetworkDevice("cellular")); |
- cros_mock_->test_api()->SetLoginLibrary(mock_login_library_, true); |
- EXPECT_CALL(*mock_login_library_, EmitLoginPromptReady()) |
+ EXPECT_CALL(*mock_session_manager_client_, EmitLoginPromptReady()) |
.Times(1); |
- EXPECT_CALL(*mock_login_library_,RequestRetrievePolicy(_,_)) |
+ EXPECT_CALL(*mock_session_manager_client_, RetrievePolicy(_)) |
.Times(AnyNumber()); |
// Minimal set of expectations needed on NetworkScreen initialization. |
@@ -131,7 +134,7 @@ class NetworkScreenTest : public WizardInProcessBrowserTest { |
virtual void TearDownInProcessBrowserTestFixture() { |
network_screen_->screen_observer_ = controller(); |
CrosInProcessBrowserTest::TearDownInProcessBrowserTestFixture(); |
- cros_mock_->test_api()->SetLoginLibrary(NULL, false); |
+ DBusThreadManager::Shutdown(); |
} |
void EmulateContinueButtonExit(NetworkScreen* network_screen) { |
@@ -145,7 +148,7 @@ class NetworkScreenTest : public WizardInProcessBrowserTest { |
} |
scoped_ptr<MockScreenObserver> mock_screen_observer_; |
- MockLoginLibrary* mock_login_library_; |
+ MockSessionManagerClient* mock_session_manager_client_; |
MockNetworkLibrary* mock_network_library_; |
scoped_ptr<NetworkDevice> cellular_; |
NetworkScreen* network_screen_; |